From 7caf1f7226bb8b874fd646ff4321281531c626e3 Mon Sep 17 00:00:00 2001
From: gs-olive <113141689+gs-olive@users.noreply.github.com>
Date: Fri, 17 Mar 2023 12:52:20 -0700
Subject: [PATCH] fix: Update default minimum torch version for aten tracer

---
 py/torch_tensorrt/fx/tracer/dispatch_tracer/aten_tracer.py | 3 +++
 1 file changed, 3 insertions(+)

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 2a252bd965..081175aff1 100644
--- a/py/torch_tensorrt/fx/tracer/dispatch_tracer/aten_tracer.py
+++ b/py/torch_tensorrt/fx/tracer/dispatch_tracer/aten_tracer.py
@@ -91,6 +91,7 @@ def setting_python_recursive_limit(limit: int = 10000) -> Generator[None, None,
         sys.setrecursionlimit(default)
 
 
+@req_torch_version("2.dev")
 def dynamo_trace(
     f: Callable[..., Value],
     # pyre-ignore
@@ -131,11 +132,13 @@ def dynamo_trace(
             ) from exc
 
 
+@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.dev")
 def opt_trace(f, args, *rest):
     """
     Optimized trace with necessary passes which re-compose some ops or replace some ops