diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_ldexp.py b/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_ldexp.py new file mode 100644 index 000000000000..d8ad2564600c --- /dev/null +++ b/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_ldexp.py @@ -0,0 +1,73 @@ +# SPDX-FileCopyrightText: © 2023 Tenstorrent Inc. + +# SPDX-License-Identifier: Apache-2.0 + +from loguru import logger +import random +import pytest +import torch +import ttnn + +from tests.ttnn.utils_for_testing import assert_with_pcc +from tests.ttnn.python_api_testing.sweep_tests import ttnn_ops + + +def run_ldexp_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device): + torch.manual_seed(data_seed) + + x = torch.Tensor(size=input_shape[0]).uniform_(-100, 100).to(torch.bfloat16) + y = torch.Tensor(size=input_shape[1]).uniform_(-100, 100).to(torch.bfloat16) + + try: + # get ref result + ref_value = torch.ldexp(x, y) + + x = ttnn_ops.setup_ttnn_tensor(x, device, dlayout[0], in_mem_config, dtype[0]) + y = ttnn_ops.setup_ttnn_tensor(y, device, dlayout[1], in_mem_config, dtype[1]) + + tt_result = ttnn.ldexp(x, y) + tt_result = ttnn_ops.ttnn_tensor_to_torch(tt_result, output_mem_config) + + except Exception as e: + logger.warning(f"Operation execution crashed") + raise e + + assert len(tt_result.shape) == len(ref_value.shape) + assert tt_result.shape == ref_value.shape + assert_with_pcc(ref_value, tt_result, 0.99) + + +test_sweep_args = [ + ( + [(160, 128), (160, 128)], + [ttnn.bfloat8_b, ttnn.bfloat16], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), + ( + [(160, 128), (160, 128)], + [ttnn.bfloat16, ttnn.bfloat8_b], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), + ( + [(160, 128), (160, 128)], + [ttnn.bfloat8_b, ttnn.bfloat8_b], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), +] + + +@pytest.mark.parametrize( + "input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed", + (test_sweep_args), +) +def test_ldexp(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device): + run_ldexp_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device) diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_logaddexp.py b/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_logaddexp.py new file mode 100644 index 000000000000..8e41d355da4a --- /dev/null +++ b/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_logaddexp.py @@ -0,0 +1,73 @@ +# SPDX-FileCopyrightText: © 2023 Tenstorrent Inc. + +# SPDX-License-Identifier: Apache-2.0 + +from loguru import logger +import random +import pytest +import torch +import ttnn + +from tests.ttnn.utils_for_testing import assert_with_pcc +from tests.ttnn.python_api_testing.sweep_tests import ttnn_ops + + +def run_logaddexp_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device): + torch.manual_seed(data_seed) + + x = torch.Tensor(size=input_shape[0]).uniform_(-100, 100).to(torch.bfloat16) + y = torch.Tensor(size=input_shape[1]).uniform_(-100, 100).to(torch.bfloat16) + + try: + # get ref result + ref_value = torch.logaddexp(x, y) + + x = ttnn_ops.setup_ttnn_tensor(x, device, dlayout[0], in_mem_config, dtype[0]) + y = ttnn_ops.setup_ttnn_tensor(y, device, dlayout[1], in_mem_config, dtype[1]) + + tt_result = ttnn.logaddexp(x, y) + tt_result = ttnn_ops.ttnn_tensor_to_torch(tt_result, output_mem_config) + + except Exception as e: + logger.warning(f"Operation execution crashed") + raise e + + assert len(tt_result.shape) == len(ref_value.shape) + assert tt_result.shape == ref_value.shape + assert_with_pcc(ref_value, tt_result, 0.99) + + +test_sweep_args = [ + ( + [(160, 128), (160, 128)], + [ttnn.bfloat8_b, ttnn.bfloat8_b], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), + ( + [(160, 128), (160, 128)], + [ttnn.bfloat16, ttnn.bfloat8_b], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), + ( + [(160, 128), (160, 128)], + [ttnn.bfloat8_b, ttnn.bfloat16], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), +] + + +@pytest.mark.parametrize( + "input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed", + (test_sweep_args), +) +def test_logaddexp(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device): + run_logaddexp_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device) diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_logaddexp2.py b/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_logaddexp2.py new file mode 100644 index 000000000000..635097f82a69 --- /dev/null +++ b/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_logaddexp2.py @@ -0,0 +1,73 @@ +# SPDX-FileCopyrightText: © 2023 Tenstorrent Inc. + +# SPDX-License-Identifier: Apache-2.0 + +from loguru import logger +import random +import pytest +import torch +import ttnn + +from tests.ttnn.utils_for_testing import assert_with_pcc +from tests.ttnn.python_api_testing.sweep_tests import ttnn_ops + + +def run_logaddexp2_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device): + torch.manual_seed(data_seed) + + x = torch.Tensor(size=input_shape[0]).uniform_(-100, 100).to(torch.bfloat16) + y = torch.Tensor(size=input_shape[1]).uniform_(-100, 100).to(torch.bfloat16) + + try: + # get ref result + ref_value = torch.logaddexp2(x, y) + + x = ttnn_ops.setup_ttnn_tensor(x, device, dlayout[0], in_mem_config, dtype[0]) + y = ttnn_ops.setup_ttnn_tensor(y, device, dlayout[1], in_mem_config, dtype[1]) + + tt_result = ttnn.logaddexp2(x, y) + tt_result = ttnn_ops.ttnn_tensor_to_torch(tt_result, output_mem_config) + + except Exception as e: + logger.warning(f"Operation execution crashed") + raise e + + assert len(tt_result.shape) == len(ref_value.shape) + assert tt_result.shape == ref_value.shape + assert_with_pcc(ref_value, tt_result, 0.99) + + +test_sweep_args = [ + ( + [(160, 128), (160, 128)], + [ttnn.bfloat8_b, ttnn.bfloat8_b], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), + ( + [(160, 128), (160, 128)], + [ttnn.bfloat16, ttnn.bfloat8_b], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), + ( + [(160, 128), (160, 128)], + [ttnn.bfloat8_b, ttnn.bfloat16], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), +] + + +@pytest.mark.parametrize( + "input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed", + (test_sweep_args), +) +def test_logaddexp2(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device): + run_logaddexp2_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device) diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_logical_and.py b/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_logical_and.py new file mode 100644 index 000000000000..7a3cb9ab92d3 --- /dev/null +++ b/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_logical_and.py @@ -0,0 +1,73 @@ +# SPDX-FileCopyrightText: © 2023 Tenstorrent Inc. + +# SPDX-License-Identifier: Apache-2.0 + +from loguru import logger +import random +import pytest +import torch +import ttnn + +from tests.ttnn.utils_for_testing import assert_with_pcc +from tests.ttnn.python_api_testing.sweep_tests import ttnn_ops + + +def run_and_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device): + torch.manual_seed(data_seed) + + x = torch.Tensor(size=input_shape[0]).uniform_(-100, 100).to(torch.bfloat16) + y = torch.Tensor(size=input_shape[1]).uniform_(-100, 100).to(torch.bfloat16) + + try: + # get ref result + ref_value = torch.logical_and(x, y) + + x = ttnn_ops.setup_ttnn_tensor(x, device, dlayout[0], in_mem_config, dtype[0]) + y = ttnn_ops.setup_ttnn_tensor(y, device, dlayout[1], in_mem_config, dtype[1]) + + tt_result = ttnn.logical_and(x, y) + tt_result = ttnn_ops.ttnn_tensor_to_torch(tt_result, output_mem_config) + + except Exception as e: + logger.warning(f"Operation execution crashed") + raise e + + assert len(tt_result.shape) == len(ref_value.shape) + assert tt_result.shape == ref_value.shape + assert_with_pcc(ref_value, tt_result, 0.99) + + +test_sweep_args = [ + ( + [(160, 128), (160, 128)], + [ttnn.bfloat8_b, ttnn.bfloat16], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), + ( + [(160, 128), (160, 128)], + [ttnn.bfloat16, ttnn.bfloat8_b], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), + ( + [(160, 128), (160, 128)], + [ttnn.bfloat8_b, ttnn.bfloat8_b], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), +] + + +@pytest.mark.parametrize( + "input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed", + (test_sweep_args), +) +def test_and(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device): + run_and_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device) diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_logical_or.py b/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_logical_or.py new file mode 100644 index 000000000000..dcc65686c0f6 --- /dev/null +++ b/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_logical_or.py @@ -0,0 +1,73 @@ +# SPDX-FileCopyrightText: © 2023 Tenstorrent Inc. + +# SPDX-License-Identifier: Apache-2.0 + +from loguru import logger +import random +import pytest +import torch +import ttnn + +from tests.ttnn.utils_for_testing import assert_with_pcc +from tests.ttnn.python_api_testing.sweep_tests import ttnn_ops + + +def run_or_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device): + torch.manual_seed(data_seed) + + x = torch.Tensor(size=input_shape[0]).uniform_(-100, 100).to(torch.bfloat16) + y = torch.Tensor(size=input_shape[1]).uniform_(-100, 100).to(torch.bfloat16) + + try: + # get ref result + ref_value = torch.logical_or(x, y) + + x = ttnn_ops.setup_ttnn_tensor(x, device, dlayout[0], in_mem_config, dtype[0]) + y = ttnn_ops.setup_ttnn_tensor(y, device, dlayout[1], in_mem_config, dtype[1]) + + tt_result = ttnn.logical_or(x, y) + tt_result = ttnn_ops.ttnn_tensor_to_torch(tt_result, output_mem_config) + + except Exception as e: + logger.warning(f"Operation execution crashed") + raise e + + assert len(tt_result.shape) == len(ref_value.shape) + assert tt_result.shape == ref_value.shape + assert_with_pcc(ref_value, tt_result, 0.99) + + +test_sweep_args = [ + ( + [(160, 128), (160, 128)], + [ttnn.bfloat8_b, ttnn.bfloat16], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), + ( + [(160, 128), (160, 128)], + [ttnn.bfloat16, ttnn.bfloat8_b], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), + ( + [(160, 128), (160, 128)], + [ttnn.bfloat8_b, ttnn.bfloat8_b], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), +] + + +@pytest.mark.parametrize( + "input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed", + (test_sweep_args), +) +def test_or(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device): + run_or_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device) diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_logical_xor.py b/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_logical_xor.py new file mode 100644 index 000000000000..e9cc909698da --- /dev/null +++ b/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_logical_xor.py @@ -0,0 +1,73 @@ +# SPDX-FileCopyrightText: © 2023 Tenstorrent Inc. + +# SPDX-License-Identifier: Apache-2.0 + +from loguru import logger +import random +import pytest +import torch +import ttnn + +from tests.ttnn.utils_for_testing import assert_with_pcc +from tests.ttnn.python_api_testing.sweep_tests import ttnn_ops + + +def run_xor_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device): + torch.manual_seed(data_seed) + + x = torch.Tensor(size=input_shape[0]).uniform_(-100, 100).to(torch.bfloat16) + y = torch.Tensor(size=input_shape[1]).uniform_(-100, 100).to(torch.bfloat16) + + try: + # get ref result + ref_value = torch.logical_xor(x, y) + + x = ttnn_ops.setup_ttnn_tensor(x, device, dlayout[0], in_mem_config, dtype[0]) + y = ttnn_ops.setup_ttnn_tensor(y, device, dlayout[1], in_mem_config, dtype[1]) + + tt_result = ttnn.logical_xor(x, y) + tt_result = ttnn_ops.ttnn_tensor_to_torch(tt_result, output_mem_config) + + except Exception as e: + logger.warning(f"Operation execution crashed") + raise e + + assert len(tt_result.shape) == len(ref_value.shape) + assert tt_result.shape == ref_value.shape + assert_with_pcc(ref_value, tt_result, 0.99) + + +test_sweep_args = [ + ( + [(160, 128), (160, 128)], + [ttnn.bfloat8_b, ttnn.bfloat16], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), + ( + [(160, 128), (160, 128)], + [ttnn.bfloat16, ttnn.bfloat8_b], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), + ( + [(160, 128), (160, 128)], + [ttnn.bfloat8_b, ttnn.bfloat8_b], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], + (ttnn.DRAM_MEMORY_CONFIG), + (ttnn.DRAM_MEMORY_CONFIG), + 4171614, + ), +] + + +@pytest.mark.parametrize( + "input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed", + (test_sweep_args), +) +def test_xor(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device): + run_xor_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device) diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_ldexp_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_ldexp_test.yaml new file mode 100644 index 000000000000..ed9dcbd1df07 --- /dev/null +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_ldexp_test.yaml @@ -0,0 +1,28 @@ +--- +test-list: + - ttnn-logical_xor: + shape: + start-shape: [1, 1, 32, 32] + end-shape: [6, 12, 256, 256] + interval: [1, 1, 32, 32] + num-dims: [2, 3, 4] + num-shapes: 2 + num-samples: 64 + args-sampling-strategy: "all" + env: + # TT_PCI_DMA_BUF_SIZE: "1048576" + datagen: + function: gen_rand + dtype: bfloat16 + args: + low: -100 + high: 100 + comparison: + function: comp_pcc + args-gen: gen_dtype_layout_device + args: + data-layout: ["TILE"] + data-type: ["BFLOAT16"] + buffer-type: ["DRAM", "L1"] + out-buffer-type: ["DRAM", "L1"] + output-file: eltwise_eltwise_ldexp_sweep.csv diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logaddexp2_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logaddexp2_test.yaml similarity index 88% rename from tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logaddexp2_test.yaml rename to tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logaddexp2_test.yaml index fa49fbd1b173..8a45e9d92849 100644 --- a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logaddexp2_test.yaml +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logaddexp2_test.yaml @@ -15,14 +15,14 @@ test-list: function: gen_rand dtype: bfloat16 args: - low: -100 - high: 100 + low: -64 + high: 64 comparison: function: comp_pcc args-gen: gen_dtype_layout_device args: data-layout: ["TILE"] - data-type: ["BFLOAT16", "BFLOAT8_B"] + data-type: ["BFLOAT16"] buffer-type: ["DRAM", "L1"] out-buffer-type: ["DRAM", "L1"] output-file: eltwise_logaddexp2_sweep.csv diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logaddexp_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logaddexp_test.yaml similarity index 88% rename from tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logaddexp_test.yaml rename to tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logaddexp_test.yaml index 668ad84153e5..326b32879e57 100644 --- a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logaddexp_test.yaml +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logaddexp_test.yaml @@ -15,14 +15,14 @@ test-list: function: gen_rand dtype: bfloat16 args: - low: -100 - high: 100 + low: -64 + high: 64 comparison: function: comp_pcc args-gen: gen_dtype_layout_device args: data-layout: ["TILE"] - data-type: ["BFLOAT16", "BFLOAT8_B"] + data-type: ["BFLOAT16"] buffer-type: ["DRAM", "L1"] out-buffer-type: ["DRAM", "L1"] output-file: eltwise_logaddexp_sweep.csv diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logical_and_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logical_and_test.yaml similarity index 93% rename from tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logical_and_test.yaml rename to tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logical_and_test.yaml index 03df980d0d64..cfa8ccff59ad 100644 --- a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logical_and_test.yaml +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logical_and_test.yaml @@ -22,7 +22,7 @@ test-list: args-gen: gen_dtype_layout_device args: data-layout: ["TILE"] - data-type: ["BFLOAT16", "BFLOAT8_B"] + data-type: ["BFLOAT16"] buffer-type: ["DRAM", "L1"] out-buffer-type: ["DRAM", "L1"] output-file: eltwise_logical_and_sweep.csv diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logical_or_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logical_or_test.yaml similarity index 93% rename from tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logical_or_test.yaml rename to tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logical_or_test.yaml index c74a11b45342..871679277875 100644 --- a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logical_or_test.yaml +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logical_or_test.yaml @@ -22,7 +22,7 @@ test-list: args-gen: gen_dtype_layout_device args: data-layout: ["TILE"] - data-type: ["BFLOAT16", "BFLOAT8_B"] + data-type: ["BFLOAT16"] buffer-type: ["DRAM", "L1"] out-buffer-type: ["DRAM", "L1"] output-file: eltwise_logical_or_sweep.csv diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logical_xor_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logical_xor_test.yaml similarity index 93% rename from tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logical_xor_test.yaml rename to tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logical_xor_test.yaml index 2288d1c58042..a1bb58b59771 100644 --- a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_logical_xor_test.yaml +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logical_xor_test.yaml @@ -22,7 +22,7 @@ test-list: args-gen: gen_dtype_layout_device args: data-layout: ["TILE"] - data-type: ["BFLOAT16", "BFLOAT8_B"] + data-type: ["BFLOAT16"] buffer-type: ["DRAM", "L1"] out-buffer-type: ["DRAM", "L1"] output-file: eltwise_logical_xor_sweep.csv diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_pow_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_pow_test.yaml similarity index 88% rename from tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_pow_test.yaml rename to tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_pow_test.yaml index 213884bdc1a8..3c910f775d13 100644 --- a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_broken/exp/ttnn_eltwise_pow_test.yaml +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_pow_test.yaml @@ -19,10 +19,10 @@ test-list: high: 100 comparison: function: comp_pcc - args-gen: gen_dtype_layout_device + args-gen: gen_power_fp_args args: data-layout: ["TILE"] - data-type: ["BFLOAT16", "BFLOAT8_B"] + data-type: ["BFLOAT16"] buffer-type: ["DRAM", "L1"] out-buffer-type: ["DRAM", "L1"] output-file: eltwise_pow_sweep.csv diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_ldexp_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_ldexp_test.yaml new file mode 100644 index 000000000000..ed9dcbd1df07 --- /dev/null +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_ldexp_test.yaml @@ -0,0 +1,28 @@ +--- +test-list: + - ttnn-logical_xor: + shape: + start-shape: [1, 1, 32, 32] + end-shape: [6, 12, 256, 256] + interval: [1, 1, 32, 32] + num-dims: [2, 3, 4] + num-shapes: 2 + num-samples: 64 + args-sampling-strategy: "all" + env: + # TT_PCI_DMA_BUF_SIZE: "1048576" + datagen: + function: gen_rand + dtype: bfloat16 + args: + low: -100 + high: 100 + comparison: + function: comp_pcc + args-gen: gen_dtype_layout_device + args: + data-layout: ["TILE"] + data-type: ["BFLOAT16"] + buffer-type: ["DRAM", "L1"] + out-buffer-type: ["DRAM", "L1"] + output-file: eltwise_eltwise_ldexp_sweep.csv diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logaddexp2_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logaddexp2_test.yaml new file mode 100644 index 000000000000..8a45e9d92849 --- /dev/null +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logaddexp2_test.yaml @@ -0,0 +1,28 @@ +--- +test-list: + - ttnn-logaddexp2: + shape: + start-shape: [1, 1, 32, 32] + end-shape: [6, 12, 256, 256] + interval: [1, 1, 32, 32] + num-dims: [2, 3, 4] + num-shapes: 2 + num-samples: 64 + args-sampling-strategy: "all" + env: + # TT_PCI_DMA_BUF_SIZE: "1048576" + datagen: + function: gen_rand + dtype: bfloat16 + args: + low: -64 + high: 64 + comparison: + function: comp_pcc + args-gen: gen_dtype_layout_device + args: + data-layout: ["TILE"] + data-type: ["BFLOAT16"] + buffer-type: ["DRAM", "L1"] + out-buffer-type: ["DRAM", "L1"] + output-file: eltwise_logaddexp2_sweep.csv diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logaddexp_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logaddexp_test.yaml new file mode 100644 index 000000000000..326b32879e57 --- /dev/null +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logaddexp_test.yaml @@ -0,0 +1,28 @@ +--- +test-list: + - ttnn-logaddexp: + shape: + start-shape: [1, 1, 32, 32] + end-shape: [6, 12, 256, 256] + interval: [1, 1, 32, 32] + num-dims: [2, 3, 4] + num-shapes: 2 + num-samples: 64 + args-sampling-strategy: "all" + env: + # TT_PCI_DMA_BUF_SIZE: "1048576" + datagen: + function: gen_rand + dtype: bfloat16 + args: + low: -64 + high: 64 + comparison: + function: comp_pcc + args-gen: gen_dtype_layout_device + args: + data-layout: ["TILE"] + data-type: ["BFLOAT16"] + buffer-type: ["DRAM", "L1"] + out-buffer-type: ["DRAM", "L1"] + output-file: eltwise_logaddexp_sweep.csv diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logical_and_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logical_and_test.yaml new file mode 100644 index 000000000000..cfa8ccff59ad --- /dev/null +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logical_and_test.yaml @@ -0,0 +1,28 @@ +--- +test-list: + - ttnn-logical_and: + shape: + start-shape: [1, 1, 32, 32] + end-shape: [6, 12, 256, 256] + interval: [1, 1, 32, 32] + num-dims: [2, 3, 4] + num-shapes: 2 + num-samples: 64 + args-sampling-strategy: "all" + env: + # TT_PCI_DMA_BUF_SIZE: "1048576" + datagen: + function: gen_rand + dtype: bfloat16 + args: + low: -100 + high: 100 + comparison: + function: comp_pcc + args-gen: gen_dtype_layout_device + args: + data-layout: ["TILE"] + data-type: ["BFLOAT16"] + buffer-type: ["DRAM", "L1"] + out-buffer-type: ["DRAM", "L1"] + output-file: eltwise_logical_and_sweep.csv diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logical_or_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logical_or_test.yaml new file mode 100644 index 000000000000..871679277875 --- /dev/null +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logical_or_test.yaml @@ -0,0 +1,28 @@ +--- +test-list: + - ttnn-logical_or: + shape: + start-shape: [1, 1, 32, 32] + end-shape: [6, 12, 256, 256] + interval: [1, 1, 32, 32] + num-dims: [2, 3, 4] + num-shapes: 2 + num-samples: 64 + args-sampling-strategy: "all" + env: + # TT_PCI_DMA_BUF_SIZE: "1048576" + datagen: + function: gen_rand + dtype: bfloat16 + args: + low: -100 + high: 100 + comparison: + function: comp_pcc + args-gen: gen_dtype_layout_device + args: + data-layout: ["TILE"] + data-type: ["BFLOAT16"] + buffer-type: ["DRAM", "L1"] + out-buffer-type: ["DRAM", "L1"] + output-file: eltwise_logical_or_sweep.csv diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logical_xor_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logical_xor_test.yaml new file mode 100644 index 000000000000..a1bb58b59771 --- /dev/null +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logical_xor_test.yaml @@ -0,0 +1,28 @@ +--- +test-list: + - ttnn-logical_xor: + shape: + start-shape: [1, 1, 32, 32] + end-shape: [6, 12, 256, 256] + interval: [1, 1, 32, 32] + num-dims: [2, 3, 4] + num-shapes: 2 + num-samples: 64 + args-sampling-strategy: "all" + env: + # TT_PCI_DMA_BUF_SIZE: "1048576" + datagen: + function: gen_rand + dtype: bfloat16 + args: + low: -100 + high: 100 + comparison: + function: comp_pcc + args-gen: gen_dtype_layout_device + args: + data-layout: ["TILE"] + data-type: ["BFLOAT16"] + buffer-type: ["DRAM", "L1"] + out-buffer-type: ["DRAM", "L1"] + output-file: eltwise_logical_xor_sweep.csv diff --git a/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_pow_test.yaml b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_pow_test.yaml new file mode 100644 index 000000000000..3c910f775d13 --- /dev/null +++ b/tests/ttnn/python_api_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_pow_test.yaml @@ -0,0 +1,28 @@ +--- +test-list: + - ttnn-pow: + shape: + start-shape: [1, 1, 32, 32] + end-shape: [6, 12, 256, 256] + interval: [1, 1, 32, 32] + num-dims: [2, 3, 4] + num-shapes: 2 + num-samples: 64 + args-sampling-strategy: "all" + env: + # TT_PCI_DMA_BUF_SIZE: "1048576" + datagen: + function: gen_rand + dtype: bfloat16 + args: + low: -100 + high: 100 + comparison: + function: comp_pcc + args-gen: gen_power_fp_args + args: + data-layout: ["TILE"] + data-type: ["BFLOAT16"] + buffer-type: ["DRAM", "L1"] + out-buffer-type: ["DRAM", "L1"] + output-file: eltwise_pow_sweep.csv