关于plugin问题enqueue问题？

dlunion / tensorRTIntegrate

TensorRT ONNX Plugin、Inference、Compile

465 stars 114 forks source link

关于plugin问题enqueue问题？ #41

Open linhezheng19 opened 4 years ago

linhezheng19 commented 4 years ago

在PluginV2Ext中有enqueue方法需要重写，您在自定义的TRTPlugin中又重定义了一个enqueue，输入参数和原来的大不一样，在这个DEFINE_BUILTIN_OP_IMPORTER宏中虽然有转类型的操作，但是是怎么调用到自定义的那个enqueue方法的呢？看了很久没有明白，求大佬指点~

hopef commented 4 years ago

在TRTPlugin里面，也就是onnxplugin.cpp:254行开始，调用了mappingToGTensor、对数据类型做了封装，然后再调用enqueue（也就是自定义的 enqueue）是这么做到的

linhezheng19 commented 4 years ago

明白了，多谢大佬～