[llvm] [LV] Don't mark ptrs as safe to speculate if fed by UB/poison op. (PR #143204)

[Florian Hahn via llvm-commits]

================
@@ -1493,10 +1493,51 @@ bool LoopVectorizationLegality::canVectorizeWithIfConvert() {
     SmallVector<const SCEVPredicate *, 4> Predicates;
     for (Instruction &I : *BB) {
       LoadInst *LI = dyn_cast<LoadInst>(&I);
+
+      // Make sure we can execute all computations feeding into Ptr in the loop
+      // w/o triggering UB and that none of the out-of-loop operands are poison.
+      // We do not need to check if operations inside the loop can produce
+      // poison due to flags (e.g. due to an inbounds GEP going out of bounds),
----------------
fhahn wrote:

Yes it includes all relevant flags, including `nuw`, `nsw`, `inbounds`. The flags are only dropped for any op that computes a pointer for loads that are executed conditionally in the original loop but executed unconditionally in the vector loop.

https://github.com/llvm/llvm-project/pull/143204