Skip to content

Commit fce74b6

Browse files
committed
fix: Add version checking for Torch Dynamo import
- Ensure `torch._dynamo` can be imported by validating Torch version - Intended to fix legacy CI run
1 parent 7433d4b commit fce74b6

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

py/torch_tensorrt/__init__.py

+5-2
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import sys
55
import platform
66
import warnings
7+
from packaging import version
78
from torch_tensorrt._version import (
89
__version__,
910
__cuda_version__,
@@ -93,8 +94,10 @@ def _find_lib(name, paths):
9394
from torch_tensorrt._TRTModuleNext import TRTModuleNext
9495

9596
from torch_tensorrt import fx
96-
from torch_tensorrt import dynamo
97-
from torch_tensorrt.dynamo import torch_compile
97+
98+
if version.parse(torch.__version__) >= version.parse("2.dev"):
99+
from torch_tensorrt import dynamo
100+
from torch_tensorrt.dynamo import torch_compile
98101

99102

100103
def _register_with_torch():

0 commit comments

Comments
 (0)