chore: Address test failures

Signed-off-by: Dheeraj Peri <[email protected]>

Author: peri044

py/torch_tensorrt/dynamo/__init__.py

@@ -1,3 +1,7 @@
-from ._settings import *
-from .compile import compile
-from .aten_tracer import trace
+from packaging import version
+from torch_tensorrt._util import sanitized_torch_version
+
+if version.parse(sanitized_torch_version()) >= version.parse("2.1.dev"):
+    from ._settings import *
+    from .compile import compile
+    from .aten_tracer import trace

tests/py/ts/models/test_models.py

@@ -25,6 +25,7 @@ def test_resnet18(self):
                 "gpu_id": 0,
             },
             "enabled_precisions": {torch.float},
+            "ir": "ts",
         }
 
         trt_mod = torchtrt.compile(self.model, **compile_spec)
@@ -49,6 +50,7 @@ def test_mobilenet_v2(self):
                 "gpu_id": 0,
             },
             "enabled_precisions": {torch.float},
+            "ir": "ts",
         }
 
         trt_mod = torchtrt.compile(self.model, **compile_spec)
@@ -75,6 +77,7 @@ def test_efficientnet_b0(self):
                 "gpu_id": 0,
             },
             "enabled_precisions": {torch.float},
+            "ir": "ts",
         }
 
         trt_mod = torchtrt.compile(self.model, **compile_spec)
@@ -107,6 +110,7 @@ def test_bert_base_uncased(self):
             },
             "enabled_precisions": {torch.float},
             "truncate_long_and_double": True,
+            "ir": "ts",
         }
         with torchtrt.logging.errors():
             trt_mod = torchtrt.ts.compile(self.model, **compile_spec)
@@ -137,6 +141,7 @@ def test_resnet18_half(self):
                 "gpu_id": 0,
             },
             "enabled_precisions": {torch.half},
+            "ir": "ts",
         }
 
         trt_mod = torchtrt.compile(self.scripted_model, **compile_spec)

tests/py/ts/models/test_multiple_registered_engines.py

@@ -27,6 +27,7 @@ def test_multiple_engines(self):
                 "gpu_id": 0,
             },
             "enabled_precisions": {torch.float},
+            "ir": "ts",
         }
         rn18_trt_mod = torchtrt.compile(self.resnet18, **compile_spec)
         rn50_trt_mod = torchtrt.compile(self.resnet50, **compile_spec)