204 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 42; RUN: opt < %s -passes=instcombine -S | FileCheck %s3;4; Test that instcombine folds allocsize function calls properly.5; Dummy arguments are inserted to verify that allocsize is picking the right6; args, and to prove that arbitrary unfoldable values don't interfere with7; allocsize if they're not used by allocsize.8 9declare ptr @my_malloc(ptr, i32) allocsize(1)10declare ptr @my_calloc(ptr, ptr, i32, i32) allocsize(2, 3)11 12define 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 817; CHECK-NEXT: store i64 100, ptr [[R]], align 818; CHECK-NEXT: ret void19;20 %1 = call ptr @my_malloc(ptr null, i32 100)21 store ptr %1, ptr %p, align 8 ; To ensure objectsize isn't killed22 23 %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false)24 store i64 %2, ptr %r, align 825 ret void26}27 28define 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 833; CHECK-NEXT: store i64 500, ptr [[R]], align 834; CHECK-NEXT: ret void35;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 killed38 39 %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false)40 store i64 %2, ptr %r, align 841 ret void42}43 44; Failure cases with non-constant values...45define 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 850; 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 852; CHECK-NEXT: ret void53;54 %1 = call ptr @my_malloc(ptr null, i32 %n)55 store ptr %1, ptr %p, align 8 ; To ensure objectsize isn't killed56 57 %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false)58 store i64 %2, ptr %r, align 859 ret void60}61 62define 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 867; 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 869; CHECK-NEXT: [[TMP3:%.*]] = call ptr @my_calloc(ptr null, ptr null, i32 100, i32 [[N]])70; CHECK-NEXT: store ptr [[TMP3]], ptr [[P]], align 871; 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 873; CHECK-NEXT: ret void74;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 killed77 78 %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false)79 store i64 %2, ptr %r, align 880 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 killed84 85 %4 = call i64 @llvm.objectsize.i64.p0(ptr %3, i1 false)86 store i64 %4, ptr %r, align 887 ret void88}89 90declare ptr @my_malloc_outofline(ptr, i32) #091declare ptr @my_calloc_outofline(ptr, ptr, i32, i32) #192 93; Verifying that out of line allocsize is parsed correctly94define 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 899; CHECK-NEXT: store i64 100, ptr [[R]], align 8100; 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 8102; CHECK-NEXT: store i64 500, ptr [[R]], align 8103; CHECK-NEXT: ret void104;105 %1 = call ptr @my_malloc_outofline(ptr null, i32 100)106 store ptr %1, ptr %p, align 8 ; To ensure objectsize isn't killed107 108 %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false)109 store i64 %2, ptr %r, align 8110 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 killed114 115 %4 = call i64 @llvm.objectsize.i64.p0(ptr %3, i1 false)116 store i64 %4, ptr %r, align 8117 ret void118}119 120declare ptr @my_malloc_i64(ptr, i64) #0121declare ptr @my_tiny_calloc(ptr, ptr, i8, i8) #1122declare ptr @my_varied_calloc(ptr, ptr, i32, i8) #1123 124define 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 8130; 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 4132; 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 8134; CHECK-NEXT: store i32 508, ptr [[R]], align 4135; 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 8137; 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 4139; CHECK-NEXT: store i64 8589934592, ptr [[R]], align 8140; 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 8142; CHECK-NEXT: store i32 5000, ptr [[R]], align 4143; CHECK-NEXT: ret void144;145 %big_malloc = call ptr @my_calloc(ptr null, ptr null, i32 2147483649, i32 2)146 store ptr %big_malloc, ptr %p, align 8147 148 %1 = call i32 @llvm.objectsize.i32.p0(ptr %big_malloc, i1 false)149 store i32 %1, ptr %r, align 4150 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 8154 155 %2 = call i32 @llvm.objectsize.i32.p0(ptr %big_little_malloc, i1 false)156 store i32 %2, ptr %r, align 4157 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 8162 163 %3 = call i32 @llvm.objectsize.i32.p0(ptr %big_malloc_i64, i1 false)164 store i32 %3, ptr %r, align 4165 166 167 %4 = call i64 @llvm.objectsize.i64.p0(ptr %big_malloc_i64, i1 false)168 store i64 %4, ptr %r, align 8169 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 8174 175 %5 = call i32 @llvm.objectsize.i32.p0(ptr %varied_calloc, i1 false)176 store i32 %5, ptr %r, align 4177 178 ret void179}180 181; We had a bug where `nobuiltin` would cause `allocsize` to be ignored in182; @llvm.objectsize calculations.183define 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 8188; CHECK-NEXT: store i64 100, ptr [[R]], align 8189; CHECK-NEXT: ret void190;191 %1 = call ptr @my_malloc(ptr null, i32 100) nobuiltin192 store ptr %1, ptr %p, align 8193 194 %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false)195 store i64 %2, ptr %r, align 8196 ret void197}198 199attributes #0 = { allocsize(1) }200attributes #1 = { allocsize(2, 3) }201 202declare i32 @llvm.objectsize.i32.p0(ptr, i1)203declare i64 @llvm.objectsize.i64.p0(ptr, i1)204