Fix tests

tenstorrent · Dec 19, 2024 · c3f0293 · c3f0293
1 parent 537d55f
commit c3f0293
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/device/blackhole/blackhole_coordinate_manager.cpp b/device/blackhole/blackhole_coordinate_manager.cpp
@@ -38,6 +38,9 @@ BlackholeCoordinateManager::BlackholeCoordinateManager(
 }
 
 void BlackholeCoordinateManager::translate_tensix_coords() {
+    if (CoordinateManager::get_num_harvested(tensix_harvesting_mask) > tensix_grid_size.x) {
+        tensix_harvesting_mask = 0;
+    }
     size_t num_harvested_x = CoordinateManager::get_num_harvested(tensix_harvesting_mask);
     size_t grid_size_x = tensix_grid_size.x;
     size_t grid_size_y = tensix_grid_size.y;

diff --git a/device/coordinate_manager.cpp b/device/coordinate_manager.cpp
@@ -86,6 +86,9 @@ CoreCoord CoordinateManager::translate_coord_to(const CoreCoord core_coord, cons
 }
 
 void CoordinateManager::translate_tensix_coords() {
+    if (CoordinateManager::get_num_harvested(tensix_harvesting_mask) > tensix_grid_size.y) {
+        tensix_harvesting_mask = 0;
+    }
     size_t num_harvested_y = CoordinateManager::get_num_harvested(tensix_harvesting_mask);
     size_t grid_size_x = tensix_grid_size.x;
     size_t grid_size_y = tensix_grid_size.y;