Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

✨[Bug] Add lowering, decompositions, nn.Module support for convert_module_to_trt_engine API #2692

Closed
peri044 opened this issue Mar 15, 2024 · 0 comments · Fixed by #2728
Closed

✨[Bug] Add lowering, decompositions, nn.Module support for convert_module_to_trt_engine API #2692

peri044 opened this issue Mar 15, 2024 · 0 comments · Fixed by #2728
Assignees
@zewenli98
Labels
bug Something isn't working

Comments

@peri044
Copy link
Collaborator

peri044 commented Mar 15, 2024

Is your feature request related to a problem? Please describe.
Add lowering, decompositions support, nn.Module support for convert_module_to_trt_engine API. Currently the convert_module_to_trt_engine API assumes the input is a graph_module

Describe the solution you'd like

Describe alternatives you've considered

Additional context
@peri044 peri044 added the feature request New feature or request label Mar 15, 2024
@peri044 peri044 assigned narendasan and zewenli98 and unassigned narendasan Mar 15, 2024
@peri044 peri044 changed the title ✨[Feature] Add lowering, decompositions support, nn.Module support for convert_module_to_trt_engine API ✨[Feature] Add lowering, decompositions, nn.Module support for convert_module_to_trt_engine API Mar 15, 2024
@narendasan narendasan added bug Something isn't working and removed feature request New feature or request labels Mar 19, 2024
@peri044 peri044 changed the title ✨[Feature] Add lowering, decompositions, nn.Module support for convert_module_to_trt_engine API ✨[Bug] Add lowering, decompositions, nn.Module support for convert_module_to_trt_engine API Mar 20, 2024
@zewenli98 zewenli98 mentioned this issue Apr 5, 2024
Merged
7 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@zewenli98 zewenli98

Labels
bug Something isn't working
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

fix: convert_module_to_trt_engine zewenli98/TensorRT
3 participants
@narendasan @peri044 @zewenli98