[SROA] Remove UB-implying metadata when promoting speculative instruction.

[DianQK]

Summary

After D138238 introduced the then/else blocks, we should remove UB-implying metadata for the promoted speculative instruction.

This patch fixes the LTO=thin appears broken on nightly rustc? #110256.

[StephenFan]

It seems that !noundef means that the loaded value is not an undefined value instead of loading a value from an under ptr is UB.

[DianQK]

In D148456#4271868, @StephenFan wrote:

It seems that !noundef means that the loaded value is not an undefined value instead of loading a value from an under ptr is UB.

Thanks, I tried to correct the summary.

[nikic]

You are looking for dropUBImplyingAttrsAndUnknownMetadata().

[nikic]

Well, actually you are looking for dropUBImplyingAttrsAndMetadata() from D146629, but for now you can just drop all unknown metadata, and we can add the UB implying distinction later.

[DianQK]

Promote the load instruction by calling dropUBImplyingAttrsAndMetadata.

Thanks for the reminder that dropUBImplyingAttrsAndMetadata()/dropUBImplyingAttrsAndUnknownMetadata() is the method I was looking for.

[nikic]

LG

llvm/test/Transforms/SROA/select-load-with-nonnull.ll

1

I'd suggest adding this test to llvm/test/Transforms/SROA/select-load.ll instead.

[DianQK]

Move the test to llvm/test/Transforms/SROA/select-load.ll.

Maybe we should also cherry-pick to 16.0.2?

llvm/lib/Transforms/Scalar/SROA.cpp

static void rewriteMemOpOfSelect(SelectInst &SI, T &I,
   Tail->setName(Head->getName() + ".cont");
   PHINode *PN;
   if (isa<LoadInst>(I))
     PN = PHINode::Create(I.getType(), 2, "", &I);
   for (BasicBlock *SuccBB : successors(Head)) {
     bool IsThen = SuccBB == HeadBI->getSuccessor(0);
     int SuccIdx = IsThen ? 0 : 1;
     auto *NewMemOpBB = SuccBB == Tail ? Head : SuccBB;
+    auto &CondMemOp = cast<T>(*I.clone());
     if (NewMemOpBB != Head) {
       NewMemOpBB->setName(Head->getName() + (IsThen ? ".then" : ".else"));
       if (isa<LoadInst>(I))
         ++NumLoadsPredicated;
       else
         ++NumStoresPredicated;
-    } else
+    } else {
+      CondMemOp.dropUBImplyingAttrsAndUnknownMetadata();
       ++NumLoadsSpeculated;
-    auto &CondMemOp = cast<T>(*I.clone());
+    }
     CondMemOp.insertBefore(NewMemOpBB->getTerminator());
     Value *Ptr = SI.getOperand(1 + SuccIdx);
     if (auto *PtrTy = Ptr->getType();
         !PtrTy->isOpaquePointerTy() &&
         PtrTy != CondMemOp.getPointerOperandType())
       Ptr = BitCastInst::CreatePointerBitCastOrAddrSpaceCast(
           Ptr, CondMemOp.getPointerOperandType(), "", &CondMemOp);
     CondMemOp.setOperand(I.getPointerOperandIndex(), Ptr);

llvm/test/Transforms/SROA/select-load-with-nonnull.ll

+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt < %s -passes='sroa<modify-cfg>' -S | FileCheck %s
+
+; When promoting speculative instruction, metadata that may trigger immediate UB should be dropped.
+define void @load_of_select_with_noundef_nonnull(ptr %buffer, i1 %b) {
+; CHECK-LABEL: @load_of_select_with_noundef_nonnull(
+; CHECK-NEXT:    br i1 [[B:%.*]], label [[DOTTHEN:%.*]], label [[DOTCONT:%.*]]
+; CHECK:       .then:
+; CHECK-NEXT:    [[LOAD_PTR_THEN_VAL:%.*]] = load ptr, ptr [[BUFFER:%.*]], align 8, !nonnull !0, !noundef !0
+; CHECK-NEXT:    br label [[DOTCONT]]
+; CHECK:       .cont:
+; CHECK-NEXT:    [[LOAD_PTR:%.*]] = phi ptr [ [[LOAD_PTR_THEN_VAL]], [[DOTTHEN]] ], [ undef, [[TMP0:%.*]] ]
+; CHECK-NEXT:    ret void
+;
+  %ub_ptr = alloca ptr
+  %select_ptr = select i1 %b, ptr %buffer, ptr %ub_ptr
+  %load_ptr = load ptr, ptr %select_ptr, !nonnull !1, !noundef !1
+  ret void
+}
+
+!1 = !{}