From f4178283988fb05ba1fa5f8f86fdc01cbdabd84d Mon Sep 17 00:00:00 2001
From: Sanja Djukic <sdjukic@tenstorrent.com>
Date: Wed, 19 Feb 2025 15:34:04 +0000
Subject: [PATCH] storing system desc

---
 src/common/module_builder.cc                      | 1 +
 src/common/pjrt_implementation/client_instance.cc | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/common/module_builder.cc b/src/common/module_builder.cc
index 55cb112f..a800d5e9 100644
--- a/src/common/module_builder.cc
+++ b/src/common/module_builder.cc
@@ -197,6 +197,7 @@ void ModuleBuilder::convertFromTTIRToTTNN(
   mlir::PassManager ttir_to_ttnn_pm(mlir_module.get()->getName());
 
   mlir::tt::ttnn::TTIRToTTNNBackendPipelineOptions options;
+  options.systemDescPath = "system_desc.ttsys";
   mlir::tt::ttnn::createTTIRToTTNNBackendPipeline(ttir_to_ttnn_pm, options);
 
   // Run the pass manager.
diff --git a/src/common/pjrt_implementation/client_instance.cc b/src/common/pjrt_implementation/client_instance.cc
index 003a7985..99624bb6 100644
--- a/src/common/pjrt_implementation/client_instance.cc
+++ b/src/common/pjrt_implementation/client_instance.cc
@@ -164,6 +164,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();
+  system_desc.store("system_desc.ttsys");
   int devices_count = chip_ids.size();
 
   devices_.resize(devices_count);