#6390: Update the test file and compare function

tenstorrent · Mar 18, 2024 · 61eacf2 · 61eacf2
1 parent b1d3e1b
commit 61eacf2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_l1_loss_sum.py b/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_l1_loss_sum.py
@@ -10,7 +10,7 @@
 
 from tests.ttnn.utils_for_testing import assert_with_pcc
 from tests.ttnn.python_api_testing.sweep_tests import ttnn_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_allclose
 
 
 def run_l1_loss_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device):
@@ -38,7 +38,7 @@ def run_l1_loss_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_con
     tt_result = tt_result.squeeze(0)
     tt_result = tt_result.squeeze(0)
 
-    success, pcc_value = comp_pcc(ref_value, tt_result)
+    success, pcc_value = comp_allclose(ref_value, tt_result[0, 0], atol=4, rtol=1e-1)
     logger.debug(pcc_value)
     logger.debug(success)
 
@@ -48,7 +48,7 @@ def run_l1_loss_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_con
 test_sweep_args = [
     (
         [(224, 128), (224, 128)],
-        [ttnn.bfloat8_b, ttnn.bfloat8_b],
+        [ttnn.bfloat16, ttnn.bfloat16],
         [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT],
         [ttnn.DRAM_MEMORY_CONFIG, ttnn.DRAM_MEMORY_CONFIG, ttnn.DRAM_MEMORY_CONFIG],
         ttnn.DRAM_MEMORY_CONFIG,