From 817ace65cd9e7a32be6b9470a47e0f0aa47c3854 Mon Sep 17 00:00:00 2001
From: pjanevski <pjanevski@tenstorrent.com>
Date: Fri, 8 Nov 2024 16:39:13 +0000
Subject: [PATCH] Add board type enums

---
 device/tt_cluster_descriptor.cpp | 9 ++++++---
 device/tt_cluster_descriptor.h   | 6 ++++--
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/device/tt_cluster_descriptor.cpp b/device/tt_cluster_descriptor.cpp
index f94d2dd2..920d74ce 100644
--- 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
index 7c701736..d364b0d0 100644
--- 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 {