pytorch-lightning
Torch-Tensorrt Integration with LightningModule
#20808
Merged
Overview
Commits
60
Changes
View On GitHub

Torch-Tensorrt Integration with LightningModule #20808

Borda merged 60 commits into Lightning-AI:master from GdoongMathew:feat/to_tensorrt
GdoongMathew
GdoongMathew feat: add `to_tensorrt` in the `LightningModule`.
f589e31d
GdoongMathew feat: add `to_tensorrt` in the `LightningModule`.
14b9f294
GdoongMathew refactor: fix `to_tensorrt` impl
314c4630
GdoongMathew test: add test_torch_tensorrt.py
534c6c43
GdoongMathew Merge remote-tracking branch 'origin/feat/to_tensorrt' into feat/to_t…
e8a7fd26
GdoongMathew add dependency in test requirement.
5c01acb7
GdoongMathew GdoongMathew requested a review from lantiga lantiga 1 year ago
GdoongMathew GdoongMathew requested a review from Borda Borda 1 year ago
GdoongMathew GdoongMathew requested a review from tchaton tchaton 1 year ago
GdoongMathew GdoongMathew requested a review from justusschock justusschock 1 year ago
GdoongMathew GdoongMathew requested a review from ethanwharris ethanwharris 1 year ago
github-actions github-actions added pl
github-actions github-actions added dependencies
GdoongMathew update dependency in test requirement.
26d2788c
GdoongMathew GdoongMathew changed the title Torch-Tensorrt Integration with LightningModule WIP: Torch-Tensorrt Integration with LightningModule 1 year ago
GdoongMathew fix mypy error.
d42ebe4a
GdoongMathew fix: fix unittest.
958968da
GdoongMathew Merge branch 'master' into feat/to_tensorrt
87a048b6
GdoongMathew fix: fix unittest.
53257f96
GdoongMathew Merge remote-tracking branch 'origin/feat/to_tensorrt' into feat/to_t…
f0762330
GdoongMathew fix: fix type annotation.
07230712
GdoongMathew
GdoongMathew fix: fix runif tensorrt logic.
c97ae0d0
GdoongMathew test: add test `test_missing_tensorrt_package`.
484f9ced
GdoongMathew req: remove mac from the tensorrt dependency.
b35c60ca
GdoongMathew Merge branch 'master' into feat/to_tensorrt
e6f8097a
GdoongMathew fix: fix default device logics.
a9047fe7
GdoongMathew test: add test `test_tensorrt_with_wrong_default_device`.
a36907f3
GdoongMathew GdoongMathew changed the title WIP: Torch-Tensorrt Integration with LightningModule Torch-Tensorrt Integration with LightningModule 363 days ago
GdoongMathew fix: reorder the import sequence.
937dedf2
Borda
Borda
Borda commented on 2025-06-11
GdoongMathew feat: add exception when torch is below 2.2.0.
15f19619
GdoongMathew add unittests.
67546dc9
GdoongMathew Merge branch 'master' into feat/to_tensorrt
dcac65dd
GdoongMathew Merge branch 'master' into feat/to_tensorrt
d13e52d9
GdoongMathew Merge branch 'master' into feat/to_tensorrt
1c45011e
GdoongMathew torch-tensorrt deps.
9d92ddfa
GdoongMathew Merge branch 'master' into feat/to_tensorrt
8489571b
GdoongMathew fix: fix unittest `test_missing_tensorrt_package` to run only when mi…
f16deedc
GdoongMathew Merge branch 'master' into feat/to_tensorrt
398cc10a
github-actions github-actions added has conflicts
Borda
Borda approved these changes on 2025-08-08
GdoongMathew Merge branch 'master' into feat/to_tensorrt
1929cb5c
GdoongMathew Merge remote-tracking branch 'origin/feat/to_tensorrt' into feat/to_t…
c4103e15
github-actions github-actions removed has conflicts
GdoongMathew revert test dependencies.
42fafb7d
GdoongMathew add the minimum support in torch-tensorrt when testing.
7016bdb8
GdoongMathew add torch-tensorrt support when python is lower then py312.
69c87b35
GdoongMathew limit the torch-tensorrt condition again.
63e3cc72
GdoongMathew limit the torch-tensorrt condition again.
bf63faae
GdoongMathew docs: add tensorrt description in docstring.
02757a39
GdoongMathew loosen the torch-tensorrt version limits.
a2b36cdd
GdoongMathew update tensorrt version.
87b2f013
GdoongMathew
GdoongMathew Merge branch 'master' into feat/to_tensorrt
e0ad517c
GdoongMathew Update test.txt description
f9821f66
github-actions github-actions added has conflicts
GdoongMathew Merge branch 'master' into feat/to_tensorrt
817e1455
github-actions github-actions removed has conflicts
Borda Merge branch 'master' into feat/to_tensorrt
ce2cdda6
GdoongMathew
GdoongMathew update tensorrt source.
ffc466ab
GdoongMathew update test.txt
af70c86d
GdoongMathew GdoongMathew force pushed from d1eec19b to af70c86d 296 days ago
GdoongMathew update test.txt
3ae0978c
GdoongMathew update test.txt
553bea50
GdoongMathew
GdoongMathew Merge branch 'master' into feat/to_tensorrt
3dea3736
GdoongMathew
GdoongMathew move tensorrt to a dedicated requirement file.
25c5d09b
GdoongMathew fix: tmp skip test_tensorrt_export_reload in torch-tensorrt 2.8.0
c59b4a12
github-actions github-actions added ci
GdoongMathew ci: extra-index-url
550d01d5
GdoongMathew reverse requirements and ci.
62d7666d
GdoongMathew reverse requirements and ci.
d8ce87d2
GdoongMathew update test.txt
076564a8
GdoongMathew ci: add extra-index
4f3c6154
GdoongMathew ci: use find-links instead. works on my computer...
a1bfa06e
GdoongMathew fix: fix bug in torch-tensorrt 2.8.0
c4d2263e
GdoongMathew add find links in ci test.
aa744ed4
GdoongMathew
GdoongMathew GdoongMathew requested a review from Borda Borda 295 days ago
Borda
Borda approved these changes on 2025-08-14
Borda chlog
f02970c7
github-actions github-actions added has conflicts
Borda Merge branch 'master' into feat/to_tensorrt
ea9e38cf
github-actions github-actions removed has conflicts
Borda Borda merged 39e24f5d into master 294 days ago
GdoongMathew GdoongMathew deleted the feat/to_tensorrt branch 294 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
ci pl dependencies
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum