[PATCH] D107692: [DAGCombine] Prevent the transform of combine for multi-use operand

[Allen zhong via Phabricator via llvm-commits]

Allen updated this revision to Diff 367226.
Allen marked an inline comment as done.
Allen edited the summary of this revision.

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

https://reviews.llvm.org/D107692

Files:
  llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
  llvm/test/CodeGen/AArch64/arm64-srl-and.ll


Index: llvm/test/CodeGen/AArch64/arm64-srl-and.ll
===================================================================
--- llvm/test/CodeGen/AArch64/arm64-srl-and.ll
+++ llvm/test/CodeGen/AArch64/arm64-srl-and.ll
@@ -1,7 +1,8 @@
 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
 ; RUN: llc -mtriple=aarch64-linux-gnu -O3 < %s | FileCheck %s
 
-; Disable the dagcombine if operand has multi use
+; The output register of insn add has multi use,
+; so disable the dagcombine
 
 @g = global i16 0, align 4
 define i32 @srl_and()  {
@@ -12,7 +13,8 @@
 ; CHECK-NEXT:    mov w9, #50
 ; CHECK-NEXT:    ldrh w8, [x8]
 ; CHECK-NEXT:    eor w8, w8, w9
-; CHECK-NEXT:    sub w8, w8, #1
+; CHECK-NEXT:    mov w9, #65535
+; CHECK-NEXT:    add w8, w8, w9
 ; CHECK-NEXT:    and w0, w8, w8, lsr #16
 ; CHECK-NEXT:    ret
 entry:
Index: llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
===================================================================
--- llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
+++ llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
@@ -5149,8 +5149,11 @@
   if (SDValue V = foldLogicOfSetCCs(true, N0, N1, DL))
     return V;
 
+  // FIXME: Add one-use check as there is CombineTo call on N0, which will
+  // affect all users of N0. Need rewrite the transform as extra instruction
+  // bloat after this quick fix.
   if (N0.getOpcode() == ISD::ADD && N1.getOpcode() == ISD::SRL &&
-      VT.getSizeInBits() <= 64) {
+      VT.getSizeInBits() <= 64 && N0->hasOneUse()) {
     if (ConstantSDNode *ADDI = dyn_cast<ConstantSDNode>(N0.getOperand(1))) {
       if (ConstantSDNode *SRLI = dyn_cast<ConstantSDNode>(N1.getOperand(1))) {
         // Look for (and (add x, c1), (lshr y, c2)). If C1 wasn't a legal


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D107692.367226.patch
Type: text/x-patch
Size: 1736 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20210818/6aaf4a8f/attachment.bin>