diff options
-rw-r--r-- | llvm/lib/Transforms/Utils/BuildLibCalls.cpp | 11 | ||||
-rw-r--r-- | llvm/test/CodeGen/X86/no-plt-libcalls.ll | 19 |
2 files changed, 30 insertions, 0 deletions
diff --git a/llvm/lib/Transforms/Utils/BuildLibCalls.cpp b/llvm/lib/Transforms/Utils/BuildLibCalls.cpp index d4cf03c326d..24af7c463f4 100644 --- a/llvm/lib/Transforms/Utils/BuildLibCalls.cpp +++ b/llvm/lib/Transforms/Utils/BuildLibCalls.cpp @@ -105,12 +105,23 @@ static bool setRetNonNull(Function &F) { return true; } +static bool setNonLazyBind(Function &F) { + if (F.hasFnAttribute(Attribute::NonLazyBind)) + return false; + F.addFnAttr(Attribute::NonLazyBind); + return true; +} + bool llvm::inferLibFuncAttributes(Function &F, const TargetLibraryInfo &TLI) { LibFunc TheLibFunc; if (!(TLI.getLibFunc(F, TheLibFunc) && TLI.has(TheLibFunc))) return false; bool Changed = false; + + if (F.getParent() != nullptr && F.getParent()->getRtLibUseGOT()) + Changed |= setNonLazyBind(F); + switch (TheLibFunc) { case LibFunc_strlen: case LibFunc_wcslen: diff --git a/llvm/test/CodeGen/X86/no-plt-libcalls.ll b/llvm/test/CodeGen/X86/no-plt-libcalls.ll new file mode 100644 index 00000000000..795ad57f9a5 --- /dev/null +++ b/llvm/test/CodeGen/X86/no-plt-libcalls.ll @@ -0,0 +1,19 @@ +; Check if "RtLibUseGOT" works correctly when lib calls are simplified. +; RUN: opt < %s -instcombine -S | FileCheck %s + +@percent_s = constant [4 x i8] c"%s\0A\00" +@hello_world = constant [13 x i8] c"hello world\0A\00" +declare i32 @printf(i8*, ...) +define void @printf_call() { + %fmt = getelementptr [4 x i8], [4 x i8]* @percent_s, i32 0, i32 0 + %str = getelementptr [13 x i8], [13 x i8]* @hello_world, i32 0, i32 0 + call i32 (i8*, ...) @printf(i8* %fmt, i8* %str) +; CHECK: call i32 @puts(i8* getelementptr inbounds ([13 x i8], [13 x i8]* @hello_world, i64 0, i64 0)) + ret void +} + +; CHECK: Function Attrs: nounwind nonlazybind +; CHECK-NEXT: declare i32 @puts(i8* nocapture readonly) + +!llvm.module.flags = !{!0} +!0 = !{i32 7, !"RtLibUseGOT", i32 1} |