Skip to content

✨[Feature] Toggle usage of TRTModuleNext based on C++ availability #2019

Closed
@gs-olive

Description

@gs-olive

Description

Based on this comment by @narendasan, the usage of the TRTModuleNext functionality, now named TorchTensorRTModule, could be automatically toggled based on the availability of the C++ functionality, like so:

try: 
    import TorchTensorRTModule
    USE_PYTHON_RT = False
except:
    USE_PYTHON_RT = True

Related Issues

#1993, #1943

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions