#3015: Fix std_hw

tenstorrent · Apr 22, 2024 · d9150f0 · d9150f0
1 parent 201cc3b
commit d9150f0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_stats_std_hw.py b/tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_stats_std_hw.py
@@ -8,7 +8,7 @@
 import tt_lib as ttl
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
-from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc
+from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc, comp_allclose
 from tests.tt_eager.python_api_testing.sweep_tests.tt_lib_ops import std_hw as tt_std_hw
 
 
@@ -31,7 +31,7 @@ def run_std_hw_tests(input_shape, dtype, dlayout, in_mem_config, out_mem_config,
     )
 
     # compare tt and golden outputs
-    success, pcc_value = comp_pcc(ref_value, tt_result)
+    success, pcc_value = comp_allclose(ref_value, tt_result, atol=0.06)
     logger.debug(pcc_value)
 
     assert success

diff --git a/...ng/sweep_tests/test_configs/ci_sweep_tests_broken/wormhole/pytorch_stats_std_hw_test.yaml b/...ng/sweep_tests/test_configs/ci_sweep_tests_broken/wormhole/pytorch_stats_std_hw_test.yaml
@@ -16,7 +16,9 @@ test-list:
           low: -10
           high: 10
       comparison:
-        function: comp_pcc
+        function: comp_allclose
+        args:
+          atol: 0.1
       args-gen: gen_dtype_layout_device
       output-file: stats_std_hw_sweep.csv
       args:
@@ -42,7 +44,9 @@ test-list:
           low: -10
           high: 10
       comparison:
-        function: comp_pcc
+        function: comp_allclose
+        args:
+          atol: 0.1
       args-gen: gen_dtype_layout_device
       output-file: stats_std_hw_sweep.csv
       args: