From 61f2ff98df81ad749cf050bfa430cc9312b9e720 Mon Sep 17 00:00:00 2001
From: ben-zhang-609 <110140741+ben-zhang-609@users.noreply.github.com>
Date: Thu, 3 Nov 2022 18:50:20 +0800
Subject: [PATCH] [triton-mlir] add flag "Link only needed" for external libs.
 (#838)

---
 lib/Target/LLVMIR/LLVMIRTranslation.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/lib/Target/LLVMIR/LLVMIRTranslation.cpp b/lib/Target/LLVMIR/LLVMIRTranslation.cpp
index be2e65b31..eaabb7c24 100644
--- a/lib/Target/LLVMIR/LLVMIRTranslation.cpp
+++ b/lib/Target/LLVMIR/LLVMIRTranslation.cpp
@@ -197,7 +197,8 @@ translateTritonGPUToLLVMIR(llvm::LLVMContext *llvmContext,
     ext_mod->setTargetTriple(llvmir->getTargetTriple());
     ext_mod->setDataLayout(llvmir->getDataLayout());
 
-    if (llvm::Linker::linkModules(*llvmir, std::move(ext_mod))) {
+    if (llvm::Linker::linkModules(*llvmir, std::move(ext_mod),
+                                  llvm::Linker::Flags::LinkOnlyNeeded)) {
       llvm::errs() << "Failed to link extern lib " << lib.first;
       return nullptr;
     }