Address comments

tenstorrent · Oct 25, 2024 · d8104e5 · d8104e5
1 parent 31414cf
commit d8104e5
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/device/coordinate_manager.cpp b/device/coordinate_manager.cpp
@@ -172,13 +172,14 @@ CoordinateManager* CoordinateManager::get_coordinate_manager(
     const std::vector<tt_xy_pair>& workers,
     std::size_t harvesting_mask) {
 
-    if (arch == tt::ARCH::GRAYSKULL) {
-        return new GrayskullCoordinateManager(worker_grid_size, workers, harvesting_mask);
-    } else if (arch == tt::ARCH::WORMHOLE_B0) {
-        return new WormholeCoordinateManager(worker_grid_size, workers, harvesting_mask);
-    } else if (arch == tt::ARCH::BLACKHOLE) {
-        return new BlackholeCoordinateManager(worker_grid_size, workers, harvesting_mask);
-    } else {
-        throw std::runtime_error("Invalid architecture for coordinate manager");
+    switch (arch) {
+        case tt::ARCH::GRAYSKULL:
+            return new GrayskullCoordinateManager(worker_grid_size, workers, harvesting_mask);
+        case tt::ARCH::WORMHOLE_B0:
+            return new WormholeCoordinateManager(worker_grid_size, workers, harvesting_mask);
+        case tt::ARCH::BLACKHOLE:
+            return new BlackholeCoordinateManager(worker_grid_size, workers, harvesting_mask);
     }
+
+    throw std::runtime_error("Invalid architecture for creating coordinate manager");
 }