Uplift third_party/tt-mlir to origin/main (54c5bb24909cc3a05fb993316a33083c047f5bf7) 2025-10-28 (#3004)

This PR uplifts the third_party/tt-mlir submodule to the origin/main

---------

Co-authored-by: nvukobratTT <[email protected]>
Co-authored-by: Aleksandar Zecevic <[email protected]>

Author: 3 people

forge/csrc/passes/mlir_compiler.cpp

@@ -143,11 +143,11 @@ auto run_mlir_compiler_generic(tt::ForgeGraphModule& module, const std::optional
 
         tt::log_info(LogMLIRCompiler, "C++ code for SharedObject generated successfully.");
 
-        const char* TT_METAL_HOME = std::getenv("TT_METAL_HOME");
+        const char* TT_METAL_RUNTIME_ROOT = std::getenv("TT_METAL_RUNTIME_ROOT");
         const char* FORGE_HOME = std::getenv("FORGE_HOME");
-        if (TT_METAL_HOME == nullptr)
+        if (TT_METAL_RUNTIME_ROOT == nullptr)
         {
-            throw std::runtime_error("TT_METAL_HOME environment variable is not set.");
+            throw std::runtime_error("TT_METAL_RUNTIME_ROOT environment variable is not set.");
         }
         if (FORGE_HOME == nullptr)
         {
@@ -159,14 +159,14 @@ auto run_mlir_compiler_generic(tt::ForgeGraphModule& module, const std::optional
         fs::path standalone_dir;
         if (std::getenv("FORGE_IN_WHEEL"))
         {
-            metal_src_dir = fs::path(std::string(TT_METAL_HOME));
+            metal_src_dir = fs::path(std::string(TT_METAL_RUNTIME_ROOT));
             metal_lib_dir = fs::path(std::string(FORGE_HOME)) / "forge/lib";
             standalone_dir = fs::path(std::string(FORGE_HOME)) / "forge/tools/ttnn-standalone";
         }
         else if (std::getenv("FORGE_IN_SOURCE"))
         {
-            metal_src_dir = fs::path(std::string(TT_METAL_HOME));
-            metal_lib_dir = fs::path(std::string(TT_METAL_HOME)).parent_path() / "tt-metal/build/lib";
+            metal_src_dir = fs::path(std::string(TT_METAL_RUNTIME_ROOT));
+            metal_lib_dir = fs::path(std::string(TT_METAL_RUNTIME_ROOT)).parent_path() / "tt-metal/build/lib";
             standalone_dir =
                 fs::path(std::string(FORGE_HOME)).parent_path() / "third_party/tt-mlir/tools/ttnn-standalone";
         }

forge/forge/__init__.py

@@ -26,18 +26,18 @@ def set_home_paths():
         logger.remove()
         logger.add(sys.stdout)
 
-    # TT_METAL_HOME should be one of the following:
+    # TT_METAL_RUNTIME_ROOT should be one of the following:
     in_wheel_path = forge_path / "forge/tt-metal"
     in_source_path = (forge_path.parent.resolve() / "third_party/tt-mlir/third_party/tt-metal/src/tt-metal").resolve()
 
-    if "TT_METAL_HOME" not in os.environ:
+    if "TT_METAL_RUNTIME_ROOT" not in os.environ:
         if in_wheel_path.exists():
-            os.environ["TT_METAL_HOME"] = str(in_wheel_path)
+            os.environ["TT_METAL_RUNTIME_ROOT"] = str(in_wheel_path)
         elif in_source_path.exists():
-            os.environ["TT_METAL_HOME"] = str(in_source_path)
+            os.environ["TT_METAL_RUNTIME_ROOT"] = str(in_source_path)
         else:
             logger.error(
-                f"TT_METAL_HOME environment variable is not set. Tried setting it to {in_wheel_path} and {in_source_path}, but neither exists. Something is wrong with the installation."
+                f"TT_METAL_RUNTIME_ROOT environment variable is not set. Tried setting it to {in_wheel_path} and {in_source_path}, but neither exists. Something is wrong with the installation."
             )
             exit(1)
 
@@ -49,14 +49,14 @@ def set_home_paths():
     else:
         logger.error("Neither wheel nor source path exist for tt-metal. Please check your installation.")
 
-    if pathlib.Path(os.environ["TT_METAL_HOME"]) not in [in_wheel_path, in_source_path]:
-        if pathlib.Path(os.environ["TT_METAL_HOME"]).exists():
+    if pathlib.Path(os.environ["TT_METAL_RUNTIME_ROOT"]) not in [in_wheel_path, in_source_path]:
+        if pathlib.Path(os.environ["TT_METAL_RUNTIME_ROOT"]).exists():
             logger.warning(
-                f"TT_METAL_HOME environment variable is set to {os.environ['TT_METAL_HOME']}, which looks like a non-standard path. Please check if this is intentional. If set incorrectly, it will cause issues during runtime."
+                f"TT_METAL_RUNTIME_ROOT environment variable is set to {os.environ['TT_METAL_RUNTIME_ROOT']}, which looks like a non-standard path. Please check if this is intentional. If set incorrectly, it will cause issues during runtime."
             )
         else:
             logger.error(
-                f"TT_METAL_HOME environment variable is set to {os.environ['TT_METAL_HOME']}, which does not exist. Please check if this is intentional. Unset it so that the default path is used or set it to the correct path."
+                f"TT_METAL_RUNTIME_ROOT environment variable is set to {os.environ['TT_METAL_RUNTIME_ROOT']}, which does not exist. Please check if this is intentional. Unset it so that the default path is used or set it to the correct path."
             )
             exit(1)