Add board type enums

tenstorrent · Nov 8, 2024 · 817ace6 · 817ace6
1 parent 0cad2b9
commit 817ace6
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/device/tt_cluster_descriptor.cpp b/device/tt_cluster_descriptor.cpp
@@ -573,10 +573,13 @@ void tt_ClusterDescriptor::load_chips_from_connectivity_descriptor(YAML::Node &y
                 board_type = BoardType::N300;
             } else if (chip_board_type.second == "GALAXY") {
                 board_type = BoardType::GALAXY;
+            } else if (chip_board_type.second == "e150") {
+                board_type = BoardType::E150;
+            }
+            else if (chip_board_type.second == "p150A") {
+                board_type = BoardType::P150A;
             } else {
-                if (chip_board_type.second != "e150" && chip_board_type.second != "p150A") {
-                    log_warning(LogSiliconDriver, "Unknown board type for chip {}. This might happen because chip is running old firmware. Defaulting to DEFAULT", chip);
-                }
+                log_warning(LogSiliconDriver, "Unknown board type for chip {}. This might happen because chip is running old firmware. Defaulting to DEFAULT", chip);
                 board_type = BoardType::DEFAULT;
             }
             desc.chip_board_type.insert({chip, board_type});

diff --git a/device/tt_cluster_descriptor.h b/device/tt_cluster_descriptor.h
@@ -25,8 +25,10 @@ namespace YAML { class Node; }
 enum BoardType : uint32_t {
     N150 = 0,
     N300 = 1,
-    GALAXY = 2,
-    DEFAULT = 3,
+    E150 = 2,
+    P150A = 3,
+    GALAXY = 4,
+    DEFAULT = 5,
 };
 
 class tt_ClusterDescriptor {