#12448: Update 1d matmul sweep test to use CoreRangeSet for core rang…

…e parameters
tenstorrent · Sep 11, 2024 · c79568c · c79568c
1 parent 226a14b
commit c79568c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/sweep_framework/sweeps/matmul/short/matmul_user_program_config_mcast_1d.py b/tests/sweep_framework/sweeps/matmul/short/matmul_user_program_config_mcast_1d.py
@@ -77,7 +77,7 @@ class TensorMemoryConfigs(enum.Enum):
                 memory_layout=ttnn.TensorMemoryLayout.WIDTH_SHARDED,
                 buffer_type=ttnn.BufferType.L1,
                 shard_spec=ttnn.ShardSpec(
-                    ttnn.num_cores_to_corerange_set(28, core_grid, row_wise=True),
+                    ttnn.CoreRangeSet(ttnn.num_cores_to_corerange_set(28, core_grid, row_wise=True)),
                     (64, IN0_INNER_DIM_PER_CORE),
                     ttnn.ShardOrientation.ROW_MAJOR,
                     False,
@@ -105,7 +105,7 @@ class TensorMemoryConfigs(enum.Enum):
                 memory_layout=ttnn.TensorMemoryLayout.WIDTH_SHARDED,
                 buffer_type=ttnn.BufferType.L1,
                 shard_spec=ttnn.ShardSpec(
-                    ttnn.num_cores_to_corerange_set(35, core_grid, row_wise=True),
+                    ttnn.CoreRangeSet(ttnn.num_cores_to_corerange_set(35, core_grid, row_wise=True)),
                     (64, IN0_INNER_DIM_PER_CORE),
                     ttnn.ShardOrientation.ROW_MAJOR,
                     False,
@@ -134,7 +134,7 @@ class TensorMemoryConfigs(enum.Enum):
                 memory_layout=ttnn.TensorMemoryLayout.WIDTH_SHARDED,
                 buffer_type=ttnn.BufferType.L1,
                 shard_spec=ttnn.ShardSpec(
-                    ttnn.num_cores_to_corerange_set(28, core_grid, row_wise=True),
+                    ttnn.CoreRangeSet(ttnn.num_cores_to_corerange_set(28, core_grid, row_wise=True)),
                     (64, IN0_INNER_DIM_PER_CORE),
                     ttnn.ShardOrientation.ROW_MAJOR,
                     False,
@@ -163,7 +163,7 @@ class TensorMemoryConfigs(enum.Enum):
                 memory_layout=ttnn.TensorMemoryLayout.WIDTH_SHARDED,
                 buffer_type=ttnn.BufferType.L1,
                 shard_spec=ttnn.ShardSpec(
-                    ttnn.num_cores_to_corerange_set(30, core_grid, row_wise=True),
+                    ttnn.CoreRangeSet(ttnn.num_cores_to_corerange_set(30, core_grid, row_wise=True)),
                     (64, IN0_INNER_DIM_PER_CORE),
                     ttnn.ShardOrientation.ROW_MAJOR,
                     False,