removing precision in harness.py interpreter compilationsettings and interpreter.run

apbose · apbose · commit 3710b86b5fc9 · 2024-04-29T15:54:55.000-07:00
diff --git a/tests/py/dynamo/conversion/harness.py b/tests/py/dynamo/conversion/harness.py
@@ -140,9 +140,7 @@ def run_test_custom_compare_results(
             if len(expected_ops):
                 self.assert_has_op(mod, expected_ops)
 
-            interpreter_result = interpreter.run(
-                precision=torch.half if fp16_mode else torch.float
-            )
+            interpreter_result = interpreter.run()
             trt_mod = PythonTorchTensorRTModule(
                 interpreter_result.engine,
                 interpreter_result.input_names,
@@ -288,7 +286,9 @@ def run_test_compare_tensor_attributes_only(
         # Previous instance of the interpreter auto-casted 64-bit inputs
         # We replicate this behavior here
         compilation_settings = CompilationSettings(
-            precision=precision, truncate_long_and_double=True
+            enabled_precisions={dtype._from(precision)},
+            truncate_long_and_double=True,
+            debug=True,
         )
 
         interp = TRTInterpreter(
