llvm
diff --git a/‎llvm/lib/Target/AMDGPU/AMDGPUTargetMachine.cpp
Copy file name to clipboardExpand all lines: llvm/lib/Target/AMDGPU/AMDGPUTargetMachine.cpp
+9Lines changed: 9 additions & 0 deletions b/‎llvm/lib/Target/AMDGPU/AMDGPUTargetMachine.cpp
Copy file name to clipboardExpand all lines: llvm/lib/Target/AMDGPU/AMDGPUTargetMachine.cpp
+9Lines changed: 9 additions & 0 deletions
@@ -793,6 +793,15 @@ void AMDGPUTargetMachine::registerPassBuilderCallbacks(
 
         PM.addPass(createCGSCCToFunctionPassAdaptor(std::move(FPM)));
       });
+
+  PB.registerFullLinkTimeOptimizationLastEPCallback(
+      [this](ModulePassManager &PM, OptimizationLevel Level) {
+        // We want to support the -lto-partitions=N option as "best effort".
+        // For that, we need to lower LDS earlier in the pipeline before the
+        // module is partitioned for codegen.
+        if (EnableLowerModuleLDS)
+          PM.addPass(AMDGPULowerModuleLDSPass(*this));
+      });
 }
 
 int64_t AMDGPUTargetMachine::getNullPointerValue(unsigned AddrSpace) {