Home | History | Annotate | Download | only in CodeGen
      1 // RUN: %clang_cc1 -fblocks -emit-llvm %s -o %t
      2 // RUN: FileCheck %s -input-file=%t -check-prefix=CHECK1
      3 // RUN: FileCheck %s -input-file=%t -check-prefix=CHECK2
      4 
      5 struct A {
      6   int a;
      7   float b;
      8   char c;
      9 };
     10 
     11 void test_nest_captured_stmt(int param, int size, int param_arr[size]) {
     12   int w;
     13   int arr[param][size];
     14   // CHECK1: %struct.anon{{.*}} = type { [[INT:i.+]]*, [[INT]]*, [[SIZE_TYPE:i.+]], [[INT]]**, [[INT]]*, [[SIZE_TYPE]], [[SIZE_TYPE]], [[INT]]* }
     15   // CHECK1: %struct.anon{{.*}} = type { [[INT]]*, [[INT]]*, [[INT]]**, [[INT]]*, [[SIZE_TYPE]], [[INT]]**, [[INT]]*, [[SIZE_TYPE]], [[SIZE_TYPE]], [[INT]]* }
     16   // CHECK1: [[T:%struct.anon.*]] = type { [[INT]]*, [[INT]]*, %struct.A*, [[INT]]**, [[INT]]*, [[SIZE_TYPE]], [[INT]]**, [[INT]]*, [[SIZE_TYPE]], [[SIZE_TYPE]], [[INT]]* }
     17   #pragma clang __debug captured
     18   {
     19     int x;
     20     int *y = &w;
     21     #pragma clang __debug captured
     22     {
     23       struct A z;
     24       #pragma clang __debug captured
     25       {
     26         w = x = z.a = 1;
     27         *y = param;
     28         z.b = 0.1f;
     29         z.c = 'c';
     30         param_arr[size - 1] = 2;
     31         arr[10][z.a] = 12;
     32 
     33         // CHECK1: define internal {{.*}}void @__captured_stmt{{.*}}([[T]]
     34         // CHECK1: [[PARAM_ARR_SIZE_REF:%.+]] = getelementptr inbounds [[T]], [[T]]* {{.+}}, i{{[0-9]+}} 0, i{{[0-9]+}} 5
     35         // CHECK1: [[PARAM_ARR_SIZE:%.+]] = load [[SIZE_TYPE]], [[SIZE_TYPE]]* [[PARAM_ARR_SIZE_REF]]
     36         // CHECK1: [[ARR_SIZE1_REF:%.+]] = getelementptr inbounds [[T]], [[T]]* {{.+}}, i{{[0-9]+}} 0, i{{[0-9]+}} 8
     37         // CHECK1: [[ARR_SIZE1:%.+]] = load [[SIZE_TYPE]], [[SIZE_TYPE]]* [[ARR_SIZE1_REF]]
     38         // CHECK1: [[ARR_SIZE2_REF:%.+]] = getelementptr inbounds [[T]], [[T]]* {{.+}}, i{{[0-9]+}} 0, i{{[0-9]+}} 9
     39         // CHECK1: [[ARR_SIZE2:%.+]] = load [[SIZE_TYPE]], [[SIZE_TYPE]]* [[ARR_SIZE2_REF]]
     40         //
     41         // CHECK1: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 2
     42         // CHECK1-NEXT: load %struct.A*, %struct.A**
     43         // CHECK1-NEXT: getelementptr inbounds %struct.A, %struct.A*
     44         // CHECK1-NEXT: store i{{.+}} 1
     45         //
     46         // CHECK1: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 1
     47         // CHECK1-NEXT: load i{{[0-9]+}}*, i{{[0-9]+}}**
     48         // CHECK1-NEXT: store i{{[0-9]+}} 1
     49         //
     50         // CHECK1: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 0
     51         // CHECK1-NEXT: load i{{[0-9]+}}*, i{{[0-9]+}}**
     52         // CHECK1-NEXT: store i{{[0-9]+}} 1
     53         //
     54         // CHECK1: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 4
     55         // CHECK1-NEXT: load i{{[0-9]+}}*, i{{[0-9]+}}**
     56         // CHECK1-NEXT: load i{{[0-9]+}}, i{{[0-9]+}}*
     57         // CHECK1-NEXT: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 3
     58         // CHECK1-NEXT: load i{{[0-9]+}}**, i{{[0-9]+}}***
     59         // CHECK1-NEXT: load i{{[0-9]+}}*, i{{[0-9]+}}**
     60         // CHECK1-NEXT: store i{{[0-9]+}}
     61         //
     62         // CHECK1: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 2
     63         // CHECK1-NEXT: load %struct.A*, %struct.A**
     64         // CHECK1-NEXT: getelementptr inbounds %struct.A, %struct.A*
     65         // CHECK1-NEXT: store float
     66         //
     67         // CHECK1: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 2
     68         // CHECK1-NEXT: load %struct.A*, %struct.A**
     69         // CHECK1-NEXT: getelementptr inbounds %struct.A, %struct.A*
     70         // CHECK1-NEXT: store i8 99
     71         //
     72         // CHECK1: [[SIZE_ADDR_REF:%.*]] = getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{.+}} 0, i{{.+}} 7
     73         // CHECK1-DAG: [[SIZE_ADDR:%.*]] = load i{{.+}}*, i{{.+}}** [[SIZE_ADDR_REF]]
     74         // CHECK1-DAG: [[SIZE:%.*]] = load i{{.+}}, i{{.+}}* [[SIZE_ADDR]]
     75         // CHECK1-DAG: [[PARAM_ARR_IDX:%.*]] = sub nsw i{{.+}} [[SIZE]], 1
     76         // CHECK1-DAG: [[PARAM_ARR_ADDR_REF:%.*]] = getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{.+}} 0, i{{.+}} 6
     77         // CHECK1-DAG: [[PARAM_ARR_ADDR:%.*]] = load i{{.+}}**, i{{.+}}*** [[PARAM_ARR_ADDR_REF]]
     78         // CHECK1-DAG: [[PARAM_ARR:%.*]] = load i{{.+}}*, i{{.+}}** [[PARAM_ARR_ADDR]]
     79         // CHECK1-DAG: [[PARAM_ARR_SIZE_MINUS_1_ADDR:%.*]] = getelementptr inbounds i{{.+}}, i{{.+}}* [[PARAM_ARR]], i{{.*}}
     80         // CHECK1: store i{{.+}} 2, i{{.+}}* [[PARAM_ARR_SIZE_MINUS_1_ADDR]]
     81         //
     82         // CHECK1: [[Z_ADDR_REF:%.*]] = getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{.+}} 0, i{{.+}} 2
     83         // CHECK1-DAG: [[Z_ADDR:%.*]] = load %struct.A*, %struct.A** [[Z_ADDR_REF]]
     84         // CHECK1-DAG: [[Z_A_ADDR:%.*]] = getelementptr inbounds %struct.A, %struct.A* [[Z_ADDR]], i{{.+}} 0, i{{.+}} 0
     85         // CHECK1-DAG: [[ARR_IDX_2:%.*]] = load i{{.+}}, i{{.+}}* [[Z_A_ADDR]]
     86         // CHECK1-DAG: [[ARR_ADDR_REF:%.*]] = getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{.+}} 0, i{{.+}} 10
     87         // CHECK1-DAG: [[ARR_ADDR:%.*]] = load i{{.+}}*, i{{.+}}** [[ARR_ADDR_REF]]
     88         // CHECK1-DAG: [[ARR_IDX_1:%.*]] = mul {{.*}} 10
     89         // CHECK1-DAG: [[ARR_10_ADDR:%.*]] = getelementptr inbounds i{{.+}}, i{{.+}}* [[ARR_ADDR]], i{{.*}} [[ARR_IDX_1]]
     90         // CHECK1-DAG: [[ARR_10_Z_A_ADDR:%.*]] = getelementptr inbounds i{{.+}}, i{{.+}}* [[ARR_10_ADDR]], i{{.*}}
     91         // CHECK1: store i{{.+}} 12, i{{.+}}* [[ARR_10_Z_A_ADDR]]
     92       }
     93     }
     94   }
     95 }
     96 
     97 void test_nest_block() {
     98   __block int x;
     99   int y;
    100   ^{
    101     int z;
    102     x = z;
    103     #pragma clang __debug captured
    104     {
    105       z = y; // OK
    106     }
    107   }();
    108 
    109   // CHECK2: define internal {{.*}}void @{{.*}}test_nest_block_block_invoke
    110   //
    111   // CHECK2: [[Z:%[0-9a-z_]*]] = alloca i{{[0-9]+}},
    112   // CHECK2: alloca %struct.anon{{.*}}
    113   //
    114   // CHECK2: store i{{[0-9]+}}
    115   // CHECK2: store i{{[0-9]+}}* [[Z]]
    116   //
    117   // CHECK2: getelementptr inbounds %struct.anon
    118   // CHECK2-NEXT: getelementptr inbounds
    119   // CHECK2-NEXT: store i{{[0-9]+}}*
    120   //
    121   // CHECK2: call void @__captured_stmt
    122 
    123   int a;
    124   #pragma clang __debug captured
    125   {
    126     __block int b;
    127     int c;
    128     __block int d;
    129     ^{
    130       b = a;
    131       b = c;
    132       b = d;
    133     }();
    134   }
    135 
    136   // CHECK2: alloca %struct.__block_byref_b
    137   // CHECK2-NEXT: [[C:%[0-9a-z_]*]] = alloca i{{[0-9]+}}
    138   // CHECK2-NEXT: alloca %struct.__block_byref_d
    139   //
    140   // CHECK2: bitcast %struct.__block_byref_b*
    141   // CHECK2-NEXT: store i8*
    142   //
    143   // CHECK2: [[CapA:%[0-9a-z_.]*]] = getelementptr inbounds {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 7
    144   //
    145   // CHECK2: getelementptr inbounds %struct.anon{{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 0
    146   // CHECK2: load i{{[0-9]+}}*, i{{[0-9]+}}**
    147   // CHECK2: load i{{[0-9]+}}, i{{[0-9]+}}*
    148   // CHECK2: store i{{[0-9]+}} {{.*}}, i{{[0-9]+}}* [[CapA]]
    149   //
    150   // CHECK2: [[CapC:%[0-9a-z_.]*]] = getelementptr inbounds {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 8
    151   // CHECK2-NEXT: [[Val:%[0-9a-z_]*]] = load i{{[0-9]+}}, i{{[0-9]+}}* [[C]]
    152   // CHECK2-NEXT: store i{{[0-9]+}} [[Val]], i{{[0-9]+}}* [[CapC]]
    153   //
    154   // CHECK2: bitcast %struct.__block_byref_d*
    155   // CHECK2-NEXT: store i8*
    156 }
    157