diff --git a/include/ttmlir/Dialect/TTIR/Transforms/Passes.td b/include/ttmlir/Dialect/TTIR/Transforms/Passes.td index 913c5aea16..fa9d421709 100644 --- 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 index 333fc3f0e0..6a327412c1 100644 --- 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;