search

TrojanXu / onnxparser-trt-plugin-sample

A sample for onnxparser working with trt user defined plugins for TRT7.0
Apache License 2.0
166 stars 36 forks source link

Make it work in TRT6 #12

Open AromaticJ opened 3 years ago

AromaticJ commented 3 years ago

Hi @TrojanXu,

Thanks for your wonderful work, it helps me immensely!

I have implemented it in TRT7 successfully on PC. However, for my platform is the Nvidia Drive Xavier, which is not able to support or upgrade to TRT7, I have to make it work in TRT6. I have read your code test_plugin_result.py, in my opinion, you replaced GridSampler with TRT_PluginV2 which belongs to TRT7 ops set by using onnx-graphsurgeon of TRT7. Is there possibility to replace GridSampler with an op belongs to TRT6 by using onnx-graphsurgeon of TRT7, and save the modified .onnx file, then using the modified .onnx file directly in TRT6? For I think the TRT7 have to be used just because we have to use onnx-graphsurgeon.

I have compiled your plugin GridSampler in TRT6 without error occured.

Thanks and Regards, AJ.