minor fix: Update default minimum torch version for aten tracer (#1747)

pytorch · Mar 17, 2023 · a1d4af0 · a1d4af0
1 parent 497d8ac
commit a1d4af0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/py/torch_tensorrt/fx/tracer/dispatch_tracer/aten_tracer.py b/py/torch_tensorrt/fx/tracer/dispatch_tracer/aten_tracer.py
@@ -91,7 +91,7 @@ def setting_python_recursive_limit(limit: int = 10000) -> Generator[None, None,
         sys.setrecursionlimit(default)
 
 
-@req_torch_version("2.0")
+@req_torch_version("2.dev")
 def dynamo_trace(
     f: Callable[..., Value],
     # pyre-ignore
@@ -127,13 +127,13 @@ def dynamo_trace(
             ) from exc
 
 
-@req_torch_version("2.0")
+@req_torch_version("2.dev")
 def trace(f, args, *rest):
     graph_module, guards = dynamo_trace(f, args, True, "symbolic")
     return graph_module, guards
 
 
-@req_torch_version("2.0")
+@req_torch_version("2.dev")
 def opt_trace(f, args, *rest):
     """
     Optimized trace with necessary passes which re-compose some ops or replace some ops