Skip to content

Commit 1ec8153

Browse files
gs-olivebowang007
authored andcommitted
minor fix: Update default minimum torch version for aten tracer (#1747)
1 parent 4e8da34 commit 1ec8153

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

py/torch_tensorrt/fx/tracer/dispatch_tracer/aten_tracer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def setting_python_recursive_limit(limit: int = 10000) -> Generator[None, None,
9191
sys.setrecursionlimit(default)
9292

9393

94-
@req_torch_version("2.0")
94+
@req_torch_version("2.dev")
9595
def dynamo_trace(
9696
f: Callable[..., Value],
9797
# pyre-ignore
@@ -127,13 +127,13 @@ def dynamo_trace(
127127
) from exc
128128

129129

130-
@req_torch_version("2.0")
130+
@req_torch_version("2.dev")
131131
def trace(f, args, *rest):
132132
graph_module, guards = dynamo_trace(f, args, True, "symbolic")
133133
return graph_module, guards
134134

135135

136-
@req_torch_version("2.0")
136+
@req_torch_version("2.dev")
137137
def opt_trace(f, args, *rest):
138138
"""
139139
Optimized trace with necessary passes which re-compose some ops or replace some ops

0 commit comments

Comments
 (0)