NVIDIA-AI-IOT / tensorrt_plugin_generator

A simple tool that can generate TensorRT plugin code quickly.
MIT License
220 stars 35 forks source link

Adding cuda kernels #4

Open sandeepnmenon opened 1 year ago

sandeepnmenon commented 1 year ago

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 I want to call my cuda binding. How can I make the cuda code and header come in scope of the project created by

trtpg generate --yaml plugin.yml --output ./plugin_codes

zerollzeng commented 1 year ago

Sorry I miss this :-) You can include a kernel head files, just like you do in your development.