diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_gelu_hang.py b/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_gelu_hang.py index b7a0a09b3e1..87c123b1a12 100644 --- a/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_gelu_hang.py +++ b/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_gelu_hang.py @@ -40,7 +40,7 @@ def run_eltwise_gelu_tests(input_shape, dtype, dlayout, in_mem_config, output_me ( [(150, 72)], [ttnn.bfloat16], - [ttnn.ROW_MAJOR_LAYOUT], + [ttnn.TILE_LAYOUT], (ttnn.DRAM_MEMORY_CONFIG), (ttnn.DRAM_MEMORY_CONFIG), 17799073, @@ -48,7 +48,7 @@ def run_eltwise_gelu_tests(input_shape, dtype, dlayout, in_mem_config, output_me ( [(3, 201, 228)], [ttnn.bfloat16], - [ttnn.ROW_MAJOR_LAYOUT], + [ttnn.TILE_LAYOUT], (ttnn.DRAM_MEMORY_CONFIG), (ttnn.DRAM_MEMORY_CONFIG), 3121221, @@ -56,7 +56,7 @@ def run_eltwise_gelu_tests(input_shape, dtype, dlayout, in_mem_config, output_me ( [(6, 6, 230, 138)], [ttnn.bfloat16], - [ttnn.ROW_MAJOR_LAYOUT], + [ttnn.TILE_LAYOUT], (ttnn.DRAM_MEMORY_CONFIG), (ttnn.DRAM_MEMORY_CONFIG), 10286194,