From 964412c5b537af4b3b83aee69c93c5cb2cfb1ca4 Mon Sep 17 00:00:00 2001
From: Aleksandr Misonizhnik <misonijnik@gmail.com>
Date: Fri, 11 Nov 2022 04:51:18 +0300
Subject: [PATCH] [fix] Improve CallSplitter

---
 lib/Module/CallSplitter.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/Module/CallSplitter.cpp b/lib/Module/CallSplitter.cpp
index 37e8e8a922b..63f859560f9 100644
--- 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);