From a1d4af0f8070ba8ce570aac7f4cd0c35f97cae19 Mon Sep 17 00:00:00 2001 From: George S <113141689+gs-olive@users.noreply.github.com> Date: Fri, 17 Mar 2023 13:39:57 -0700 Subject: [PATCH] minor fix: Update default minimum torch version for aten tracer (#1747) --- py/torch_tensorrt/fx/tracer/dispatch_tracer/aten_tracer.py | 6 +++--- 1 file changed, 3 insertions(+), 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 index 6c12b80f8b..06717e400c 100644 --- 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