[PATCH] D152978: [DAGCombiner] Fix crash when trying to replace an indexed store with a narrow store.

[Amara Emerson via Phabricator via llvm-commits]

This revision was landed with ongoing or failed builds.
This revision was automatically updated to reflect the committed changes.
Closed by commit rGf79b0333fc36: [DAGCombiner] Fix crash when trying to replace an indexed store with a narrow… (authored by aemerson).

Repository:
  rG LLVM Github Monorepo

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

https://reviews.llvm.org/D152978

Files:
  llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
  llvm/test/CodeGen/AArch64/replace-load-with-shrink-store-indexed-crash.ll


Index: llvm/test/CodeGen/AArch64/replace-load-with-shrink-store-indexed-crash.ll
===================================================================
--- /dev/null
+++ llvm/test/CodeGen/AArch64/replace-load-with-shrink-store-indexed-crash.ll
@@ -0,0 +1,31 @@
+; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 2
+; RUN: llc < %s -mtriple=aarch64-- | FileCheck %s
+target datalayout = "e-m:o-i64:64-i128:128-n32:64-S128"
+target triple = "arm64-apple-macosx13.0.0"
+
+define void @_Z1hP1f(ptr %j) {
+; CHECK-LABEL: _Z1hP1f:
+; CHECK:       // %bb.0: // %entry
+; CHECK-NEXT:    mov x8, x0
+; CHECK-NEXT:  .LBB0_1: // %for.body
+; CHECK-NEXT:    // =>This Inner Loop Header: Depth=1
+; CHECK-NEXT:    ldr w9, [x8]
+; CHECK-NEXT:    mov w10, w9
+; CHECK-NEXT:    bfi w10, w9, #16, #16
+; CHECK-NEXT:    str w10, [x8], #4
+; CHECK-NEXT:    str w9, [x0]
+; CHECK-NEXT:    b .LBB0_1
+entry:
+  br label %for.body
+
+for.body:                                         ; preds = %for.body, %entry
+  %indvars.iv1 = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
+  %arrayidx3 = getelementptr i32, ptr %j, i64 %indvars.iv1
+  %0 = load i32, ptr %arrayidx3, align 4
+  %and = and i32 %0, 65535
+  %or = mul i32 %and, 65537
+  store i32 %or, ptr %arrayidx3, align 4
+  store i32 %0, ptr %j, align 4
+  %indvars.iv.next = add i64 %indvars.iv1, 1
+  br label %for.body
+}
Index: llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
===================================================================
--- llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
+++ llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
@@ -18922,6 +18922,11 @@
     UseTruncStore = true;
   else
     return SDValue();
+
+  // Can't do this for indexed stores.
+  if (St->isIndexed())
+    return SDValue();
+
   // Check that the target doesn't think this is a bad idea.
   if (St->getMemOperand() &&
       !TLI.allowsMemoryAccess(*DAG.getContext(), DAG.getDataLayout(), VT,


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D152978.531648.patch
Type: text/x-patch
Size: 1975 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20230615/e709dc59/attachment.bin>