TensorRT
TensorRT copied to clipboard
pytorch
Wrap ScriptFunctions in modules to compile
Add a section to _compiler.py that wraps ScriptFunctions in Modules to compile the graph.
This issue has not seen activity for 30 days, Remove stale label or comment or this will be closed in 5 days
![github-actions[bot] avatar](https://avatars.githubusercontent.com/in/15368?v=4 "Published by a pub.dev verified publisher"){kind=small}
This issue has not seen activity for 30 days, Remove stale label or comment or this will be closed in 5 days
This issue has not seen activity for 30 days, Remove stale label or comment or this will be closed in 5 days
This issue has not seen activity for 30 days, Remove stale label or comment or this will be closed in 5 days
This issue has not seen activity for 90 days, Remove stale label or comment or this will be closed in 10 days
I would like to work on this.
https://github.com/NVIDIA/TRTorch/blob/2fbbbc1c2cbe672a77aba7a0291059485f6ecc84/py/trtorch/_compiler.py#L135
In my understanding, TRTorch should be able to automatically wrap the function in a module and proceed with compiling the module. Do you have any other note for me @narendasan
yeah, I am not entirely sure about the relationship between ScriptFunctions and ScriptModules. Heres how we wrap modules for Python coming from C++ for embedding engines https://github.com/NVIDIA/TRTorch/blob/2fbbbc1c2cbe672a77aba7a0291059485f6ecc84/py/trtorch/_compiler.py#L140. Not sure if it will be the same here
This issue has not seen activity for 90 days, Remove stale label or comment or this will be closed in 10 days
This issue has not seen activity for 90 days, Remove stale label or comment or this will be closed in 10 days
This issue has not seen activity for 90 days, Remove stale label or comment or this will be closed in 10 days
This issue has not seen activity for 90 days, Remove stale label or comment or this will be closed in 10 days
This issue has not seen activity for 90 days, Remove stale label or comment or this will be closed in 10 days
This issue has not seen activity for 90 days, Remove stale label or comment or this will be closed in 10 days