#8361: add reshard sweep test

tenstorrent · May 10, 2024 · a40cbeb · a40cbeb
1 parent ee6f9d6
commit a40cbeb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/ttnn/sweep_tests/sweeps/reshard_height_width.py b/tests/ttnn/sweep_tests/sweeps/reshard_height_width.py
@@ -14,8 +14,10 @@
 
 parameters = {
     "dtype": [ttnn.int32, ttnn.bfloat16, ttnn.bfloat8_b],
-    "height": [4, 8, 12, 16, 32, 64, 96, 128, 256, 512, 1024, 4096, 8192, 8196],
-    "width": [4, 8, 12, 16, 32, 64, 96, 128, 256, 512, 1024, 4096, 8192, 8196],
+    #    "height": [4, 8, 12, 16, 32, 64, 96, 128, 256, 512, 1024, 4096, 8192, 8196],
+    #    "width": [4, 8, 12, 16, 32, 64, 96, 128, 256, 512, 1024, 4096, 8192, 8196],
+    "height": [4, 8, 12, 16, 32, 64, 96, 128],
+    "width": [4, 8, 12, 16, 32, 64, 96, 128],
     "layout": [ttnn.TILE_LAYOUT, ttnn.ROW_MAJOR_LAYOUT],
     "input_shard_orientation": [ttnn.ShardOrientation.ROW_MAJOR, ttnn.ShardOrientation.COL_MAJOR],
     "input_num_cores_x": [1, 2, 4, 8],