Relax atol precision

tenstorrent · Jan 7, 2025 · efc8f83 · efc8f83
1 parent 16fb250
commit efc8f83
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/torch/test_constant_fold.py b/tests/torch/test_constant_fold.py
@@ -89,4 +89,6 @@ def forward(self, x):
     cc = CompilerConfig()
     cc.enable_consteval = True
     cc.consteval_parameters = True
-    verify_module(Basic(), input_shapes=[(32, 32)], compiler_config=cc)
+    verify_module(
+        Basic(), input_shapes=[(32, 32)], compiler_config=cc, required_atol=0.02
+    )