test/Transforms/FunctionAttrs/dereferenceable.ll

   1 ; RUN: opt -attributor -attributor-manifest-internal --attributor-disable=false -attributor-max-iterations-verify -attributor-max-iterations=1 -S < %s | FileCheck %s --check-prefixes=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 dereferenceable(4) "no-capture-maybe-returned" %0, double* nonnull 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* dereferenceable_or_null(4) "no-capture-maybe-returned" %0, double* nonnull 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 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* 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 dereferenceable(8) "no-capture-maybe-returned" %0, i32* nonnull 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 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 ; CHECK: call void @deref_phi_user(i32* 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 ; CHECK: call void @deref_phi_user(i32* %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 }