test/Transforms/FunctionAttrs/dereferenceable.ll

   1 ; RUN: opt -attributor -attributor-manifest-internal --attributor-disable=false -attributor-max-iterations-verify -attributor-max-iterations=2 -S < %s | FileCheck %s --check-prefix=ATTRIBUTOR
   2
   3
   4 declare void @deref_phi_user(i32* %a);
   5
   6 ; TEST 1
   7 ; take mininimum of return values
   8 ;
   9 define i32* @test1(i32* dereferenceable(4) %0, double* dereferenceable(8) %1, i1 zeroext %2) local_unnamed_addr {
  10 ; ATTRIBUTOR: define nonnull dereferenceable(4) i32* @test1(i32* nonnull readnone dereferenceable(4) "no-capture-maybe-returned" %0, double* nonnull readnone dereferenceable(8) "no-capture-maybe-returned" %1, i1 zeroext %2)
  11   %4 = bitcast double* %1 to i32*
  12   %5 = select i1 %2, i32* %0, i32* %4
  13   ret i32* %5
  14 }
  15
  16 ; TEST 2
  17 define i32* @test2(i32* dereferenceable_or_null(4) %0, double* dereferenceable(8) %1, i1 zeroext %2) local_unnamed_addr {
  18 ; ATTRIBUTOR: define dereferenceable_or_null(4) i32* @test2(i32* readnone dereferenceable_or_null(4) "no-capture-maybe-returned" %0, double* nonnull readnone dereferenceable(8) "no-capture-maybe-returned" %1, i1 zeroext %2)
  19   %4 = bitcast double* %1 to i32*
  20   %5 = select i1 %2, i32* %0, i32* %4
  21   ret i32* %5
  22 }
  23
  24 ; TEST 3
  25 ; GEP inbounds
  26 define i32* @test3_1(i32* dereferenceable(8) %0) local_unnamed_addr {
  27 ; ATTRIBUTOR: define nonnull dereferenceable(4) i32* @test3_1(i32* nonnull readnone dereferenceable(8) "no-capture-maybe-returned" %0)
  28   %ret = getelementptr inbounds i32, i32* %0, i64 1
  29   ret i32* %ret
  30 }
  31
  32 define i32* @test3_2(i32* dereferenceable_or_null(32) %0) local_unnamed_addr {
  33 ; FIXME: Argument should be mark dereferenceable because of GEP `inbounds`.
  34 ; ATTRIBUTOR: define nonnull dereferenceable(16) i32* @test3_2(i32* readnone dereferenceable_or_null(32) "no-capture-maybe-returned" %0)
  35   %ret = getelementptr inbounds i32, i32* %0, i64 4
  36   ret i32* %ret
  37 }
  38
  39 define i32* @test3_3(i32* dereferenceable(8) %0, i32* dereferenceable(16) %1, i1 %2) local_unnamed_addr {
  40 ; ATTRIBUTOR: define nonnull dereferenceable(4) i32* @test3_3(i32* nonnull readnone dereferenceable(8) "no-capture-maybe-returned" %0, i32* nonnull readnone dereferenceable(16) "no-capture-maybe-returned" %1, i1 %2) local_unnamed_addr
  41   %ret1 = getelementptr inbounds i32, i32* %0, i64 1
  42   %ret2 = getelementptr inbounds i32, i32* %1, i64 2
  43   %ret = select i1 %2, i32* %ret1, i32* %ret2
  44   ret i32* %ret
  45 }
  46
  47 ; TEST 4
  48 ; Better than known in IR.
  49
  50 define dereferenceable(4) i32* @test4(i32* dereferenceable(8) %0) local_unnamed_addr {
  51 ; ATTRIBUTOR: define nonnull dereferenceable(8) i32* @test4(i32* nonnull readnone returned dereferenceable(8) "no-capture-maybe-returned" %0)
  52   ret i32* %0
  53 }
  54
  55 ; TEST 5
  56 ; loop in which dereferenceabily "grows"
  57 define void @deref_phi_growing(i32* dereferenceable(4000) %a) {
  58 entry:
  59   br label %for.cond
  60
  61 for.cond:                                         ; preds = %for.inc, %entry
  62   %i.0 = phi i32 [ 0, %entry ], [ %inc, %for.inc ]
  63   %a.addr.0 = phi i32* [ %a, %entry ], [ %incdec.ptr, %for.inc ]
  64 ; ATTRIBUTOR: call void @deref_phi_user(i32* nonnull dereferenceable(4000) %a.addr.0)
  65   call void @deref_phi_user(i32* %a.addr.0)
  66   %tmp = load i32, i32* %a.addr.0, align 4
  67   %cmp = icmp slt i32 %i.0, %tmp
  68   br i1 %cmp, label %for.body, label %for.cond.cleanup
  69
  70 for.cond.cleanup:                                 ; preds = %for.cond
  71   br label %for.end
  72
  73 for.body:                                         ; preds = %for.cond
  74   br label %for.inc
  75
  76 for.inc:                                          ; preds = %for.body
  77   %incdec.ptr = getelementptr inbounds i32, i32* %a.addr.0, i64 -1
  78   %inc = add nuw nsw i32 %i.0, 1
  79   br label %for.cond
  80
  81 for.end:                                          ; preds = %for.cond.cleanup
  82   ret void
  83 }
  84
  85 ; TEST 6
  86 ; loop in which dereferenceabily "shrinks"
  87 define void @deref_phi_shrinking(i32* dereferenceable(4000) %a) {
  88 entry:
  89   br label %for.cond
  90
  91 for.cond:                                         ; preds = %for.inc, %entry
  92   %i.0 = phi i32 [ 0, %entry ], [ %inc, %for.inc ]
  93   %a.addr.0 = phi i32* [ %a, %entry ], [ %incdec.ptr, %for.inc ]
  94 ; ATTRIBUTOR: call void @deref_phi_user(i32* nonnull %a.addr.0)
  95   call void @deref_phi_user(i32* %a.addr.0)
  96   %tmp = load i32, i32* %a.addr.0, align 4
  97   %cmp = icmp slt i32 %i.0, %tmp
  98   br i1 %cmp, label %for.body, label %for.cond.cleanup
  99
 100 for.cond.cleanup:                                 ; preds = %for.cond
 101   br label %for.end
 102
 103 for.body:                                         ; preds = %for.cond
 104   br label %for.inc
 105
 106 for.inc:                                          ; preds = %for.body
 107   %incdec.ptr = getelementptr inbounds i32, i32* %a.addr.0, i64 1
 108   %inc = add nuw nsw i32 %i.0, 1
 109   br label %for.cond
 110
 111 for.end:                                          ; preds = %for.cond.cleanup
 112   ret void
 113 }
 114
 115 ; TEST 7
 116 ; share known infomation in must-be-executed-context
 117 declare i32* @unkown_ptr() willreturn nounwind
 118 declare i32 @unkown_f(i32*) willreturn nounwind
 119 define i32* @f7_0(i32* %ptr) {
 120 ; ATTRIBUTOR: define nonnull dereferenceable(8) i32* @f7_0(i32* nonnull returned dereferenceable(8) %ptr)
 121   %T = tail call i32 @unkown_f(i32* dereferenceable(8) %ptr)
 122   ret i32* %ptr
 123 }
 124
 125 ; ATTRIBUTOR: define void @f7_1(i32* nonnull dereferenceable(4) %ptr, i1 %c)
 126 define void @f7_1(i32* %ptr, i1 %c) {
 127
 128 ; ATTRIBUTOR:   %A = tail call i32 @unkown_f(i32* nonnull dereferenceable(4) %ptr)
 129   %A = tail call i32 @unkown_f(i32* %ptr)
 130
 131   %ptr.0 = load i32, i32* %ptr
 132   ; deref 4 hold
 133
 134 ; FIXME: this should be %B = tail call i32 @unkown_f(i32* nonnull dereferenceable(4) %ptr)
 135 ; ATTRIBUTOR:   %B = tail call i32 @unkown_f(i32* nonnull dereferenceable(4) %ptr)
 136   %B = tail call i32 @unkown_f(i32* dereferenceable(1) %ptr)
 137
 138   br i1%c, label %if.true, label %if.false
 139 if.true:
 140 ; ATTRIBUTOR:   %C = tail call i32 @unkown_f(i32* nonnull dereferenceable(8) %ptr)
 141   %C = tail call i32 @unkown_f(i32* %ptr)
 142
 143 ; ATTRIBUTOR:   %D = tail call i32 @unkown_f(i32* nonnull dereferenceable(8) %ptr)
 144   %D = tail call i32 @unkown_f(i32* dereferenceable(8) %ptr)
 145
 146 ; FIXME: This should be tail call i32 @unkown_f(i32* nonnull dereferenceable(8) %ptr)
 147 ;        Making must-be-executed-context backward exploration will fix this.
 148 ; ATTRIBUTOR:   %E = tail call i32 @unkown_f(i32* nonnull dereferenceable(4) %ptr)
 149   %E = tail call i32 @unkown_f(i32* %ptr)
 150
 151   ret void
 152
 153 if.false:
 154   ret void
 155 }
 156
 157 ; ATTRIBUTOR: define void @f7_2(i1 %c)
 158 define void @f7_2(i1 %c) {
 159
 160   %ptr =  tail call i32* @unkown_ptr()
 161
 162 ; ATTRIBUTOR:   %A = tail call i32 @unkown_f(i32* nonnull dereferenceable(4) %ptr)
 163   %A = tail call i32 @unkown_f(i32* %ptr)
 164
 165   %arg_a.0 = load i32, i32* %ptr
 166   ; deref 4 hold
 167
 168 ; ATTRIBUTOR:   %B = tail call i32 @unkown_f(i32* nonnull dereferenceable(4) %ptr)
 169   %B = tail call i32 @unkown_f(i32* dereferenceable(1) %ptr)
 170
 171   br i1%c, label %if.true, label %if.false
 172 if.true:
 173
 174 ; ATTRIBUTOR:   %C = tail call i32 @unkown_f(i32* nonnull dereferenceable(8) %ptr)
 175   %C = tail call i32 @unkown_f(i32* %ptr)
 176
 177 ; ATTRIBUTOR:   %D = tail call i32 @unkown_f(i32* nonnull dereferenceable(8) %ptr)
 178   %D = tail call i32 @unkown_f(i32* dereferenceable(8) %ptr)
 179
 180   %E = tail call i32 @unkown_f(i32* %ptr)
 181 ; FIXME: This should be @unkown_f(i32* nonnull dereferenceable(8) %ptr)
 182 ;        Making must-be-executed-context backward exploration will fix this.
 183 ; ATTRIBUTOR:   %E = tail call i32 @unkown_f(i32* nonnull dereferenceable(4) %ptr)
 184
 185   ret void
 186
 187 if.false:
 188   ret void
 189 }
 190
 191 define i32* @f7_3() {
 192 ; ATTRIBUTOR: define nonnull dereferenceable(4) i32* @f7_3()
 193   %ptr = tail call i32* @unkown_ptr()
 194   store i32 10, i32* %ptr, align 16
 195   ret i32* %ptr
 196 }
 197
 198 define i32* @test_for_minus_index(i32* %p) {
 199 ; FIXME: This should be define nonnull dereferenceable(8) i32* @test_for_minus_index(i32* nonnull %p)
 200 ; ATTRIBUTOR: define nonnull dereferenceable(8) i32* @test_for_minus_index(i32* writeonly "no-capture-maybe-returned" %p)
 201   %q = getelementptr inbounds i32, i32* %p, i32 -2
 202   store i32 1, i32* %q
 203   ret i32* %q
 204 }