diff --git a/models/demos/distilbert/tests/test_perf_distilbert.py b/models/demos/distilbert/tests/test_perf_distilbert.py index a68e1cbeb453..8cb8c99e59fe 100644 --- a/models/demos/distilbert/tests/test_perf_distilbert.py +++ b/models/demos/distilbert/tests/test_perf_distilbert.py @@ -154,7 +154,7 @@ def test_distilbert_perf_device(batch_size, test, reset_seeds): if is_grayskull(): expected_perf = 40.8772 elif is_wormhole_b0(): - expected_perf = 90.2505 + expected_perf = 103.884 command = f"pytest tests/ttnn/integration_tests/distilbert/test_ttnn_distilbert.py::test_distilbert_for_question_answering[sequence_size=768-batch_size=8-model_name=distilbert-base-uncased-distilled-squad]" cols = ["DEVICE FW", "DEVICE KERNEL", "DEVICE BRISC KERNEL"] diff --git a/models/demos/vgg/tests/test_perf_vgg.py b/models/demos/vgg/tests/test_perf_vgg.py index f687e217cba6..c83f13888814 100644 --- a/models/demos/vgg/tests/test_perf_vgg.py +++ b/models/demos/vgg/tests/test_perf_vgg.py @@ -137,10 +137,10 @@ def test_perf_device_bare_metal_vgg(batch_size, model_name): margin = 0.03 if model_name == "ttnn_vgg11": - expected_perf = 132.2436 if is_grayskull() else 272.8989 + expected_perf = 132.2436 if is_grayskull() else 283.289 command = f"pytest tests/ttnn/integration_tests/vgg/test_ttnn_vgg11.py" else: - expected_perf = 116.1459 if is_grayskull() else 194.4063 + expected_perf = 116.1459 if is_grayskull() else 201.3867 command = f"pytest tests/ttnn/integration_tests/vgg/test_ttnn_vgg16.py" cols = ["DEVICE FW", "DEVICE KERNEL", "DEVICE BRISC KERNEL"] diff --git a/tests/tt_eager/python_api_testing/unit_testing/misc/test_transpose.py b/tests/tt_eager/python_api_testing/unit_testing/misc/test_transpose.py index 9d6b9fb1e8fa..f5bd8d6ae453 100644 --- a/tests/tt_eager/python_api_testing/unit_testing/misc/test_transpose.py +++ b/tests/tt_eager/python_api_testing/unit_testing/misc/test_transpose.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: © 2023 Tenstorrent Inc. +# SPDX-FileCopyrightText: © 2024 Tenstorrent Inc. # SPDX-License-Identifier: Apache-2.0 @@ -911,6 +911,8 @@ def test_transpose_issue_11650_10350(shape, dims, layout, dtype, device): def test_transpose_unpadded(shape, dims, layout, dtype, pad_value, device): if pad_value is not None and is_blackhole(): pytest.skip("Blackhole reduce is needed for the full test to work") + elif dtype == ttnn.float32 and is_grayskull(): + pytest.skip("Grayskull does not support float32") torch_input = torch.randn(shape, dtype=torch.bfloat16) torch_output = torch_input.transpose(dims[0], dims[1]) diff --git a/tests/ttnn/unit_tests/operations/test_permute.py b/tests/ttnn/unit_tests/operations/test_permute.py index 96d9a3b028da..bcae95293a1d 100644 --- a/tests/ttnn/unit_tests/operations/test_permute.py +++ b/tests/ttnn/unit_tests/operations/test_permute.py @@ -8,7 +8,8 @@ import ttnn -from tests.ttnn.utils_for_testing import assert_with_pcc, is_blackhole +from tests.ttnn.utils_for_testing import assert_with_pcc +from models.utility_functions import is_blackhole torch.manual_seed(2005) diff --git a/ttnn/cpp/ttnn/operations/data_movement/concat/concat.cpp b/ttnn/cpp/ttnn/operations/data_movement/concat/concat.cpp index f6167c29a6e5..301b224c6eaf 100644 --- a/ttnn/cpp/ttnn/operations/data_movement/concat/concat.cpp +++ b/ttnn/cpp/ttnn/operations/data_movement/concat/concat.cpp @@ -161,7 +161,7 @@ MassagedConcat build_prepost_transpose_concat(uint8_t queue_id, const MemoryConf tensors.end(), std::back_inserter(itensors), [dim1, dim2](const ttnn::Tensor& input_tensor) -> ttnn::Tensor { - return ttnn::transpose(input_tensor, dim1, dim2, std::nullopt); + return ttnn::transpose(input_tensor, dim1, dim2); } ); auto norm_dim1 = tensors.front().get_shape().get_normalized_index(dim1); diff --git a/ttnn/cpp/ttnn/operations/data_movement/transpose/device/kernels/dataflow/reader_unary_transpose_hc_interleaved_tiled_padding_aware.cpp b/ttnn/cpp/ttnn/operations/data_movement/transpose/device/kernels/dataflow/reader_unary_transpose_hc_interleaved_tiled_padding_aware.cpp index 2b7cfbfe22a1..a2ae47671508 100644 --- a/ttnn/cpp/ttnn/operations/data_movement/transpose/device/kernels/dataflow/reader_unary_transpose_hc_interleaved_tiled_padding_aware.cpp +++ b/ttnn/cpp/ttnn/operations/data_movement/transpose/device/kernels/dataflow/reader_unary_transpose_hc_interleaved_tiled_padding_aware.cpp @@ -1,4 +1,4 @@ -// SPDX-FileCopyrightText: © 2023 Tenstorrent Inc. +// SPDX-FileCopyrightText: © 2024 Tenstorrent Inc. // // SPDX-License-Identifier: Apache-2.0