diff --git a/tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_stats_var_hw.py b/tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_stats_var_hw.py index a30f8c0ccba6..e976035c4118 100644 --- a/tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_stats_var_hw.py +++ b/tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_stats_var_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 var_hw as tt_var_hw @@ -31,7 +31,7 @@ def run_var_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=1.7) 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_var_hw_test.yaml b/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/grayskull/pytorch_stats_var_hw_test.yaml index b58d36206028..924fe97234e7 100644 --- a/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/grayskull/pytorch_stats_var_hw_test.yaml +++ b/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/grayskull/pytorch_stats_var_hw_test.yaml @@ -16,7 +16,9 @@ test-list: low: -10 high: 10 comparison: - function: comp_pcc + function: comp_allclose + args: + atol: 1.7 args-gen: gen_dtype_layout_device output-file: stats_var_hw_sweep.csv args: @@ -42,7 +44,9 @@ test-list: low: -10 high: 10 comparison: - function: comp_pcc + function: comp_allclose + args: + atol: 1.7 args-gen: gen_dtype_layout_device output-file: stats_var_hw_sweep.csv args: diff --git a/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/wormhole/pytorch_stats_var_hw_test.yaml b/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/wormhole/pytorch_stats_var_hw_test.yaml index b58d36206028..924fe97234e7 100644 --- a/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/wormhole/pytorch_stats_var_hw_test.yaml +++ b/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/wormhole/pytorch_stats_var_hw_test.yaml @@ -16,7 +16,9 @@ test-list: low: -10 high: 10 comparison: - function: comp_pcc + function: comp_allclose + args: + atol: 1.7 args-gen: gen_dtype_layout_device output-file: stats_var_hw_sweep.csv args: @@ -42,7 +44,9 @@ test-list: low: -10 high: 10 comparison: - function: comp_pcc + function: comp_allclose + args: + atol: 1.7 args-gen: gen_dtype_layout_device output-file: stats_var_hw_sweep.csv args: