From 3e32efc6585d4d2ec6ae8d6543197ae7c678690f Mon Sep 17 00:00:00 2001 From: VirdhatchaniKN Date: Mon, 22 Apr 2024 13:10:53 +0000 Subject: [PATCH] #3015: Fix std_hw --- .../non_working_unit_tests/grayskull/test_stats_std_hw.py | 4 ++-- .../grayskull/pytorch_stats_std_hw_test.yaml | 8 ++++++-- .../wormhole/pytorch_stats_std_hw_test.yaml | 8 ++++++-- 3 files changed, 14 insertions(+), 6 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 index 29fd8e74e96..359c92c0a58 100644 --- 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.1) logger.debug(pcc_value) assert success diff --git a/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/grayskull/pytorch_stats_std_hw_test.yaml b/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/grayskull/pytorch_stats_std_hw_test.yaml index e7444f0ff05..5c5a4ba6a89 100644 --- a/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/grayskull/pytorch_stats_std_hw_test.yaml +++ b/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/grayskull/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: diff --git a/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/wormhole/pytorch_stats_std_hw_test.yaml b/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/wormhole/pytorch_stats_std_hw_test.yaml index e7444f0ff05..5c5a4ba6a89 100644 --- a/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/wormhole/pytorch_stats_std_hw_test.yaml +++ b/tests/tt_eager/python_api_testing/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: