From 37895867ac780ffb213792dd9c5a6679c757dc25 Mon Sep 17 00:00:00 2001 From: umadevimcw Date: Fri, 13 Sep 2024 09:40:05 +0000 Subject: [PATCH] #4702: Update test files --- .../wormhole/test_eltwise_mul_hang.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_mul_hang.py b/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_mul_hang.py index 1436803c3d8..3d9a6fd5fe3 100644 --- a/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_mul_hang.py +++ b/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_mul_hang.py @@ -40,25 +40,25 @@ def run_eltwise_mul_tests(input_shape, dtype, dlayout, in_mem_config, output_mem test_sweep_args = [ ( - [(150, 72), (150, 72)], + [(160, 96), (160, 96)], [ttnn.bfloat16, ttnn.bfloat16], - [ttnn.ROW_MAJOR_LAYOUT, ttnn.ROW_MAJOR_LAYOUT], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], (ttnn.DRAM_MEMORY_CONFIG), (ttnn.DRAM_MEMORY_CONFIG), 17799073, ), ( - [(3, 201, 228), (3, 201, 228)], + [(3, 224, 256), (3, 224, 256)], [ttnn.bfloat16, ttnn.bfloat16], - [ttnn.ROW_MAJOR_LAYOUT, ttnn.ROW_MAJOR_LAYOUT], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], (ttnn.DRAM_MEMORY_CONFIG), (ttnn.DRAM_MEMORY_CONFIG), 3121221, ), ( - [(6, 6, 230, 138), (6, 6, 230, 138)], + [(6, 6, 256, 160), (6, 6, 256, 160)], [ttnn.bfloat16, ttnn.bfloat16], - [ttnn.ROW_MAJOR_LAYOUT, ttnn.ROW_MAJOR_LAYOUT], + [ttnn.TILE_LAYOUT, ttnn.TILE_LAYOUT], (ttnn.DRAM_MEMORY_CONFIG), (ttnn.DRAM_MEMORY_CONFIG), 10286194,