Skip to content

Commit 7d77bd2

Browse files
committed
changing the compilation setting function call in run_test_compare_tensor_attributes_only
1 parent 2ee77fa commit 7d77bd2

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/py/dynamo/conversion/harness.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -288,7 +288,9 @@ def run_test_compare_tensor_attributes_only(
288288
# Previous instance of the interpreter auto-casted 64-bit inputs
289289
# We replicate this behavior here
290290
compilation_settings = CompilationSettings(
291-
precision=precision, truncate_long_and_double=True
291+
enabled_precisions={dtype._from(precision)},
292+
truncate_long_and_double=True,
293+
debug=True,
292294
)
293295

294296
interp = TRTInterpreter(

0 commit comments

Comments
 (0)