#4702: Change layout for the tests

tenstorrent · Oct 29, 2024 · 92ed31f · 92ed31f
1 parent 3789586
commit 92ed31f
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 12 deletions.
diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_add_hang.py b/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_add_hang.py
@@ -42,23 +42,23 @@ def run_eltwise_add_tests(input_shape, dtype, dlayout, in_mem_config, output_mem
     (
         [(150, 72), (150, 72)],
         [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)],
         [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)],
         [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,

diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_exp_hang.py b/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_exp_hang.py
@@ -40,23 +40,23 @@ def run_eltwise_exp_tests(input_shape, dtype, dlayout, in_mem_config, output_mem
     (
         [(150, 72)],
         [ttnn.bfloat16],
-        [ttnn.ROW_MAJOR_LAYOUT],
+        [ttnn.TILE_LAYOUT],
         (ttnn.DRAM_MEMORY_CONFIG),
         (ttnn.DRAM_MEMORY_CONFIG),
         17799073,
     ),
     (
         [(3, 201, 228)],
         [ttnn.bfloat16],
-        [ttnn.ROW_MAJOR_LAYOUT],
+        [ttnn.TILE_LAYOUT],
         (ttnn.DRAM_MEMORY_CONFIG),
         (ttnn.DRAM_MEMORY_CONFIG),
         3121221,
     ),
     (
         [(6, 6, 230, 138)],
         [ttnn.bfloat16],
-        [ttnn.ROW_MAJOR_LAYOUT],
+        [ttnn.TILE_LAYOUT],
         (ttnn.DRAM_MEMORY_CONFIG),
         (ttnn.DRAM_MEMORY_CONFIG),
         10286194,

diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_sub_hang.py b/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_sub_hang.py
@@ -42,23 +42,23 @@ def run_eltwise_sub_tests(input_shape, dtype, dlayout, in_mem_config, output_mem
     (
         [(150, 72), (150, 72)],
         [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)],
         [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)],
         [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,

diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_tanh_hang.py b/tests/ttnn/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_tanh_hang.py
@@ -40,23 +40,23 @@ def run_eltwise_tanh_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,
     ),
     (
         [(3, 201, 228)],
         [ttnn.bfloat16],
-        [ttnn.ROW_MAJOR_LAYOUT],
+        [ttnn.ttnn.TILE_LAYOUT],
         (ttnn.DRAM_MEMORY_CONFIG),
         (ttnn.DRAM_MEMORY_CONFIG),
         3121221,
     ),
     (
         [(6, 6, 230, 138)],
         [ttnn.bfloat16],
-        [ttnn.ROW_MAJOR_LAYOUT],
+        [ttnn.ttnn.TILE_LAYOUT],
         (ttnn.DRAM_MEMORY_CONFIG),
         (ttnn.DRAM_MEMORY_CONFIG),
         10286194,