test/Transforms/CorrelatedValuePropagation/ashr.ll

   1 ; RUN: opt < %s -correlated-propagation -S | FileCheck %s
   2
   3 ; Check that debug locations are preserved. For more info see:
   4 ;   https://llvm.org/docs/SourceLevelDebugging.html#fixing-errors
   5 ; RUN: opt < %s -enable-debugify -correlated-propagation -S 2>&1 | \
   6 ; RUN:   FileCheck %s -check-prefix=DEBUG
   7 ; DEBUG: CheckModuleDebugify: PASS
   8
   9 ; CHECK-LABEL: @test1
  10 define void @test1(i32 %n) {
  11 entry:
  12   br label %for.cond
  13
  14 for.cond:                                         ; preds = %for.body, %entry
  15   %a = phi i32 [ %n, %entry ], [ %shr, %for.body ]
  16   %cmp = icmp sgt i32 %a, 1
  17   br i1 %cmp, label %for.body, label %for.end
  18
  19 for.body:                                         ; preds = %for.cond
  20 ; CHECK: lshr i32 %a, 5
  21   %shr = ashr i32 %a, 5
  22   br label %for.cond
  23
  24 for.end:                                          ; preds = %for.cond
  25   ret void
  26 }
  27
  28 ;; Negative test to show transform doesn't happen unless n > 0.
  29 ; CHECK-LABEL: @test2
  30 define void @test2(i32 %n) {
  31 entry:
  32   br label %for.cond
  33
  34 for.cond:                                         ; preds = %for.body, %entry
  35   %a = phi i32 [ %n, %entry ], [ %shr, %for.body ]
  36   %cmp = icmp sgt i32 %a, -2
  37   br i1 %cmp, label %for.body, label %for.end
  38
  39 for.body:                                         ; preds = %for.cond
  40 ; CHECK: ashr i32 %a, 2
  41   %shr = ashr i32 %a, 2
  42   br label %for.cond
  43
  44 for.end:                                          ; preds = %for.cond
  45   ret void
  46 }
  47
  48 ;; Non looping test case.
  49 ; CHECK-LABEL: @test3
  50 define void @test3(i32 %n) {
  51 entry:
  52   %cmp = icmp sgt i32 %n, 0
  53   br i1 %cmp, label %bb, label %exit
  54
  55 bb:
  56 ; CHECK: lshr exact i32 %n, 4
  57   %shr = ashr exact i32 %n, 4
  58   br label %exit
  59
  60 exit:
  61   ret void
  62 }
  63
  64 ; looping case where loop has exactly one block
  65 ; at the point of ashr, we know that the operand is always greater than 0,
  66 ; because of the guard before it, so we can transform it to lshr.
  67 declare void @llvm.experimental.guard(i1,...)
  68 ; CHECK-LABEL: @test4
  69 define void @test4(i32 %n) {
  70 entry:
  71   %cmp = icmp sgt i32 %n, 0
  72   br i1 %cmp, label %loop, label %exit
  73
  74 loop:
  75 ; CHECK: lshr i32 %a, 1
  76   %a = phi i32 [ %n, %entry ], [ %shr, %loop ]
  77   %cond = icmp sgt i32 %a, 2
  78   call void(i1,...) @llvm.experimental.guard(i1 %cond) [ "deopt"() ]
  79   %shr = ashr i32 %a, 1
  80   br i1 %cond, label %loop, label %exit
  81
  82 exit:
  83   ret void
  84 }
  85
  86 ; same test as above with assume instead of guard.
  87 declare void @llvm.assume(i1)
  88 ; CHECK-LABEL: @test5
  89 define void @test5(i32 %n) {
  90 entry:
  91   %cmp = icmp sgt i32 %n, 0
  92   br i1 %cmp, label %loop, label %exit
  93
  94 loop:
  95 ; CHECK: lshr i32 %a, 1
  96   %a = phi i32 [ %n, %entry ], [ %shr, %loop ]
  97   %cond = icmp sgt i32 %a, 4
  98   call void @llvm.assume(i1 %cond)
  99   %shr = ashr i32 %a, 1
 100   %loopcond = icmp sgt i32 %shr, 8
 101   br i1 %loopcond, label %loop, label %exit
 102
 103 exit:
 104   ret void
 105 }