[PATCH] D135867: [FuncSpec] Fix missed opportunities for function specialisation

[Momchil Velikov via Phabricator via llvm-commits]

This revision was automatically updated to reflect the committed changes.
Closed by commit rG1a525dec7f3f: [FuncSpec] Fix missed opportunities for function specialisation (authored by chill).

Repository:
  rG LLVM Github Monorepo

CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D135867/new/

https://reviews.llvm.org/D135867

Files:
  llvm/lib/Transforms/IPO/FunctionSpecialization.cpp
  llvm/test/Transforms/FunctionSpecialization/get-possible-constants.ll


Index: llvm/test/Transforms/FunctionSpecialization/get-possible-constants.ll
===================================================================
--- /dev/null
+++ llvm/test/Transforms/FunctionSpecialization/get-possible-constants.ll
@@ -0,0 +1,42 @@
+; RUN: opt -S --passes=function-specialization < %s | FileCheck %s
+define dso_local i32 @p0(i32 noundef %x) {
+entry:
+  %add = add nsw i32 %x, 1
+  ret i32 %add
+}
+
+define dso_local i32 @p1(i32 noundef %x) {
+entry:
+  %sub = add nsw i32 %x, -1
+  ret i32 %sub
+}
+
+define dso_local i32 @g0(i32 noundef %x) {
+entry:
+  %call = tail call fastcc i32 @f(i32 noundef %x, ptr noundef nonnull @p0)
+  ret i32 %call
+}
+
+define internal fastcc i32 @f(i32 noundef %x, ptr nocapture noundef readonly %p) noinline  {
+entry:
+  %call = tail call i32 %p(i32 noundef %x)
+  %add = add nsw i32 %call, %x
+  ret i32 %add
+}
+
+define dso_local i32 @g1(i32 noundef %x) {
+entry:
+  %call = tail call fastcc i32 @f(i32 noundef %x, ptr noundef nonnull @p1)
+  ret i32 %call
+}
+
+define dso_local i32 @g2(i32 noundef %x) {
+entry:
+  %call = tail call fastcc i32 @f(i32 noundef %x, ptr poison)
+  ret i32 %call
+}
+
+; Check that a single argument, that cannot be used for specialisation, does not
+; prevent specialisation based on other arguments.
+; CHECK: @f.1
+; CHECK: @f.2
Index: llvm/lib/Transforms/IPO/FunctionSpecialization.cpp
===================================================================
--- llvm/lib/Transforms/IPO/FunctionSpecialization.cpp
+++ llvm/lib/Transforms/IPO/FunctionSpecialization.cpp
@@ -731,18 +731,17 @@
 
       auto *V = CS.getArgOperand(A->getArgNo());
       if (isa<PoisonValue>(V))
-        return;
+        continue;
 
       // TrackValueOfGlobalVariable only tracks scalar global variables.
       if (auto *GV = dyn_cast<GlobalVariable>(V)) {
         // Check if we want to specialize on the address of non-constant
         // global values.
-        if (!GV->isConstant())
-          if (!SpecializeOnAddresses)
-            return;
+        if (!GV->isConstant() && !SpecializeOnAddresses)
+          continue;
 
         if (!GV->getValueType()->isSingleValueType())
-          return;
+          continue;
       }
 
       if (isa<Constant>(V) && (Solver.getLatticeValueFor(V).isConstant() ||


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D135867.470629.patch
Type: text/x-patch
Size: 2285 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20221025/7aad6ec7/attachment.bin>