[fix] Improve CallSplitter

misonijnik · Nov 11, 2022 · 964412c · 964412c
1 parent 72b0bb9
commit 964412c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/Module/CallSplitter.cpp b/lib/Module/CallSplitter.cpp
@@ -41,8 +41,8 @@ bool CallSplitter::runOnFunction(Function &F) {
         if (callInst != firstInst) {
           fbb = fbb->splitBasicBlock(callInst);
         }
-        if (afterCallInst->isTerminator() && !isa<InvokeInst>(afterCallInst) &&
-            !isa<ReturnInst>(afterCallInst)) {
+        if (isa<BranchInst>(afterCallInst) &&
+            cast<BranchInst>(afterCallInst)->isUnconditional()) {
           break;
         }
         fbb = fbb->splitBasicBlock(afterCallInst);