Skip to content

Commit 10917bf

Browse files
authored
small fix: Packaging version switch (#2315)
1 parent a03a585 commit 10917bf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

py/torch_tensorrt/dynamo/backend/backends.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
from packaging import version
2121

2222
# Modify import location of utilities based on Torch version
23-
if version.parse(torch_tensorrt.sanitized_torch_version()) <= version.parse("2.1.0"):
23+
if version.parse(torch_tensorrt.sanitized_torch_version()) < version.parse("2.1.1"):
2424
from torch._inductor.freezing import ConstantFolder, replace_node_with_constant
2525
else:
2626
from torch._inductor.constant_folding import (

0 commit comments

Comments
 (0)