From 7514b14f98c9a62bbe0377633aa3fcda8f0211b9 Mon Sep 17 00:00:00 2001 From: Artem Yerofieiev Date: Wed, 22 May 2024 21:43:43 +0000 Subject: [PATCH] #8364: Disable implicit fallback for ttnn.split --- tests/ttnn/unit_tests/operations/test_split.py | 1 + ttnn/ttnn/operations/data_movement.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/ttnn/unit_tests/operations/test_split.py b/tests/ttnn/unit_tests/operations/test_split.py index 32ab7fd9f4e..03b78cfb567 100644 --- a/tests/ttnn/unit_tests/operations/test_split.py +++ b/tests/ttnn/unit_tests/operations/test_split.py @@ -12,6 +12,7 @@ from models.utility_functions import skip_for_wormhole_b0 +@pytest.mark.skip(reason="ttnn.split is not implemented") @pytest.mark.parametrize("h", [32]) @pytest.mark.parametrize("w", [64]) @pytest.mark.parametrize("split_size", [2, 4]) diff --git a/ttnn/ttnn/operations/data_movement.py b/ttnn/ttnn/operations/data_movement.py index 5f0c04bd1be..b674ae323b6 100644 --- a/ttnn/ttnn/operations/data_movement.py +++ b/ttnn/ttnn/operations/data_movement.py @@ -265,7 +265,7 @@ def _split_validate_input_tensors(operation_name, input_tensor, *args, **kwargs) name="ttnn.split", validate_input_tensors=_split_validate_input_tensors, golden_function=_golden_function, - allow_to_fallback_to_golden_function_on_failure=True, + allow_to_fallback_to_golden_function_on_failure=False, ) def split(input_tensor: ttnn.Tensor, split_size: int, dim: int) -> ttnn.Tensor: r"""