Add support for P300 board type

tenstorrent · Mar 6, 2025 · 0ba1eac · 0ba1eac
1 parent 924a26e
commit 0ba1eac
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/device/api/umd/device/types/cluster_descriptor_types.h b/device/api/umd/device/types/cluster_descriptor_types.h
@@ -122,6 +122,8 @@ inline BoardType get_board_type_from_board_id(const uint64_t board_id) {
         return BoardType::P100;
     } else if (upi == 0x40 || upi == 0x41 || upi == 0x42) {
         return BoardType::P150;
+    } else if (upi == 0x44 || upi == 0x45 || upi == 0x46) {
+        return BoardType::P300;
     }
 
     throw std::runtime_error(fmt::format("No existing board type for board id {}", board_id));

diff --git a/device/tt_cluster_descriptor.cpp b/device/tt_cluster_descriptor.cpp
@@ -776,7 +776,9 @@ void tt_ClusterDescriptor::load_chips_from_connectivity_descriptor(YAML::Node &y
                 chip_board_type.second == "p150" || chip_board_type.second == "p150A" ||
                 chip_board_type.second == "p150C") {
                 board_type = BoardType::P150;
-            } else if (chip_board_type.second == "p300") {
+            } else if (
+                chip_board_type.second == "p300" || chip_board_type.second == "p300A" ||
+                chip_board_type.second == "p300C") {
                 board_type = BoardType::P300;
             } else if (chip_board_type.second == "GALAXY") {
                 board_type = BoardType::GALAXY;