tensorrt_plugin_generator icon indicating copy to clipboard operation
tensorrt_plugin_generator copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA-AI-IOT
→

Metadata

A simple tool that can generate TensorRT plugin code quickly.

Results 4 tensorrt_plugin_generator issues
Sort by recently updated
recently updated
newest added

Debugging the Custom Plugins

13 comment

Thank you so much for this wonderful repository. This helped to create custom plugin library in the most efficient way. Is there a way to test the enqueue function because...

jeethesh-pai avatar jeethesh-pai

Error Code 9: Internal Error (node_of_bevpool_out: could not find any supported formats consistent with input/output data types)

1 comment

log: ```shell ----> debug debug debug debug debug debug debug debug debug debug debug debug debug debug debug debug debug

lDarryll avatar lDarryll

Adding cuda kernels

1 comment

I was able to create a plugin and this repo really helped with the boiler plate code. One addition that I want to make in my `enqueue` function is that...

sandeepnmenon avatar sandeepnmenon

question

int64 should be supported

int64 is a very common data type and should be supported. https://github.com/NVIDIA-AI-IOT/tensorrt_plugin_generator/blob/81b95cf4902cc8a2ba0054c8f84dbd1f5bded6a1/tpg/config.py#L39

demuxin avatar demuxin

← Metadata

201
Stars
36
Forks
Watchers

Owner

verified publisher icon NVIDIA-AI-IOT

Metadata

A simple tool that can generate TensorRT plugin code quickly.

Back