changing the data type in interpretor setting

apbose · apbose · commit f503b61a37af · 2024-04-25T17:41:01.000-07:00
diff --git a/tests/py/dynamo/conversion/harness.py b/tests/py/dynamo/conversion/harness.py
@@ -273,7 +273,7 @@ def run_test_comparator(
         inputs,
         expected_ops,
         comparators: List[Tuple[Callable, List]],
-        precision=torch.float32,
+        precision=torch.float,
         output_dtypes=None,
         use_dynamo_tracer=False,
         enable_passes=False,
diff --git a/tests/py/dynamo/conversion/test_rand_aten.py b/tests/py/dynamo/conversion/test_rand_aten.py
@@ -121,7 +121,7 @@ def forward(self, x):
             x.dtype == y.dtype if check_dtype else True
         )
         expected_ops = []
-        # TRT-np returns int32 while torch returns float32
+        # TRT-TRT returns int32  while torch returns int64
         self.run_test_comparator(
             rand_model,
             inputs,

Original file line number	Diff line number	Diff line change
`@@ -121,7 +121,7 @@ def forward(self, x):`
`121`	`121`	`x.dtype == y.dtype if check_dtype else True`
`122`	`122`	`)`
`123`	`123`	`expected_ops = []`
`124`		`- # TRT-np returns int32 while torch returns float32`
	`124`	`+ # TRT-TRT returns int32 while torch returns int64`
`125`	`125`	`self.run_test_comparator(`
`126`	`126`	`rand_model,`
`127`	`127`	`inputs,`