#12619: Update matmul sweep timeout and core range set usage

tenstorrent · Sep 13, 2024 · 14f4341 · 14f4341
1 parent bbfec73
commit 14f4341
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 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
@@ -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
@@ -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
@@ -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 = {