#0: fix alloc after harvesting row

tenstorrent · Jun 3, 2024 · b8f5f53 · b8f5f53
1 parent 16eae04
commit b8f5f53
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/...ore_reuse_mcast_dram_sharded_optimized/bmm_op_multi_core_reuse_dram_sharded_optimized.cpp b/...ore_reuse_mcast_dram_sharded_optimized/bmm_op_multi_core_reuse_dram_sharded_optimized.cpp
@@ -236,15 +236,28 @@ void get_dram_reader_core_coords_wormhole_b0(
             if (std::find(harvested_rows.begin(), harvested_rows.end(), y) != harvested_rows.end() ||
                 std::count(group_y.begin(), group_y.end(), y) >= 2) {
                 auto adjust_coord = [&](int start, int end, int step) {
+                    bool found_new_row = false;
                     for (int j = start; step > 0 ? j <= end : j >= end; j += step) {
                         if (std::find(harvested_rows.begin(), harvested_rows.end(), j) == harvested_rows.end() &&
                             std::count(group_y.begin(), group_y.end(), j) == 0) {
                             coord.y = j;
                             coord.x += x_step;
                             x_step--;
+                            found_new_row = true;
                             break;
                         }
                     }
+                    if (not found_new_row) {
+                        for (int j = start; step > 0 ? j <= end : j >= end; j += step) {
+                            if (std::find(harvested_rows.begin(), harvested_rows.end(), j) == harvested_rows.end()) {
+                                coord.y = j;
+                                coord.x += x_step;
+                                x_step--;
+                                found_new_row = true;
+                                break;
+                            }
+                        }
+                    }
                 };
 
                 if (y >= max_bank_id) {