From c79568c5d427812455c46a39388ef0fa807543d0 Mon Sep 17 00:00:00 2001 From: Borys Bradel Date: Wed, 11 Sep 2024 16:41:40 +0000 Subject: [PATCH] #12448: Update 1d matmul sweep test to use CoreRangeSet for core range parameters --- .../matmul/short/matmul_user_program_config_mcast_1d.py | 8 ++++---- 1 file changed, 4 insertions(+), 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 index 75deaf15493..55928553fb8 100644 --- 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,