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.

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

Support TensorRT export with dynamo #6577

Open
binliunls opened this issue Jun 4, 2023 · 0 comments · May be fixed by #6578
Open

Support TensorRT export with dynamo #6577

binliunls opened this issue Jun 4, 2023 · 0 comments · May be fixed by #6578

Comments

@binliunls
Copy link
Contributor

Is your feature request related to a problem? Please describe.
PyTorch introduced the dynamo as a part to compile PyTorch models to different backends. And it also works for the Torch-TensorRT compiler and the Torch-TensorRT team is working on supporting it. Therefore, it would be better to have MONAI trt_export API also support this feature.

Describe the solution you'd like
Add the use_torchscript parameter to the trt_export API to choose whether to use the torchscript as IR. Because the dynamo supports to directly export models from pytorch models instead of the torchscript models.

@binliunls binliunls linked a pull request Jun 4, 2023 that will close this issue
7 tasks
@ericspod ericspod moved this to Todo in MONAI v1.3 Jul 11, 2023
@wyli wyli removed the status in MONAI v1.3 Jul 18, 2023
@wyli wyli removed this from MONAI v1.3 Oct 13, 2023
@wyli wyli added this to MONAI v1.4 Oct 13, 2023
@wyli wyli moved this to Todo in MONAI v1.4 Oct 13, 2023
@wyli wyli removed the status in MONAI v1.4 Nov 3, 2023
@vikashg vikashg closed this as completed Dec 20, 2023
@KumoLiu KumoLiu reopened this Dec 27, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
No open projects
Status: Done
Development

Successfully merging a pull request may close this issue.

3 participants