From 1a34a01906c1317f859c326dcd3d191a79f252e2 Mon Sep 17 00:00:00 2001 From: ngrujic Date: Thu, 2 Nov 2023 13:13:33 +0000 Subject: [PATCH] #3561: Modify recip unit test for wormhole --- .../grayskull/test_eltwise_recip.py | 3 -- .../pytorch_eltwise_recip_test.yaml | 52 +++++++++++++++++++ 2 files changed, 52 insertions(+), 3 deletions(-) create mode 100644 tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests/broken_wormhole/pytorch_eltwise_recip_test.yaml diff --git a/tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_recip.py b/tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_recip.py index 52c9d44c39a9..290ae351dfaf 100644 --- a/tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_recip.py +++ b/tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_recip.py @@ -5,12 +5,10 @@ from loguru import logger import pytest import torch - 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.common import is_wormhole_b0, skip_for_wormhole_b0 from tests.tt_eager.python_api_testing.sweep_tests.tt_lib_ops import eltwise_recip as tt_eltwise_recip @@ -78,7 +76,6 @@ def run_eltwise_recip_tests(input_shape, dtype, dlayout, in_mem_config, out_mem_ ] -@skip_for_wormhole_b0 @pytest.mark.parametrize( "input_shape, dtype, dlayout, in_mem_config, out_mem_config, data_seed", (test_sweep_args), diff --git a/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests/broken_wormhole/pytorch_eltwise_recip_test.yaml b/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests/broken_wormhole/pytorch_eltwise_recip_test.yaml new file mode 100644 index 000000000000..e408df605624 --- /dev/null +++ b/tests/tt_eager/python_api_testing/sweep_tests/test_configs/ci_sweep_tests/broken_wormhole/pytorch_eltwise_recip_test.yaml @@ -0,0 +1,52 @@ +--- +test-list: + - eltwise-recip: + shape: + start-shape: [1, 1, 32, 32] + end-shape: [6, 12, 256, 256] + interval: [1, 1, 32, 32] + num-shapes: 1 + num-samples: 64 + args-sampling-strategy: "all" + env: + TT_METAL_SLOW_DISPATCH_MODE: ["", "1"] + # TT_PCI_DMA_BUF_SIZE: "1048576" + datagen: + function: gen_rand + 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", "SYSTEM_MEMORY"] + out-buffer-type: ["DRAM"] + output-file: eltwise_recip_sweep.csv + - eltwise-recip: + shape: + start-shape: [1, 1, 2, 2] + end-shape: [6, 12, 256, 256] + interval: [1, 1, 1, 2] + num-shapes: 1 + num-samples: 64 + args-sampling-strategy: "all" + env: + TT_METAL_SLOW_DISPATCH_MODE: ["", "1"] + # TT_PCI_DMA_BUF_SIZE: "1048576" + datagen: + function: gen_rand + args: + low: -100 + high: 100 + comparison: + function: comp_pcc + args-gen: gen_dtype_layout_device + args: + data-layout: ["ROW_MAJOR"] + data-type: ["BFLOAT16"] + buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"] + out-buffer-type: ["DRAM"] + output-file: eltwise_recip_sweep.csv