From a40cbeb54d1d5796699ed87eaf4ad4b380713e62 Mon Sep 17 00:00:00 2001 From: tarafdarTT Date: Fri, 10 May 2024 18:30:31 +0000 Subject: [PATCH] #8361: add reshard sweep test --- tests/ttnn/sweep_tests/sweeps/reshard_height_width.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/ttnn/sweep_tests/sweeps/reshard_height_width.py b/tests/ttnn/sweep_tests/sweeps/reshard_height_width.py index 409612ae6ccb..e62eef8953ef 100644 --- 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],