[LoopInstSimplify] Ignore users in unreachable blocks. PR55072

[mkazantsev]

Summary

Logic in this pass assumes that all users of loop instructions are
either in the same loop or are LCSSA Phis. In fact, there can also
be users in unreachable blocks that currently break assertions.
Such users don't need to go to the next round of simplifications.

[fhahn]

LGTM, thanks! I think it would be good to remove the branch on undef, as per inline comment and possible remove some of blocks that seem unneeded.

llvm/test/Transforms/LoopInstSimplify/pr55072.ll

6

nit: the test shouldn't be X86 specific, can this be removed?

39

Those branches look like an artifact of llvm-reduce, where it fails to remove them. Are they needed?

43

Avoiding switching on undef might make the test less prone to unnecessary changes in case loop-instsimplify becomes more powerful in the future.

Branch/switch on undef is UB and if the source has UB this will mean verification with tools like Alive2 will always succeed, regardless of what the result of the transform is.

[mkazantsev]

llvm/test/Transforms/LoopInstSimplify/pr55072.ll

6

I guess so. Will update the test before commiting changes.

[mkazantsev]

llvm/test/Transforms/LoopInstSimplify/pr55072.ll

39

https://github.com/llvm/llvm-project/commit/ab17873ee696dfbd6e7b10c5f78192f3990db970

[fhahn]

llvm/test/Transforms/LoopInstSimplify/pr55072.ll

39

Thanks!

llvm/lib/Transforms/Scalar/LoopInstSimplify.cpp

for (BasicBlock *BB : RPOT) {
         Value *V = SimplifyInstruction(&I, SQ.getWithInstruction(&I));
         if (!V || !LI.replacementPreservesLCSSAForm(&I, V))
           continue;
 
         for (Use &U : llvm::make_early_inc_range(I.uses())) {
           auto *UserI = cast<Instruction>(U.getUser());
           U.set(V);
 
+          // Do not bother dealing with unreachable code.
+          if (!DT.isReachableFromEntry(UserI->getParent()))
+            continue;
+
           // If the instruction is used by a PHI node we have already processed
           // we'll need to iterate on the loop body to converge, so add it to
           // the next set.
           if (auto *UserPI = dyn_cast<PHINode>(UserI))
             if (VisitedPHIs.count(UserPI)) {
               Next->insert(UserPI);
               continue;
             }

llvm/test/Transforms/LoopInstSimplify/pr55072.ll

+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
 ; RUN: opt -S %s -passes=loop-instsimplify | FileCheck %s
 ; RUN: opt -S %s -passes='loop-mssa(loop-instsimplify)' -verify-memoryssa | FileCheck %s
 
-; XFAIL: *
-; REQUIRES: asserts
-
 target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128-ni:1-p2:32:8:8:32-ni:2"
 target triple = "x86_64-unknown-linux-gnu"
 
 define i32 @test_01() {
-; CHECK-LABEL: test_01
+; CHECK-LABEL: @test_01(
+; CHECK-NEXT:  bb:
+; CHECK-NEXT:    br label [[BB1:%.*]]
+; CHECK:       bb1:
+; CHECK-NEXT:    br label [[BB3:%.*]]
+; CHECK:       bb2:
+; CHECK-NEXT:    br label [[BB2:%.*]]
+; CHECK:       bb3:
+; CHECK-NEXT:    switch i32 undef, label [[BB8:%.*]] [
+; CHECK-NEXT:    i32 0, label [[BB4:%.*]]
+; CHECK-NEXT:    i32 1, label [[BB6:%.*]]
+; CHECK-NEXT:    i32 2, label [[BB5:%.*]]
+; CHECK-NEXT:    ]
+; CHECK:       bb4:
+; CHECK-NEXT:    br label [[BB6]]
+; CHECK:       bb5:
+; CHECK-NEXT:    br label [[BB8]]
+; CHECK:       bb6:
+; CHECK-NEXT:    br label [[BB8]]
+; CHECK:       bb7:
+; CHECK-NEXT:    ret i32 0
+; CHECK:       bb8:
+; CHECK-NEXT:    br label [[BB3]]
+;
 bb:
   br label %bb1
 
 bb1:                                              ; preds = %bb
   br label %bb3
 
 bb2:                                              ; preds = %bb2
   br label %bb2
 
 bb3:                                              ; preds = %bb8, %bb1
   switch i32 undef, label %bb8 [
-    i32 0, label %bb4
-    i32 1, label %bb6
-    i32 2, label %bb5
   ]
 
 bb4:                                              ; preds = %bb3
   %tmp = lshr exact i32 undef, 16
   br label %bb6
 
 bb5:                                              ; preds = %bb3
   br label %bb8