[LLVM][Transforms] Add more conditions to setVolatileForInlinedBB

backengineering · Jul 14, 2024 · 0264815 · 0264815
1 parent 06e9d66
commit 0264815
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/llvm/lib/Transforms/Utils/InlineFunction.cpp b/llvm/lib/Transforms/Utils/InlineFunction.cpp
@@ -2019,7 +2019,14 @@ inlineRetainOrClaimRVCalls(CallBase &CB, objcarc::ARCInstKind RVCallKind,
 // This is used to preserve the volatile attribute of the original call instruction.
 static void setVolatileForInlinedBB(BasicBlock *BB) {
 
+  if (!BB)
+    return;
+
   // We need BBs belong to a function.
+  if (!BB->getParent())
+    return;
+
+  // We need BBs belong to a module.
   if (!BB->getParent()->getParent())
     return;