llvm/test/Transforms/IndVarSimplify/pr55925.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt -S -passes=indvars < %s | FileCheck %s
   3
   4 ; This tests the case where a terminator can be modeled by SCEV,
   5 ; because it has a returned attribute.
   6
   7 target datalayout = "e-m:o-i64:64-i128:128-n32:64-S128"
   8
   9 declare i32 @foo(i32)
  10
  11 define void @test(ptr %p) personality ptr undef {
  12 ; CHECK-LABEL: @test(
  13 ; CHECK-NEXT:  entry:
  14 ; CHECK-NEXT:    br label [[LOOP:%.*]]
  15 ; CHECK:       loop:
  16 ; CHECK-NEXT:    [[INDVARS_IV:%.*]] = phi i64 [ [[INDVARS_IV_NEXT:%.*]], [[LOOP_LATCH:%.*]] ], [ 0, [[ENTRY:%.*]] ]
  17 ; CHECK-NEXT:    [[TMP0:%.*]] = trunc nuw i64 [[INDVARS_IV]] to i32
  18 ; CHECK-NEXT:    [[RES:%.*]] = invoke i32 @foo(i32 returned [[TMP0]])
  19 ; CHECK-NEXT:            to label [[LOOP_LATCH]] unwind label [[EXIT:%.*]]
  20 ; CHECK:       loop.latch:
  21 ; CHECK-NEXT:    [[TMP1:%.*]] = trunc nuw i64 [[INDVARS_IV]] to i32
  22 ; CHECK-NEXT:    [[TMP2:%.*]] = call i32 @foo(i32 [[TMP1]])
  23 ; CHECK-NEXT:    [[INDVARS_IV_NEXT]] = add i64 [[INDVARS_IV]], 1
  24 ; CHECK-NEXT:    br label [[LOOP]]
  25 ; CHECK:       exit:
  26 ; CHECK-NEXT:    [[LP:%.*]] = landingpad { ptr, i32 }
  27 ; CHECK-NEXT:            cleanup
  28 ; CHECK-NEXT:    ret void
  29 ;
  30 entry:
  31   br label %loop
  32
  33 loop:
  34   %iv = phi i32 [ 0, %entry ], [ %iv.next, %loop.latch ]
  35   %res = invoke i32 @foo(i32 returned %iv)
  36   to label %loop.latch unwind label %exit
  37
  38 loop.latch:
  39   %ext = zext i32 %iv to i64
  40   %tmp5 = getelementptr inbounds i8, ptr %p, i64 %ext
  41   %iv.next = add nuw i32 %iv, 1
  42   call i32 @foo(i32 %res)
  43   br label %loop
  44
  45 exit:
  46   %lp = landingpad { ptr, i32 }
  47   cleanup
  48   ret void
  49 }
  50
  51 define void @test_critedge(i1 %c, ptr %p) personality ptr undef {
  52 ; CHECK-LABEL: @test_critedge(
  53 ; CHECK-NEXT:  entry:
  54 ; CHECK-NEXT:    br label [[LOOP:%.*]]
  55 ; CHECK:       loop:
  56 ; CHECK-NEXT:    [[INDVARS_IV:%.*]] = phi i64 [ [[INDVARS_IV_NEXT:%.*]], [[LOOP_LATCH:%.*]] ], [ 0, [[ENTRY:%.*]] ]
  57 ; CHECK-NEXT:    br i1 [[C:%.*]], label [[LOOP_INVOKE:%.*]], label [[LOOP_OTHER:%.*]]
  58 ; CHECK:       loop.invoke:
  59 ; CHECK-NEXT:    [[TMP0:%.*]] = trunc nuw i64 [[INDVARS_IV]] to i32
  60 ; CHECK-NEXT:    [[TMP1:%.*]] = trunc nuw i64 [[INDVARS_IV]] to i32
  61 ; CHECK-NEXT:    [[RES:%.*]] = invoke i32 @foo(i32 returned [[TMP0]])
  62 ; CHECK-NEXT:            to label [[LOOP_LATCH]] unwind label [[EXIT:%.*]]
  63 ; CHECK:       loop.other:
  64 ; CHECK-NEXT:    br label [[LOOP_LATCH]]
  65 ; CHECK:       loop.latch:
  66 ; CHECK-NEXT:    [[PHI:%.*]] = phi i32 [ [[TMP1]], [[LOOP_INVOKE]] ], [ 0, [[LOOP_OTHER]] ]
  67 ; CHECK-NEXT:    [[TMP2:%.*]] = call i32 @foo(i32 [[PHI]])
  68 ; CHECK-NEXT:    [[INDVARS_IV_NEXT]] = add i64 [[INDVARS_IV]], 1
  69 ; CHECK-NEXT:    br label [[LOOP]]
  70 ; CHECK:       exit:
  71 ; CHECK-NEXT:    [[LP:%.*]] = landingpad { ptr, i32 }
  72 ; CHECK-NEXT:            cleanup
  73 ; CHECK-NEXT:    ret void
  74 ;
  75 entry:
  76   br label %loop
  77
  78 loop:
  79   %iv = phi i32 [ 0, %entry ], [ %iv.next, %loop.latch ]
  80   br i1 %c, label %loop.invoke, label %loop.other
  81
  82 loop.invoke:
  83   %res = invoke i32 @foo(i32 returned %iv)
  84   to label %loop.latch unwind label %exit
  85
  86 loop.other:
  87   br label %loop.latch
  88
  89 loop.latch:
  90   %phi = phi i32 [ %res, %loop.invoke ], [ 0, %loop.other ]
  91   %ext = zext i32 %iv to i64
  92   %tmp5 = getelementptr inbounds i8, ptr %p, i64 %ext
  93   %iv.next = add nuw i32 %iv, 1
  94   call i32 @foo(i32 %phi)
  95   br label %loop
  96
  97 exit:
  98   %lp = landingpad { ptr, i32 }
  99   cleanup
 100   ret void
 101 }