llvm/test/Transforms/InstCombine/allocsize.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 4
   2 ; RUN: opt < %s -passes=instcombine -S | FileCheck %s
   3 ;
   4 ; Test that instcombine folds allocsize function calls properly.
   5 ; Dummy arguments are inserted to verify that allocsize is picking the right
   6 ; args, and to prove that arbitrary unfoldable values don't interfere with
   7 ; allocsize if they're not used by allocsize.
   8
   9 declare ptr @my_malloc(ptr, i32) allocsize(1)
  10 declare ptr @my_calloc(ptr, ptr, i32, i32) allocsize(2, 3)
  11
  12 define void @test_malloc(ptr %p, ptr %r) {
  13 ; CHECK-LABEL: define void @test_malloc(
  14 ; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]]) {
  15 ; CHECK-NEXT:    [[TMP1:%.*]] = call dereferenceable_or_null(100) ptr @my_malloc(ptr null, i32 100)
  16 ; CHECK-NEXT:    store ptr [[TMP1]], ptr [[P]], align 8
  17 ; CHECK-NEXT:    store i64 100, ptr [[R]], align 8
  18 ; CHECK-NEXT:    ret void
  19 ;
  20   %1 = call ptr @my_malloc(ptr null, i32 100)
  21   store ptr %1, ptr %p, align 8 ; To ensure objectsize isn't killed
  22
  23   %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false)
  24   store i64 %2, ptr %r, align 8
  25   ret void
  26 }
  27
  28 define void @test_calloc(ptr %p, ptr %r) {
  29 ; CHECK-LABEL: define void @test_calloc(
  30 ; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]]) {
  31 ; CHECK-NEXT:    [[TMP1:%.*]] = call dereferenceable_or_null(500) ptr @my_calloc(ptr null, ptr null, i32 100, i32 5)
  32 ; CHECK-NEXT:    store ptr [[TMP1]], ptr [[P]], align 8
  33 ; CHECK-NEXT:    store i64 500, ptr [[R]], align 8
  34 ; CHECK-NEXT:    ret void
  35 ;
  36   %1 = call ptr @my_calloc(ptr null, ptr null, i32 100, i32 5)
  37   store ptr %1, ptr %p, align 8 ; To ensure objectsize isn't killed
  38
  39   %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false)
  40   store i64 %2, ptr %r, align 8
  41   ret void
  42 }
  43
  44 ; Failure cases with non-constant values...
  45 define void @test_malloc_fails(ptr %p, ptr %r, i32 %n) {
  46 ; CHECK-LABEL: define void @test_malloc_fails(
  47 ; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]], i32 [[N:%.*]]) {
  48 ; CHECK-NEXT:    [[TMP1:%.*]] = call ptr @my_malloc(ptr null, i32 [[N]])
  49 ; CHECK-NEXT:    store ptr [[TMP1]], ptr [[P]], align 8
  50 ; CHECK-NEXT:    [[TMP2:%.*]] = call i64 @llvm.objectsize.i64.p0(ptr [[TMP1]], i1 false, i1 false, i1 false)
  51 ; CHECK-NEXT:    store i64 [[TMP2]], ptr [[R]], align 8
  52 ; CHECK-NEXT:    ret void
  53 ;
  54   %1 = call ptr @my_malloc(ptr null, i32 %n)
  55   store ptr %1, ptr %p, align 8 ; To ensure objectsize isn't killed
  56
  57   %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false)
  58   store i64 %2, ptr %r, align 8
  59   ret void
  60 }
  61
  62 define void @test_calloc_fails(ptr %p, ptr %r, i32 %n) {
  63 ; CHECK-LABEL: define void @test_calloc_fails(
  64 ; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]], i32 [[N:%.*]]) {
  65 ; CHECK-NEXT:    [[TMP1:%.*]] = call ptr @my_calloc(ptr null, ptr null, i32 [[N]], i32 5)
  66 ; CHECK-NEXT:    store ptr [[TMP1]], ptr [[P]], align 8
  67 ; CHECK-NEXT:    [[TMP2:%.*]] = call i64 @llvm.objectsize.i64.p0(ptr [[TMP1]], i1 false, i1 false, i1 false)
  68 ; CHECK-NEXT:    store i64 [[TMP2]], ptr [[R]], align 8
  69 ; CHECK-NEXT:    [[TMP3:%.*]] = call ptr @my_calloc(ptr null, ptr null, i32 100, i32 [[N]])
  70 ; CHECK-NEXT:    store ptr [[TMP3]], ptr [[P]], align 8
  71 ; CHECK-NEXT:    [[TMP4:%.*]] = call i64 @llvm.objectsize.i64.p0(ptr [[TMP3]], i1 false, i1 false, i1 false)
  72 ; CHECK-NEXT:    store i64 [[TMP4]], ptr [[R]], align 8
  73 ; CHECK-NEXT:    ret void
  74 ;
  75   %1 = call ptr @my_calloc(ptr null, ptr null, i32 %n, i32 5)
  76   store ptr %1, ptr %p, align 8 ; To ensure objectsize isn't killed
  77
  78   %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false)
  79   store i64 %2, ptr %r, align 8
  80
  81
  82   %3 = call ptr @my_calloc(ptr null, ptr null, i32 100, i32 %n)
  83   store ptr %3, ptr %p, align 8 ; To ensure objectsize isn't killed
  84
  85   %4 = call i64 @llvm.objectsize.i64.p0(ptr %3, i1 false)
  86   store i64 %4, ptr %r, align 8
  87   ret void
  88 }
  89
  90 declare ptr @my_malloc_outofline(ptr, i32) #0
  91 declare ptr @my_calloc_outofline(ptr, ptr, i32, i32) #1
  92
  93 ; Verifying that out of line allocsize is parsed correctly
  94 define void @test_outofline(ptr %p, ptr %r) {
  95 ; CHECK-LABEL: define void @test_outofline(
  96 ; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]]) {
  97 ; CHECK-NEXT:    [[TMP1:%.*]] = call dereferenceable_or_null(100) ptr @my_malloc_outofline(ptr null, i32 100)
  98 ; CHECK-NEXT:    store ptr [[TMP1]], ptr [[P]], align 8
  99 ; CHECK-NEXT:    store i64 100, ptr [[R]], align 8
 100 ; CHECK-NEXT:    [[TMP2:%.*]] = call dereferenceable_or_null(500) ptr @my_calloc_outofline(ptr null, ptr null, i32 100, i32 5)
 101 ; CHECK-NEXT:    store ptr [[TMP2]], ptr [[P]], align 8
 102 ; CHECK-NEXT:    store i64 500, ptr [[R]], align 8
 103 ; CHECK-NEXT:    ret void
 104 ;
 105   %1 = call ptr @my_malloc_outofline(ptr null, i32 100)
 106   store ptr %1, ptr %p, align 8 ; To ensure objectsize isn't killed
 107
 108   %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false)
 109   store i64 %2, ptr %r, align 8
 110
 111
 112   %3 = call ptr @my_calloc_outofline(ptr null, ptr null, i32 100, i32 5)
 113   store ptr %3, ptr %p, align 8 ; To ensure objectsize isn't killed
 114
 115   %4 = call i64 @llvm.objectsize.i64.p0(ptr %3, i1 false)
 116   store i64 %4, ptr %r, align 8
 117   ret void
 118 }
 119
 120 declare ptr @my_malloc_i64(ptr, i64) #0
 121 declare ptr @my_tiny_calloc(ptr, ptr, i8, i8) #1
 122 declare ptr @my_varied_calloc(ptr, ptr, i32, i8) #1
 123
 124 define void @test_overflow(ptr %p, ptr %r) {
 125   ; (2**31 + 1) * 2 > 2**31. So overflow. Yay.
 126 ; CHECK-LABEL: define void @test_overflow(
 127 ; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]]) {
 128 ; CHECK-NEXT:    [[BIG_MALLOC:%.*]] = call dereferenceable_or_null(4294967298) ptr @my_calloc(ptr null, ptr null, i32 -2147483647, i32 2)
 129 ; CHECK-NEXT:    store ptr [[BIG_MALLOC]], ptr [[P]], align 8
 130 ; CHECK-NEXT:    [[TMP1:%.*]] = call i32 @llvm.objectsize.i32.p0(ptr [[BIG_MALLOC]], i1 false, i1 false, i1 false)
 131 ; CHECK-NEXT:    store i32 [[TMP1]], ptr [[R]], align 4
 132 ; CHECK-NEXT:    [[BIG_LITTLE_MALLOC:%.*]] = call dereferenceable_or_null(508) ptr @my_tiny_calloc(ptr null, ptr null, i8 127, i8 4)
 133 ; CHECK-NEXT:    store ptr [[BIG_LITTLE_MALLOC]], ptr [[P]], align 8
 134 ; CHECK-NEXT:    store i32 508, ptr [[R]], align 4
 135 ; CHECK-NEXT:    [[BIG_MALLOC_I64:%.*]] = call dereferenceable_or_null(8589934592) ptr @my_malloc_i64(ptr null, i64 8589934592)
 136 ; CHECK-NEXT:    store ptr [[BIG_MALLOC_I64]], ptr [[P]], align 8
 137 ; CHECK-NEXT:    [[TMP2:%.*]] = call i32 @llvm.objectsize.i32.p0(ptr [[BIG_MALLOC_I64]], i1 false, i1 false, i1 false)
 138 ; CHECK-NEXT:    store i32 [[TMP2]], ptr [[R]], align 4
 139 ; CHECK-NEXT:    store i64 8589934592, ptr [[R]], align 8
 140 ; CHECK-NEXT:    [[VARIED_CALLOC:%.*]] = call dereferenceable_or_null(5000) ptr @my_varied_calloc(ptr null, ptr null, i32 1000, i8 5)
 141 ; CHECK-NEXT:    store ptr [[VARIED_CALLOC]], ptr [[P]], align 8
 142 ; CHECK-NEXT:    store i32 5000, ptr [[R]], align 4
 143 ; CHECK-NEXT:    ret void
 144 ;
 145   %big_malloc = call ptr @my_calloc(ptr null, ptr null, i32 2147483649, i32 2)
 146   store ptr %big_malloc, ptr %p, align 8
 147
 148   %1 = call i32 @llvm.objectsize.i32.p0(ptr %big_malloc, i1 false)
 149   store i32 %1, ptr %r, align 4
 150
 151
 152   %big_little_malloc = call ptr @my_tiny_calloc(ptr null, ptr null, i8 127, i8 4)
 153   store ptr %big_little_malloc, ptr %p, align 8
 154
 155   %2 = call i32 @llvm.objectsize.i32.p0(ptr %big_little_malloc, i1 false)
 156   store i32 %2, ptr %r, align 4
 157
 158
 159   ; malloc(2**33)
 160   %big_malloc_i64 = call ptr @my_malloc_i64(ptr null, i64 8589934592)
 161   store ptr %big_malloc_i64, ptr %p, align 8
 162
 163   %3 = call i32 @llvm.objectsize.i32.p0(ptr %big_malloc_i64, i1 false)
 164   store i32 %3, ptr %r, align 4
 165
 166
 167   %4 = call i64 @llvm.objectsize.i64.p0(ptr %big_malloc_i64, i1 false)
 168   store i64 %4, ptr %r, align 8
 169
 170
 171   ; Just intended to ensure that we properly handle args of different types...
 172   %varied_calloc = call ptr @my_varied_calloc(ptr null, ptr null, i32 1000, i8 5)
 173   store ptr %varied_calloc, ptr %p, align 8
 174
 175   %5 = call i32 @llvm.objectsize.i32.p0(ptr %varied_calloc, i1 false)
 176   store i32 %5, ptr %r, align 4
 177
 178   ret void
 179 }
 180
 181 ; We had a bug where `nobuiltin` would cause `allocsize` to be ignored in
 182 ; @llvm.objectsize calculations.
 183 define void @test_nobuiltin(ptr %p, ptr %r) {
 184 ; CHECK-LABEL: define void @test_nobuiltin(
 185 ; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]]) {
 186 ; CHECK-NEXT:    [[TMP1:%.*]] = call dereferenceable_or_null(100) ptr @my_malloc(ptr null, i32 100) #[[ATTR3:[0-9]+]]
 187 ; CHECK-NEXT:    store ptr [[TMP1]], ptr [[P]], align 8
 188 ; CHECK-NEXT:    store i64 100, ptr [[R]], align 8
 189 ; CHECK-NEXT:    ret void
 190 ;
 191   %1 = call ptr @my_malloc(ptr null, i32 100) nobuiltin
 192   store ptr %1, ptr %p, align 8
 193
 194   %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false)
 195   store i64 %2, ptr %r, align 8
 196   ret void
 197 }
 198
 199 attributes #0 = { allocsize(1) }
 200 attributes #1 = { allocsize(2, 3) }
 201
 202 declare i32 @llvm.objectsize.i32.p0(ptr, i1)
 203 declare i64 @llvm.objectsize.i64.p0(ptr, i1)