#7006: Fix reduce sum2 fail issue

tenstorrent · May 6, 2024 · f9dd970 · f9dd970
1 parent f85ffe7
commit f9dd970
Showing 1 changed file with 38 additions and 6 deletions.
diff --git a/tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_sum_2.py b/tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_sum_2.py
@@ -48,23 +48,31 @@ def run_sum_2_tests(input_shape, dtype, dlayout, in_mem_config, out_mem_config,
         (6, 2, 216, 186),
         [ttl.tensor.DataType.BFLOAT16],
         [ttl.tensor.Layout.ROW_MAJOR],
-        [None],
+        [ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM)],
         ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM),
         13482735,
     ),
     (
         (6, 1, 140, 110),
         [ttl.tensor.DataType.BFLOAT16],
         [ttl.tensor.Layout.ROW_MAJOR],
-        [None],
+        [ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM)],
+        ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM),
+        2348954,
+    ),
+    (
+        (6, 1, 140, 192),
+        [ttl.tensor.DataType.BFLOAT16],
+        [ttl.tensor.Layout.ROW_MAJOR],
+        [ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM)],
         ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM),
         2348954,
     ),
     (
         (6, 1, 160, 256),
-        [ttl.tensor.DataType.BFLOAT8_B],
-        [ttl.tensor.Layout.TILE],
-        [None],
+        [ttl.tensor.DataType.BFLOAT16],
+        [ttl.tensor.Layout.ROW_MAJOR],
+        [ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM)],
         ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM),
         2348954,
     ),
@@ -78,7 +86,15 @@ def run_sum_2_tests(input_shape, dtype, dlayout, in_mem_config, out_mem_config,
     ),
     (
         (10, 21, 480, 128),
-        [ttl.tensor.DataType.BFLOAT8_B],
+        [ttl.tensor.DataType.BFLOAT16],
+        [ttl.tensor.Layout.ROW_MAJOR],
+        [ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM)],
+        ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM),
+        13482735,
+    ),
+    (
+        (8, 4, 160, 288),
+        [ttl.tensor.DataType.BFLOAT16],
         [ttl.tensor.Layout.TILE],
         [ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM)],
         ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM),
@@ -92,6 +108,22 @@ def run_sum_2_tests(input_shape, dtype, dlayout, in_mem_config, out_mem_config,
         ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM),
         13482735,
     ),
+    (
+        (7, 14, 32, 160),
+        [ttl.tensor.DataType.BFLOAT16],
+        [ttl.tensor.Layout.ROW_MAJOR],
+        [ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM)],
+        ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM),
+        13482735,
+    ),
+    (
+        (7, 14, 32, 160),
+        [ttl.tensor.DataType.BFLOAT16],
+        [ttl.tensor.Layout.TILE],
+        [ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM)],
+        ttl.tensor.MemoryConfig(ttl.tensor.TensorMemoryLayout.INTERLEAVED, ttl.tensor.BufferType.DRAM),
+        13482735,
+    ),
     (
         (7, 14, 32, 160),
         [ttl.tensor.DataType.BFLOAT8_B],