[PATCH] D117168: [SROA] Bail out on PHIs in catchswitch BBs

[Heejin Ahn via Phabricator via llvm-commits]

aheejin updated this revision to Diff 399785.
aheejin marked an inline comment as done.
aheejin added a comment.

Add `CHECK-LABEL`


Repository:
  rG LLVM Github Monorepo

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

https://reviews.llvm.org/D117168

Files:
  llvm/lib/Transforms/Scalar/SROA.cpp
  llvm/test/Transforms/SROA/phi-catchswitch.ll


Index: llvm/test/Transforms/SROA/phi-catchswitch.ll
===================================================================
--- /dev/null
+++ llvm/test/Transforms/SROA/phi-catchswitch.ll
@@ -0,0 +1,45 @@
+; RUN: opt < %s -sroa -S | FileCheck %s
+
+target datalayout = "e-m:e-p:32:32-i64:64-n32:64-S128"
+target triple = "wasm32-unknown-unknown"
+
+%struct.foo = type { i32 }
+declare i32 @__gxx_wasm_personality_v0(...)
+declare void @foo()
+
+; Tests if the SROA pass correctly bails out on rewriting PHIs in a catchswitch
+; BB.
+; CHECK-LABEL: @test_phi_catchswitch
+define void @test_phi_catchswitch() personality i8* bitcast (i32 (...)* @__gxx_wasm_personality_v0 to i8*) {
+entry:
+  ; CHECK: alloca
+  %tmp = alloca %struct.foo, align 4
+  %tmp2 = getelementptr inbounds %struct.foo, %struct.foo* %tmp, i32 0, i32 0
+  invoke void @foo()
+          to label %bb3 unwind label %bb10
+
+bb3:
+  invoke void @foo()
+          to label %bb9 unwind label %catch.dispatch
+
+catch.dispatch:
+  ; While rewriting the alloca in the entry BB, the SROA pass tries to insert a
+  ; non-PHI instruction in this BB by calling getFirstInsertionPt(), which is
+  ; not possible in a catchswitch BB. This test checks if we correctly bail out
+  ; on these cases.
+  %tmp5 = phi i32* [ %tmp2, %bb3 ]
+  %tmp6 = catchswitch within none [label %catch.start] unwind label %bb10
+
+catch.start:
+  %tmp8 = catchpad within %tmp6 [i8* null]
+  unreachable
+
+bb9:
+  unreachable
+
+bb10:
+  %tmp11 = phi i32* [ %tmp2, %entry ], [ %tmp5, %catch.dispatch ]
+  %tmp12 = cleanuppad within none []
+  store i32 0, i32* %tmp11, align 4
+  unreachable
+}
Index: llvm/lib/Transforms/Scalar/SROA.cpp
===================================================================
--- llvm/lib/Transforms/Scalar/SROA.cpp
+++ llvm/lib/Transforms/Scalar/SROA.cpp
@@ -1008,6 +1008,13 @@
     if (I.use_empty())
       return markAsDead(I);
 
+    // If this is a PHI node before a catchswitch, we cannot insert any non-PHI
+    // instructions in this BB, which may be required during rewriting. Bail out
+    // on these cases.
+    if (isa<PHINode>(I) &&
+        I.getParent()->getFirstInsertionPt() == I.getParent()->end())
+      return PI.setAborted(&I);
+
     // TODO: We could use SimplifyInstruction here to fold PHINodes and
     // SelectInsts. However, doing so requires to change the current
     // dead-operand-tracking mechanism. For instance, suppose neither loading


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D117168.399785.patch
Type: text/x-patch
Size: 2440 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20220113/9bddb819/attachment.bin>