[triton-mlir] add flag "Link only needed" for external libs. (#838)

This commit is contained in:
ben-zhang-609
2022-11-03 18:50:20 +08:00
committed by GitHub
parent 91a9773b38
commit 61f2ff98df

View File

@@ -197,7 +197,8 @@ translateTritonGPUToLLVMIR(llvm::LLVMContext *llvmContext,
ext_mod->setTargetTriple(llvmir->getTargetTriple()); ext_mod->setTargetTriple(llvmir->getTargetTriple());
ext_mod->setDataLayout(llvmir->getDataLayout()); 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; llvm::errs() << "Failed to link extern lib " << lib.first;
return nullptr; return nullptr;
} }