uncommenting the tests

pytorch · May 2, 2024 · 3844d35 · 3844d35
1 parent c4de771
commit 3844d35
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/py/dynamo/conversion/harness.py b/tests/py/dynamo/conversion/harness.py
@@ -249,7 +249,7 @@ def run_test(
             input = inputs[num_input]
             if input.dtype in (torch.int64, torch.float64):
                 dtype_32bit = (
-                    torch.int32 if (input.dtype == torch.int64) else torch.int64
+                    torch.int32 if (input.dtype == torch.int64) else torch.float32
                 )
                 # should we modify graph here to insert clone nodes?
                 # ideally not required
@@ -259,7 +259,7 @@ def run_test(
                         input.to(dtype_32bit),
                     ]
                     + list(trt_inputs[num_input + 1 :])
-                )        
+                )
 
         trt_input_specs = [Input.from_tensor(i) for i in trt_inputs]
         input_specs = [Input.from_tensor(i) for i in inputs]
@@ -270,7 +270,7 @@ def run_test(
                 mod,
                 input_specs,
                 compilation_settings.device,
-                truncate_long_and_double=compilation_settings.truncate_long_and_double,
+                truncate_double=compilation_settings.truncate_double,
             )
 
         _LOGGER.debug(f"Compilation settings: {compilation_settings}")
@@ -316,7 +316,7 @@ def run_test_compare_tensor_attributes_only(
         # We replicate this behavior here
         compilation_settings = CompilationSettings(
             enabled_precisions={dtype._from(precision)},
-            truncate_long_and_double=True,
+            truncate_and_double=True,
             debug=True,
         )