Add AttachMetalLayout to the metal backend pipeline (#1968)

Set pipeline pass option default memory space to L1
tenstorrent · Jan 28, 2025 · b7596a4 · b7596a4
1 parent d739ed9
commit b7596a4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/include/ttmlir/Dialect/TTIR/Transforms/Passes.td b/include/ttmlir/Dialect/TTIR/Transforms/Passes.td
@@ -29,7 +29,7 @@ def TTIRAttachMetalLayout: Pass<"ttir-attach-metal-layout", "::mlir::ModuleOp">
   let options = [
     Option<"initMemorySpace", "init-memory-space",
           "::mlir::tt::MemorySpace",
-          /*default=*/"::mlir::tt::MemorySpace::System",
+          /*default=*/"::mlir::tt::MemorySpace::DeviceL1",
            "Set the initial memory space for tensors to start in">
   ];
 }

diff --git a/lib/Dialect/TTMetal/Pipelines/TTMetalPipelines.cpp b/lib/Dialect/TTMetal/Pipelines/TTMetalPipelines.cpp
@@ -24,6 +24,7 @@ void createTTIRToTTMetalBackendPipeline(
       options.meshShape.begin(), options.meshShape.end());
   pm.addPass(mlir::tt::ttir::createTTIRImplicitDevice(implicitDeviceOptions));
   pm.addPass(mlir::tt::ttir::createTTIRConstantAsFill());
+  pm.addPass(mlir::tt::ttir::createTTIRAttachMetalLayout());
   pm.addPass(mlir::tt::ttir::createTTIRGenericRegion());
   mlir::tt::ttir::TTIRLayoutOptions layoutOptions;
   layoutOptions.initMemorySpace = mlir::tt::MemorySpace::DeviceL1;