[llvm] dcd23d8 - [FuncSpec] Don't specialise functions with attribute NoDuplicate.
Sjoerd Meijer via llvm-commits
llvm-commits at lists.llvm.org
Thu Jun 17 02:45:02 PDT 2021
Author: Sjoerd Meijer
Date: 2021-06-17T10:32:29+01:00
New Revision: dcd23d875a7e3750903ca4ec865508544757dd99
URL: https://github.com/llvm/llvm-project/commit/dcd23d875a7e3750903ca4ec865508544757dd99
DIFF: https://github.com/llvm/llvm-project/commit/dcd23d875a7e3750903ca4ec865508544757dd99.diff
LOG: [FuncSpec] Don't specialise functions with attribute NoDuplicate.
Differential Revision: https://reviews.llvm.org/D104378
Added:
llvm/test/Transforms/FunctionSpecialization/function-specialization-nodup.ll
Modified:
llvm/lib/Transforms/IPO/FunctionSpecialization.cpp
Removed:
################################################################################
diff --git a/llvm/lib/Transforms/IPO/FunctionSpecialization.cpp b/llvm/lib/Transforms/IPO/FunctionSpecialization.cpp
index 2c381b7c1ef4..df289a62fa31 100644
--- a/llvm/lib/Transforms/IPO/FunctionSpecialization.cpp
+++ b/llvm/lib/Transforms/IPO/FunctionSpecialization.cpp
@@ -555,6 +555,8 @@ bool llvm::runFunctionSpecialization(
for (Function &F : M) {
if (F.isDeclaration())
continue;
+ if (F.hasFnAttribute(Attribute::NoDuplicate))
+ continue;
LLVM_DEBUG(dbgs() << "\nFnSpecialization: Analysing decl: " << F.getName()
<< "\n");
diff --git a/llvm/test/Transforms/FunctionSpecialization/function-specialization-nodup.ll b/llvm/test/Transforms/FunctionSpecialization/function-specialization-nodup.ll
new file mode 100644
index 000000000000..fa726b66fe59
--- /dev/null
+++ b/llvm/test/Transforms/FunctionSpecialization/function-specialization-nodup.ll
@@ -0,0 +1,39 @@
+; RUN: opt -function-specialization -force-function-specialization -S < %s | FileCheck %s
+
+; Function @foo has function attribute 'noduplicate', so check that we don't
+; specialize it:
+
+; CHECK-NOT: @foo.1(
+; CHECK-NOT: @foo.2(
+
+target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
+
+ at A = external dso_local constant i32, align 4
+ at B = external dso_local constant i32, align 4
+
+define dso_local i32 @bar(i32 %x, i32 %y) {
+entry:
+ %tobool = icmp ne i32 %x, 0
+ br i1 %tobool, label %if.then, label %if.else
+
+if.then:
+ %call = call i32 @foo(i32 %x, i32* @A)
+ br label %return
+
+if.else:
+ %call1 = call i32 @foo(i32 %y, i32* @B)
+ br label %return
+
+return:
+ %retval.0 = phi i32 [ %call, %if.then ], [ %call1, %if.else ]
+ ret i32 %retval.0
+}
+
+define internal i32 @foo(i32 %x, i32* %b) #0 {
+entry:
+ %0 = load i32, i32* %b, align 4
+ %add = add nsw i32 %x, %0
+ ret i32 %add
+}
+
+attributes #0 = { noduplicate }
More information about the llvm-commits
mailing list