From e9aa47a56fb5e7c7d33dcc21736fa91151b89503 Mon Sep 17 00:00:00 2001 From: "Zewen (Evan) Li" Date: Thu, 2 May 2024 18:26:41 -0700 Subject: [PATCH] fix: a bug in func run_test_compare_tensor_attributes_only (#2809) --- tests/py/dynamo/conversion/harness.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/py/dynamo/conversion/harness.py b/tests/py/dynamo/conversion/harness.py index d3d3580359..7bed68a11e 100644 --- a/tests/py/dynamo/conversion/harness.py +++ b/tests/py/dynamo/conversion/harness.py @@ -289,7 +289,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_double=True, debug=True, )