[PATCH] D129025: [SimplifyCFG] Skip hoisting common instructions that return token type

[Yuanfang Chen via Phabricator via llvm-commits]

ychen updated this revision to Diff 441817.
ychen added a comment.

run update_test_checks.py on llvm/test/Transforms/LowerMatrixIntrinsics/transpose-opts.ll


Repository:
  rG LLVM Github Monorepo

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

https://reviews.llvm.org/D129025

Files:
  llvm/lib/Transforms/Utils/SimplifyCFG.cpp
  llvm/test/Transforms/LowerMatrixIntrinsics/transpose-opts.ll
  llvm/test/Transforms/SimplifyCFG/hoist-skip-token.ll


Index: llvm/test/Transforms/SimplifyCFG/hoist-skip-token.ll
===================================================================
--- /dev/null
+++ llvm/test/Transforms/SimplifyCFG/hoist-skip-token.ll
@@ -0,0 +1,36 @@
+; RUN: opt < %s -passes='simplifycfg<hoist-common-insts>' -S | FileCheck %s
+
+; CHECK: call token @llvm.coro.save(ptr null)
+; CHECK: call token @llvm.coro.save(ptr null)
+
+declare token @llvm.coro.save(ptr)
+declare i8 @llvm.coro.suspend(token, i1)
+
+define void @f(i32 %x) {
+entry:
+  br label %while.cond
+
+while.cond:
+  %cmp = icmp slt i32 %x, 0
+  br i1 %cmp, label %await.suspend, label %final.suspend
+
+await.suspend:
+  %0 = call token @llvm.coro.save(ptr null)
+  %1 = call i8 @llvm.coro.suspend(token %0, i1 false)
+  switch i8 %1, label %coro.ret [
+    i8 0, label %while.cond
+    i8 1, label %cleanup
+  ]
+
+final.suspend:
+  %2 = call token @llvm.coro.save(ptr null)
+  %3 = call i8 @llvm.coro.suspend(token %2, i1 true)
+  %switch = icmp ult i8 %3, 2
+  br i1 %switch, label %cleanup, label %coro.ret
+
+cleanup:
+  br label %coro.ret
+
+coro.ret:
+  ret void
+}
Index: llvm/test/Transforms/LowerMatrixIntrinsics/transpose-opts.ll
===================================================================
--- llvm/test/Transforms/LowerMatrixIntrinsics/transpose-opts.ll
+++ llvm/test/Transforms/LowerMatrixIntrinsics/transpose-opts.ll
@@ -972,7 +972,7 @@
 ; CHECK-NEXT:    [[TMP16:%.*]] = extractelement <2 x float> [[TMP13]], i64 0
 ; CHECK-NEXT:    [[TMP17:%.*]] = insertelement <2 x float> [[TMP15]], float [[TMP16]], i64 1
 ; CHECK-NEXT:    [[TMP18:%.*]] = extractelement <2 x float> [[TMP6]], i64 1
-; CHECK-NEXT:    [[TMP19:%.*]] = insertelement <2 x float> poison float [[TMP18]], i64 0
+; CHECK-NEXT:    [[TMP19:%.*]] = insertelement <2 x float> poison, float [[TMP18]], i64 0
 ; CHECK-NEXT:    [[TMP20:%.*]] = extractelement <2 x float> [[TMP13]], i64 1
 ; CHECK-NEXT:    [[TMP21:%.*]] = insertelement <2 x float> [[TMP19]], float [[TMP20]], i64 1
 ; CHECK-NEXT:    [[TMP22:%.*]] = shufflevector <2 x float> [[TMP17]], <2 x float> [[TMP21]], <4 x i32> <i32 0, i32 1, i32 2, i32 3>
Index: llvm/lib/Transforms/Utils/SimplifyCFG.cpp
===================================================================
--- llvm/lib/Transforms/Utils/SimplifyCFG.cpp
+++ llvm/lib/Transforms/Utils/SimplifyCFG.cpp
@@ -1476,6 +1476,7 @@
   // FIXME: Can we define a safety predicate for CallBr?
   if (isa<PHINode>(I1) || !I1->isIdenticalToWhenDefined(I2) ||
       (isa<InvokeInst>(I1) && !isSafeToHoistInvoke(BB1, BB2, I1, I2)) ||
+      I1->getType()->isTokenTy() || I2->getType()->isTokenTy() ||
       isa<CallBrInst>(I1))
     return false;
 


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D129025.441817.patch
Type: text/x-patch
Size: 2668 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20220701/5a351143/attachment.bin>