From 14f43415d5686b35f71f19a2c6fabb7b439538a2 Mon Sep 17 00:00:00 2001 From: Borys Bradel Date: Fri, 13 Sep 2024 15:52:53 +0000 Subject: [PATCH] #12619: Update matmul sweep timeout and core range set usage --- .../sweeps/matmul/full/matmul_default_height_sharded.py | 2 +- .../sweeps/matmul/full/matmul_default_width_sharded.py | 2 +- tests/sweep_framework/sweeps/matmul/short/matmul_default.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/sweep_framework/sweeps/matmul/full/matmul_default_height_sharded.py b/tests/sweep_framework/sweeps/matmul/full/matmul_default_height_sharded.py index 44cd92fe128c..fff716bc2ea1 100644 --- a/tests/sweep_framework/sweeps/matmul/full/matmul_default_height_sharded.py +++ b/tests/sweep_framework/sweeps/matmul/full/matmul_default_height_sharded.py @@ -88,7 +88,7 @@ def run( # TODO: row_wise=False and ROW_MAJOR shard orientation gives bad PCC # TODO: COL_MAJOR shard orientation doesn't work for get_matmul_program_config input_a_memory_config.shard_spec = ttnn.ShardSpec( - ttnn.num_cores_to_corerange_set(num_cores_height, core_grid, row_wise=True), + ttnn.CoreRangeSet(ttnn.num_cores_to_corerange_set(num_cores_height, core_grid, row_wise=True)), (per_core_height, k_size), ttnn.ShardOrientation.ROW_MAJOR, False, diff --git a/tests/sweep_framework/sweeps/matmul/full/matmul_default_width_sharded.py b/tests/sweep_framework/sweeps/matmul/full/matmul_default_width_sharded.py index e6fb7562639f..a7f0361f4fe5 100644 --- a/tests/sweep_framework/sweeps/matmul/full/matmul_default_width_sharded.py +++ b/tests/sweep_framework/sweeps/matmul/full/matmul_default_width_sharded.py @@ -81,7 +81,7 @@ def run( # TODO: row_wise=False and ROW_MAJOR shard orientation gives bad PCC # TODO: COL_MAJOR shard orientation doesn't work for get_matmul_program_config input_a_memory_config.shard_spec = ttnn.ShardSpec( - ttnn.num_cores_to_corerange_set(num_cores_width, core_grid, row_wise=True), + ttnn.CoreRangeSet(ttnn.num_cores_to_corerange_set(num_cores_width, core_grid, row_wise=True)), (total_height, per_core_width), ttnn.ShardOrientation.ROW_MAJOR, False, diff --git a/tests/sweep_framework/sweeps/matmul/short/matmul_default.py b/tests/sweep_framework/sweeps/matmul/short/matmul_default.py index d872fcf1874d..bb12db78c588 100644 --- a/tests/sweep_framework/sweeps/matmul/short/matmul_default.py +++ b/tests/sweep_framework/sweeps/matmul/short/matmul_default.py @@ -12,7 +12,7 @@ from models.utility_functions import torch_random -TIMEOUT = 5 +TIMEOUT = 15 # TODO: Missing coverage for mixed precision; passed in dtype does nothing in current matmul path parameters = {