Hack to enable n300 until #9 is resolved

tenstorrent · Sep 23, 2024 · cf369db · cf369db
1 parent 896c37e
commit cf369db
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/build-and-test.yml b/.github/workflows/build-and-test.yml
@@ -93,7 +93,7 @@ jobs:
       matrix:
         build: [
           {runs-on: n150, name: "run"},
-          # {runs-on: n300, name: "run"}, #TODO: https://github.com/tenstorrent/tt-xla/issues/9
+          {runs-on: n300, name: "run"},
         ]
 
     runs-on:

diff --git a/src/common/api_impl.cc b/src/common/api_impl.cc
@@ -646,7 +646,7 @@ void ClientInstance::BindApi(PJRT_Api* api) {
 tt_pjrt_status ClientInstance::PopulateDevices() {
   DLOG_F(LOG_DEBUG, "ClientInstance::PopulateDevices");
   auto [system_desc, chip_ids] = tt::runtime::getCurrentSystemDesc();
-  int device_info_count_ = chip_ids.size();
+  int device_info_count_ = 1; // TODO: revert to chip_ids.size(); once https://github.com/tenstorrent/tt-xla/issues/9 is fixed
 
   devices_.resize(device_info_count_);
   for (size_t i = 0; i < device_info_count_; ++i) {