From 0360a5e31ee72df579cc53824fd53671066e66ff Mon Sep 17 00:00:00 2001 From: Borys Bradel Date: Thu, 23 Jan 2025 03:43:46 +0000 Subject: [PATCH] #16720: adjust pcc for max test on GS --- tests/ttnn/unit_tests/operations/test_max.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/ttnn/unit_tests/operations/test_max.py b/tests/ttnn/unit_tests/operations/test_max.py index 6991ab417a7d..d5fa0fe26249 100644 --- a/tests/ttnn/unit_tests/operations/test_max.py +++ b/tests/ttnn/unit_tests/operations/test_max.py @@ -8,7 +8,7 @@ import ttnn from tests.ttnn.utils_for_testing import assert_with_pcc -from models.utility_functions import torch_random +from models.utility_functions import torch_random, is_grayskull @pytest.mark.parametrize("batch_size", [1, 16, 1, 16]) @@ -129,4 +129,5 @@ def test_max_dim(device, input_shape_and_dim, keepdim): output_tensor = ttnn.to_torch(output_tensor) - assert_with_pcc(torch_output_tensor, output_tensor) + pcc = 0.999 if is_grayskull() else 0.9999 + assert_with_pcc(torch_output_tensor, output_tensor, pcc=pcc)