We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a03a585 commit 10917bfCopy full SHA for 10917bf
py/torch_tensorrt/dynamo/backend/backends.py
@@ -20,7 +20,7 @@
20
from packaging import version
21
22
# Modify import location of utilities based on Torch version
23
-if version.parse(torch_tensorrt.sanitized_torch_version()) <= version.parse("2.1.0"):
+if version.parse(torch_tensorrt.sanitized_torch_version()) < version.parse("2.1.1"):
24
from torch._inductor.freezing import ConstantFolder, replace_node_with_constant
25
else:
26
from torch._inductor.constant_folding import (
0 commit comments