221 lines · c
1// NOTE: Assertions have been autogenerated by utils/update_cc_test_checks.py UTC_ARGS: --version 52// RUN: %clang_cc1 -std=c2y -triple x86_64-unknown-unknown -emit-llvm -o - %s | FileCheck %s --check-prefix=CHECK-643// RUN: %clang_cc1 -std=c2y -triple aarch64-unknown-unknown -emit-llvm -o - %s | FileCheck %s --check-prefix=CHECK-644// RUN: %clang_cc1 -std=c2y -triple ppc64le-unknown-unknown -emit-llvm -o - %s | FileCheck %s --check-prefix=CHECK-645// RUN: %clang_cc1 -std=c2y -triple i686-unknown-unknown -emit-llvm -o - %s | FileCheck %s --check-prefix=CHECK-326 7// This tests the codegen behavior for VLA use in _Countof, which builds upon8// the code generation for sizeof. It tests a limited set of triples due to9// variations in VLA handling between platforms, but this test should10// generalize to all targets.11 12typedef typeof(sizeof(0)) size_t;13 14// CHECK-64-LABEL: define dso_local i64 @test1(15// CHECK-64-SAME: i32 noundef{{( signext)?}} [[N:%.*]]) #[[ATTR0:[0-9]+]] {16// CHECK-64-NEXT: [[ENTRY:.*:]]17// CHECK-64-NEXT: [[N_ADDR:%.*]] = alloca i32, align18// CHECK-64-NEXT: [[SAVED_STACK:%.*]] = alloca ptr, align19// CHECK-64-NEXT: [[__VLA_EXPR0:%.*]] = alloca i64, align20// CHECK-64-NEXT: store i32 [[N]], ptr [[N_ADDR]], align21// CHECK-64-NEXT: [[TMP0:%.*]] = load i32, ptr [[N_ADDR]], align22// CHECK-64-NEXT: [[TMP1:%.*]] = zext i32 [[TMP0]] to i6423// CHECK-64-NEXT: [[TMP2:%.*]] = call ptr @llvm.stacksave.p0()24// CHECK-64-NEXT: store ptr [[TMP2]], ptr [[SAVED_STACK]], align25// CHECK-64-NEXT: [[VLA:%.*]] = alloca i32, i64 [[TMP1]], align26// CHECK-64-NEXT: store i64 [[TMP1]], ptr [[__VLA_EXPR0]], align27// CHECK-64-NEXT: [[TMP3:%.*]] = load ptr, ptr [[SAVED_STACK]], align28// CHECK-64-NEXT: call void @llvm.stackrestore.p0(ptr [[TMP3]])29// CHECK-64-NEXT: ret i64 [[TMP1]]30//31// CHECK-32-LABEL: define dso_local i32 @test1(32// CHECK-32-SAME: i32 noundef [[N:%.*]]) #[[ATTR0:[0-9]+]] {33// CHECK-32-NEXT: [[ENTRY:.*:]]34// CHECK-32-NEXT: [[N_ADDR:%.*]] = alloca i32, align35// CHECK-32-NEXT: [[SAVED_STACK:%.*]] = alloca ptr, align36// CHECK-32-NEXT: [[__VLA_EXPR0:%.*]] = alloca i32, align37// CHECK-32-NEXT: store i32 [[N]], ptr [[N_ADDR]], align38// CHECK-32-NEXT: [[TMP0:%.*]] = load i32, ptr [[N_ADDR]], align39// CHECK-32-NEXT: [[TMP1:%.*]] = call ptr @llvm.stacksave.p0()40// CHECK-32-NEXT: store ptr [[TMP1]], ptr [[SAVED_STACK]], align41// CHECK-32-NEXT: [[VLA:%.*]] = alloca i32, i32 [[TMP0]], align42// CHECK-32-NEXT: store i32 [[TMP0]], ptr [[__VLA_EXPR0]], align43// CHECK-32-NEXT: [[TMP2:%.*]] = load ptr, ptr [[SAVED_STACK]], align44// CHECK-32-NEXT: call void @llvm.stackrestore.p0(ptr [[TMP2]])45// CHECK-32-NEXT: ret i32 [[TMP0]]46//47size_t test1(int n) {48 int array[n];49 return _Countof(array);50}51 52// CHECK-64-LABEL: define dso_local i64 @test2(53// CHECK-64-SAME: i32 noundef{{( signext)?}} [[N:%.*]]) #[[ATTR0]] {54// CHECK-64-NEXT: [[ENTRY:.*:]]55// CHECK-64-NEXT: [[N_ADDR:%.*]] = alloca i32, align56// CHECK-64-NEXT: store i32 [[N]], ptr [[N_ADDR]], align57// CHECK-64-NEXT: [[TMP0:%.*]] = load i32, ptr [[N_ADDR]], align58// CHECK-64-NEXT: [[TMP1:%.*]] = zext i32 [[TMP0]] to i6459// CHECK-64-NEXT: ret i64 [[TMP1]]60//61// CHECK-32-LABEL: define dso_local i32 @test2(62// CHECK-32-SAME: i32 noundef [[N:%.*]]) #[[ATTR0]] {63// CHECK-32-NEXT: [[ENTRY:.*:]]64// CHECK-32-NEXT: [[N_ADDR:%.*]] = alloca i32, align65// CHECK-32-NEXT: store i32 [[N]], ptr [[N_ADDR]], align66// CHECK-32-NEXT: [[TMP0:%.*]] = load i32, ptr [[N_ADDR]], align67// CHECK-32-NEXT: ret i32 [[TMP0]]68//69size_t test2(int n) {70 return _Countof(int[n]);71}72 73// CHECK-64-LABEL: define dso_local i64 @test3(74// CHECK-64-SAME: i32 noundef{{( signext)?}} [[N:%.*]]) #[[ATTR0]] {75// CHECK-64-NEXT: [[ENTRY:.*:]]76// CHECK-64-NEXT: [[N_ADDR:%.*]] = alloca i32, align77// CHECK-64-NEXT: [[SAVED_STACK:%.*]] = alloca ptr, align78// CHECK-64-NEXT: [[__VLA_EXPR0:%.*]] = alloca i64, align79// CHECK-64-NEXT: [[X:%.*]] = alloca i32, align80// CHECK-64-NEXT: [[Y:%.*]] = alloca i32, align81// CHECK-64-NEXT: store i32 [[N]], ptr [[N_ADDR]], align82// CHECK-64-NEXT: [[TMP0:%.*]] = load i32, ptr [[N_ADDR]], align83// CHECK-64-NEXT: [[TMP1:%.*]] = zext i32 [[TMP0]] to i6484// CHECK-64-NEXT: [[TMP2:%.*]] = call ptr @llvm.stacksave.p0()85// CHECK-64-NEXT: store ptr [[TMP2]], ptr [[SAVED_STACK]], align86// CHECK-64-NEXT: [[VLA:%.*]] = alloca [7 x i32], i64 [[TMP1]], align87// CHECK-64-NEXT: store i64 [[TMP1]], ptr [[__VLA_EXPR0]], align88// CHECK-64-NEXT: [[CONV:%.*]] = trunc i64 [[TMP1]] to i3289// CHECK-64-NEXT: store i32 [[CONV]], ptr [[X]], align90// CHECK-64-NEXT: store i32 7, ptr [[Y]], align91// CHECK-64-NEXT: [[TMP3:%.*]] = load i32, ptr [[X]], align92// CHECK-64-NEXT: [[TMP4:%.*]] = load i32, ptr [[Y]], align93// CHECK-64-NEXT: [[ADD:%.*]] = add nsw i32 [[TMP3]], [[TMP4]]94// CHECK-64-NEXT: [[CONV1:%.*]] = sext i32 [[ADD]] to i6495// CHECK-64-NEXT: [[TMP5:%.*]] = load ptr, ptr [[SAVED_STACK]], align96// CHECK-64-NEXT: call void @llvm.stackrestore.p0(ptr [[TMP5]])97// CHECK-64-NEXT: ret i64 [[CONV1]]98//99// CHECK-32-LABEL: define dso_local i32 @test3(100// CHECK-32-SAME: i32 noundef [[N:%.*]]) #[[ATTR0]] {101// CHECK-32-NEXT: [[ENTRY:.*:]]102// CHECK-32-NEXT: [[N_ADDR:%.*]] = alloca i32, align103// CHECK-32-NEXT: [[SAVED_STACK:%.*]] = alloca ptr, align104// CHECK-32-NEXT: [[__VLA_EXPR0:%.*]] = alloca i32, align105// CHECK-32-NEXT: [[X:%.*]] = alloca i32, align106// CHECK-32-NEXT: [[Y:%.*]] = alloca i32, align107// CHECK-32-NEXT: store i32 [[N]], ptr [[N_ADDR]], align108// CHECK-32-NEXT: [[TMP0:%.*]] = load i32, ptr [[N_ADDR]], align109// CHECK-32-NEXT: [[TMP1:%.*]] = call ptr @llvm.stacksave.p0()110// CHECK-32-NEXT: store ptr [[TMP1]], ptr [[SAVED_STACK]], align111// CHECK-32-NEXT: [[VLA:%.*]] = alloca [7 x i32], i32 [[TMP0]], align112// CHECK-32-NEXT: store i32 [[TMP0]], ptr [[__VLA_EXPR0]], align113// CHECK-32-NEXT: store i32 [[TMP0]], ptr [[X]], align114// CHECK-32-NEXT: store i32 7, ptr [[Y]], align115// CHECK-32-NEXT: [[TMP2:%.*]] = load i32, ptr [[X]], align116// CHECK-32-NEXT: [[TMP3:%.*]] = load i32, ptr [[Y]], align117// CHECK-32-NEXT: [[ADD:%.*]] = add nsw i32 [[TMP2]], [[TMP3]]118// CHECK-32-NEXT: [[TMP4:%.*]] = load ptr, ptr [[SAVED_STACK]], align119// CHECK-32-NEXT: call void @llvm.stackrestore.p0(ptr [[TMP4]])120// CHECK-32-NEXT: ret i32 [[ADD]]121//122size_t test3(int n) {123 int array[n][7];124 int x = _Countof(array);125 int y = _Countof(*array);126 return x + y;127}128 129// CHECK-64-LABEL: define dso_local void @test4(130// CHECK-64-SAME: i32 noundef{{( signext)?}} [[N:%.*]]) #[[ATTR0]] {131// CHECK-64-NEXT: [[ENTRY:.*:]]132// CHECK-64-NEXT: [[N_ADDR:%.*]] = alloca i32, align133// CHECK-64-NEXT: [[X:%.*]] = alloca i32, align134// CHECK-64-NEXT: [[Y:%.*]] = alloca i32, align135// CHECK-64-NEXT: store i32 [[N]], ptr [[N_ADDR]], align136// CHECK-64-NEXT: store i32 7, ptr [[X]], align137// CHECK-64-NEXT: [[TMP0:%.*]] = load i32, ptr [[N_ADDR]], align138// CHECK-64-NEXT: [[INC:%.*]] = add nsw i32 [[TMP0]], 1139// CHECK-64-NEXT: store i32 [[INC]], ptr [[N_ADDR]], align140// CHECK-64-NEXT: [[TMP1:%.*]] = zext i32 [[TMP0]] to i64141// CHECK-64-NEXT: [[CONV:%.*]] = trunc i64 [[TMP1]] to i32142// CHECK-64-NEXT: store i32 [[CONV]], ptr [[Y]], align143// CHECK-64-NEXT: ret void144//145// CHECK-32-LABEL: define dso_local void @test4(146// CHECK-32-SAME: i32 noundef [[N:%.*]]) #[[ATTR0]] {147// CHECK-32-NEXT: [[ENTRY:.*:]]148// CHECK-32-NEXT: [[N_ADDR:%.*]] = alloca i32, align149// CHECK-32-NEXT: [[X:%.*]] = alloca i32, align150// CHECK-32-NEXT: [[Y:%.*]] = alloca i32, align151// CHECK-32-NEXT: store i32 [[N]], ptr [[N_ADDR]], align152// CHECK-32-NEXT: store i32 7, ptr [[X]], align153// CHECK-32-NEXT: [[TMP0:%.*]] = load i32, ptr [[N_ADDR]], align154// CHECK-32-NEXT: [[INC:%.*]] = add nsw i32 [[TMP0]], 1155// CHECK-32-NEXT: store i32 [[INC]], ptr [[N_ADDR]], align156// CHECK-32-NEXT: store i32 [[TMP0]], ptr [[Y]], align157// CHECK-32-NEXT: ret void158//159void test4(int n) {160 // n should not be evaluated in this case because the operator does not need161 // to evaluate it to know the result is 7.162 int x = _Countof(int[7][n++]);163 // n should be evaluated in this case, however.164 int y = _Countof(int[n++][7]);165}166 167// CHECK-64-LABEL: define dso_local void @test5(168// CHECK-64-SAME: ) #[[ATTR0]] {169// CHECK-64-NEXT: [[ENTRY:.*:]]170// CHECK-64-NEXT: [[J:%.*]] = alloca i32, align171// CHECK-64-NEXT: [[SAVED_STACK:%.*]] = alloca ptr, align172// CHECK-64-NEXT: [[A:%.*]] = alloca i32, align173// CHECK-64-NEXT: [[B:%.*]] = alloca i32, align174// CHECK-64-NEXT: [[C:%.*]] = alloca i32, align175// CHECK-64-NEXT: [[D:%.*]] = alloca i32, align176// CHECK-64-NEXT: store i32 2, ptr [[J]], align177// CHECK-64-NEXT: [[TMP0:%.*]] = call ptr @llvm.stacksave.p0()178// CHECK-64-NEXT: store ptr [[TMP0]], ptr [[SAVED_STACK]], align179// CHECK-64-NEXT: [[VLA:%.*]] = alloca i32, i64 120, align180// CHECK-64-NEXT: store i32 1, ptr [[A]], align181// CHECK-64-NEXT: store i32 4, ptr [[B]], align182// CHECK-64-NEXT: store i32 5, ptr [[C]], align183// CHECK-64-NEXT: store i32 6, ptr [[D]], align184// CHECK-64-NEXT: [[TMP1:%.*]] = load ptr, ptr [[SAVED_STACK]], align185// CHECK-64-NEXT: call void @llvm.stackrestore.p0(ptr [[TMP1]])186// CHECK-64-NEXT: ret void187//188// CHECK-32-LABEL: define dso_local void @test5(189// CHECK-32-SAME: ) #[[ATTR0]] {190// CHECK-32-NEXT: [[ENTRY:.*:]]191// CHECK-32-NEXT: [[J:%.*]] = alloca i32, align192// CHECK-32-NEXT: [[SAVED_STACK:%.*]] = alloca ptr, align193// CHECK-32-NEXT: [[A:%.*]] = alloca i32, align194// CHECK-32-NEXT: [[B:%.*]] = alloca i32, align195// CHECK-32-NEXT: [[C:%.*]] = alloca i32, align196// CHECK-32-NEXT: [[D:%.*]] = alloca i32, align197// CHECK-32-NEXT: store i32 2, ptr [[J]], align198// CHECK-32-NEXT: [[TMP0:%.*]] = call ptr @llvm.stacksave.p0()199// CHECK-32-NEXT: store ptr [[TMP0]], ptr [[SAVED_STACK]], align200// CHECK-32-NEXT: [[VLA:%.*]] = alloca i32, i32 120, align201// CHECK-32-NEXT: store i32 1, ptr [[A]], align202// CHECK-32-NEXT: store i32 4, ptr [[B]], align203// CHECK-32-NEXT: store i32 5, ptr [[C]], align204// CHECK-32-NEXT: store i32 6, ptr [[D]], align205// CHECK-32-NEXT: [[TMP1:%.*]] = load ptr, ptr [[SAVED_STACK]], align206// CHECK-32-NEXT: call void @llvm.stackrestore.p0(ptr [[TMP1]])207// CHECK-32-NEXT: ret void208//209void test5() {210 // Ensure we're getting the variable-length dimensions correctly. We were211 // previously returning the size of all VLA dimensions multiplied together212 // to get the total element count rather than the element count for a single213 // array rank. See GH141409214 const int j = 2;215 int arr[1][j + 2][j + 3][j + 4];216 int a = _Countof arr;217 int b = _Countof *arr;218 int c = _Countof **arr;219 int d = _Countof ***arr;220}221