test/Transforms/LICM/argmemonly-call.ll

   1 ; RUN: opt -S -basicaa -licm -licm-n2-threshold=0 %s | FileCheck %s
   2 ; RUN: opt -licm -basicaa -licm-n2-threshold=200 < %s -S | FileCheck %s --check-prefix=ALIAS-N2
   3 ; RUN: opt -aa-pipeline=basic-aa -licm-n2-threshold=0 -passes='require<aa>,require<targetir>,require<scalar-evolution>,require<opt-remark-emit>,loop(licm)' < %s -S | FileCheck %s
   4 ; RUN: opt -aa-pipeline=basic-aa -licm-n2-threshold=200 -passes='require<aa>,require<targetir>,require<scalar-evolution>,require<opt-remark-emit>,loop(licm)' < %s -S | FileCheck %s --check-prefix=ALIAS-N2
   5 ; RUN: opt -S -basicaa -licm -licm-n2-threshold=0 -enable-mssa-loop-dependency=true -verify-memoryssa %s | FileCheck %s --check-prefix=ALIAS-N2
   6
   7 declare i32 @foo() readonly argmemonly nounwind
   8 declare i32 @foo2() readonly nounwind
   9 declare i32 @bar(i32* %loc2) readonly argmemonly nounwind
  10
  11 define void @test(i32* %loc) {
  12 ; CHECK-LABEL: @test
  13 ; CHECK: @foo
  14 ; CHECK-LABEL: loop:
  15 ; ALIAS-N2-LABEL: @test
  16 ; ALIAS-N2: @foo
  17 ; ALIAS-N2-LABEL: loop:
  18   br label %loop
  19
  20 loop:
  21   %res = call i32 @foo()
  22   store i32 %res, i32* %loc
  23   br label %loop
  24 }
  25
  26 ; Negative test: show argmemonly is required
  27 define void @test_neg(i32* %loc) {
  28 ; CHECK-LABEL: @test_neg
  29 ; CHECK-LABEL: loop:
  30 ; CHECK: @foo
  31 ; ALIAS-N2-LABEL: @test_neg
  32 ; ALIAS-N2-LABEL: loop:
  33 ; ALIAS-N2: @foo
  34   br label %loop
  35
  36 loop:
  37   %res = call i32 @foo2()
  38   store i32 %res, i32* %loc
  39   br label %loop
  40 }
  41
  42 define void @test2(i32* noalias %loc, i32* noalias %loc2) {
  43 ; CHECK-LABEL: @test2
  44 ; CHECK: @bar
  45 ; CHECK-LABEL: loop:
  46 ; ALIAS-N2-LABEL: @test2
  47 ; ALIAS-N2: @bar
  48 ; ALIAS-N2-LABEL: loop:
  49   br label %loop
  50
  51 loop:
  52   %res = call i32 @bar(i32* %loc2)
  53   store i32 %res, i32* %loc
  54   br label %loop
  55 }
  56
  57 ; Negative test: %might clobber gep
  58 define void @test3(i32* %loc) {
  59 ; CHECK-LABEL: @test3
  60 ; CHECK-LABEL: loop:
  61 ; CHECK: @bar
  62 ; ALIAS-N2-LABEL: @test3
  63 ; ALIAS-N2-LABEL: loop:
  64 ; ALIAS-N2: @bar
  65   br label %loop
  66
  67 loop:
  68   %res = call i32 @bar(i32* %loc)
  69   %gep = getelementptr i32, i32 *%loc, i64 1000000
  70   store i32 %res, i32* %gep
  71   br label %loop
  72 }
  73
  74
  75 ; Negative test: %loc might alias %loc2
  76 define void @test4(i32* %loc, i32* %loc2) {
  77 ; CHECK-LABEL: @test4
  78 ; CHECK-LABEL: loop:
  79 ; CHECK: @bar
  80 ; ALIAS-N2-LABEL: @test4
  81 ; ALIAS-N2-LABEL: loop:
  82 ; ALIAS-N2: @bar
  83   br label %loop
  84
  85 loop:
  86   %res = call i32 @bar(i32* %loc2)
  87   store i32 %res, i32* %loc
  88   br label %loop
  89 }
  90
  91 declare i32 @foo_new(i32*) readonly
  92 ; With the default AST mechanism used by LICM for alias analysis,
  93 ; we clump foo_new with bar.
  94 ; With the N2 Alias analysis diagnostic tool, we are able to hoist the
  95 ; argmemonly bar call out of the loop.
  96 ; Using MemorySSA we can also hoist bar.
  97
  98 define void @test5(i32* %loc2, i32* noalias %loc) {
  99 ; ALIAS-N2-LABEL: @test5
 100 ; ALIAS-N2: @bar
 101 ; ALIAS-N2-LABEL: loop:
 102
 103 ; CHECK-LABEL: @test5
 104 ; CHECK-LABEL: loop:
 105 ; CHECK:  @bar
 106   br label %loop
 107
 108 loop:
 109   %res1 = call i32 @bar(i32* %loc2)
 110   %res = call i32 @foo_new(i32* %loc2)
 111   store volatile i32 %res1, i32* %loc
 112   br label %loop
 113 }
 114
 115
 116 ; memcpy doesn't write to it's source argument, so loads to that location
 117 ; can still be hoisted
 118 define void @test6(i32* noalias %loc, i32* noalias %loc2) {
 119 ; CHECK-LABEL: @test6
 120 ; CHECK: %val = load i32, i32* %loc2
 121 ; CHECK-LABEL: loop:
 122 ; CHECK: @llvm.memcpy
 123 ; ALIAS-N2-LABEL: @test6
 124 ; ALIAS-N2: %val = load i32, i32* %loc2
 125 ; ALIAS-N2-LABEL: loop:
 126 ; ALIAS-N2: @llvm.memcpy
 127   br label %loop
 128
 129 loop:
 130   %val = load i32, i32* %loc2
 131   store i32 %val, i32* %loc
 132   %dest = bitcast i32* %loc to i8*
 133   %src = bitcast i32* %loc2 to i8*
 134   call void @llvm.memcpy.p0i8.p0i8.i64(i8* %dest, i8* %src, i64 8, i1 false)
 135   br label %loop
 136 }
 137
 138 define void @test7(i32* noalias %loc, i32* noalias %loc2) {
 139 ; CHECK-LABEL: @test7
 140 ; CHECK: %val = load i32, i32* %loc2
 141 ; CHECK-LABEL: loop:
 142 ; CHECK: @custom_memcpy
 143 ; ALIAS-N2-LABEL: @test7
 144 ; ALIAS-N2: %val = load i32, i32* %loc2
 145 ; ALIAS-N2-LABEL: loop:
 146 ; ALIAS-N2: @custom_memcpy
 147   br label %loop
 148
 149 loop:
 150   %val = load i32, i32* %loc2
 151   store i32 %val, i32* %loc
 152   %dest = bitcast i32* %loc to i8*
 153   %src = bitcast i32* %loc2 to i8*
 154   call void @custom_memcpy(i8* %dest, i8* %src)
 155   br label %loop
 156 }
 157
 158 declare void @llvm.memcpy.p0i8.p0i8.i64(i8* nocapture writeonly, i8* nocapture readonly, i64, i1)
 159 declare void @custom_memcpy(i8* nocapture writeonly, i8* nocapture readonly) argmemonly nounwind