[llvm] 1789534 - [SelectOpt] Don't convert constant selects to branches. (#110858)

[via llvm-commits]

Author: David Green
Date: 2024-10-05T16:17:22+01:00
New Revision: 1789534ac1ac95f9a3369625e331b9a446be60d9

URL: https://github.com/llvm/llvm-project/commit/1789534ac1ac95f9a3369625e331b9a446be60d9
DIFF: https://github.com/llvm/llvm-project/commit/1789534ac1ac95f9a3369625e331b9a446be60d9.diff

LOG: [SelectOpt] Don't convert constant selects to branches. (#110858)

Selects that choose between two constants will be less profitable to
turn into branches, especially if the constants can be folded somehow
into the surrounding instructions. They will also be cost modelled in a
way that can make them over-optimistically converted to branches, as
neither branch will have a latency depth but the constants still need to
be materialized.

This patch disabled selectopt for selects with two constant branches. It
is currently in the target independent part, as it sounds generic, but I
could move it into AArch64 if needed.

Added: 
    llvm/test/CodeGen/AArch64/selectopt-const.ll

Modified: 
    llvm/include/llvm/Analysis/TargetTransformInfoImpl.h

Removed: 
    


################################################################################
diff  --git a/llvm/include/llvm/Analysis/TargetTransformInfoImpl.h b/llvm/include/llvm/Analysis/TargetTransformInfoImpl.h
index eca8818cc25e62..50040dc8f6165b 100644
--- a/llvm/include/llvm/Analysis/TargetTransformInfoImpl.h
+++ b/llvm/include/llvm/Analysis/TargetTransformInfoImpl.h
@@ -407,9 +407,13 @@ class TargetTransformInfoImplBase {
   bool enableSelectOptimize() const { return true; }
 
   bool shouldTreatInstructionLikeSelect(const Instruction *I) {
+    // A select with two constant operands will usually be better left as a
+    // select.
+    using namespace llvm::PatternMatch;
+    if (match(I, m_Select(m_Value(), m_Constant(), m_Constant())))
+      return false;
     // If the select is a logical-and/logical-or then it is better treated as a
     // and/or by the backend.
-    using namespace llvm::PatternMatch;
     return isa<SelectInst>(I) &&
            !match(I, m_CombineOr(m_LogicalAnd(m_Value(), m_Value()),
                                  m_LogicalOr(m_Value(), m_Value())));

diff  --git a/llvm/test/CodeGen/AArch64/selectopt-const.ll b/llvm/test/CodeGen/AArch64/selectopt-const.ll
new file mode 100644
index 00000000000000..f10327e136ad13
--- /dev/null
+++ b/llvm/test/CodeGen/AArch64/selectopt-const.ll
@@ -0,0 +1,72 @@
+; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
+; RUN: llc -mtriple=aarch64-linux-gnu -mcpu=neoverse-v2 -O3 < %s | FileCheck %s
+
+define i32 @test_const(ptr %in1, ptr %in2, ptr %out, i32 %n, ptr %tbl) {
+; CHECK-LABEL: test_const:
+; CHECK:       // %bb.0: // %entry
+; CHECK-NEXT:    cmp w3, #1
+; CHECK-NEXT:    b.lt .LBB0_3
+; CHECK-NEXT:  // %bb.1: // %for.body.preheader
+; CHECK-NEXT:    mov w9, #1267 // =0x4f3
+; CHECK-NEXT:    fmov s1, #1.00000000
+; CHECK-NEXT:    fmov d2, #5.00000000
+; CHECK-NEXT:    mov w8, w3
+; CHECK-NEXT:    movk w9, #16309, lsl #16
+; CHECK-NEXT:    fmov s0, w9
+; CHECK-NEXT:    .p2align 5, , 16
+; CHECK-NEXT:  .LBB0_2: // %for.body
+; CHECK-NEXT:    // =>This Inner Loop Header: Depth=1
+; CHECK-NEXT:    ldr s4, [x1], #4
+; CHECK-NEXT:    ldr w9, [x0], #4
+; CHECK-NEXT:    add w9, w9, #10
+; CHECK-NEXT:    scvtf d3, w9
+; CHECK-NEXT:    fmadd s4, s4, s0, s1
+; CHECK-NEXT:    fabs s4, s4
+; CHECK-NEXT:    fcvt d4, s4
+; CHECK-NEXT:    fdiv d3, d3, d4
+; CHECK-NEXT:    fcmp d3, d2
+; CHECK-NEXT:    cset w9, lt
+; CHECK-NEXT:    subs x8, x8, #1
+; CHECK-NEXT:    ubfiz x9, x9, #4, #32
+; CHECK-NEXT:    ldr s3, [x4, x9]
+; CHECK-NEXT:    fcvtzs w9, s3
+; CHECK-NEXT:    str w9, [x2], #4
+; CHECK-NEXT:    b.ne .LBB0_2
+; CHECK-NEXT:  .LBB0_3: // %for.cond.cleanup
+; CHECK-NEXT:    mov w0, wzr
+; CHECK-NEXT:    ret
+entry:
+  %cmp15 = icmp sgt i32 %n, 0
+  br i1 %cmp15, label %for.body.preheader, label %for.cond.cleanup
+
+for.body.preheader:                               ; preds = %entry
+  %wide.trip.count = zext nneg i32 %n to i64
+  br label %for.body
+
+for.cond.cleanup:                                 ; preds = %for.body, %entry
+  ret i32 0
+
+for.body:                                         ; preds = %for.body.preheader, %for.body
+  %indvars.iv = phi i64 [ 0, %for.body.preheader ], [ %indvars.iv.next, %for.body ]
+  %arrayidx = getelementptr inbounds i32, ptr %in1, i64 %indvars.iv
+  %0 = load i32, ptr %arrayidx, align 4
+  %add = add nsw i32 %0, 10
+  %conv = sitofp i32 %add to double
+  %arrayidx2 = getelementptr inbounds float, ptr %in2, i64 %indvars.iv
+  %1 = load float, ptr %arrayidx2, align 4
+  %mul = fmul fast float %1, 0x3FF6A09E60000000
+  %add3 = fadd fast float %mul, 1.000000e+00
+  %2 = tail call fast float @llvm.fabs.f32(float %add3)
+  %3 = fpext float %2 to double
+  %div = fdiv fast double %conv, %3
+  %cmp5 = fcmp fast olt double %div, 5.000000e+00
+  %idxprom6 = select i1 %cmp5, i64 4, i64 0
+  %arrayidx7 = getelementptr inbounds float, ptr %tbl, i64 %idxprom6
+  %4 = load float, ptr %arrayidx7, align 4
+  %conv8 = fptosi float %4 to i32
+  %arrayidx10 = getelementptr inbounds i32, ptr %out, i64 %indvars.iv
+  store i32 %conv8, ptr %arrayidx10, align 4
+  %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
+  %exitcond.not = icmp eq i64 %indvars.iv.next, %wide.trip.count
+  br i1 %exitcond.not, label %for.cond.cleanup, label %for.body
+}