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 { i32*, i32*, i{{.+}}*, i32**, i32* }
     15   // CHECK1: %struct.anon{{.*}} = type { i32*, i32*, i32**, i32*, i{{.+}}*, i32**, i32* }
     16   // CHECK1: [[T:%struct.anon.*]] = type { i32*, i32*, %struct.A*, i32**, i32*, i{{.+}}*, i32**, i32* }
     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         //
     35         // CHECK1: getelementptr inbounds [[T]]* {{.*}}, i32 0, i32 2
     36         // CHECK1-NEXT: load %struct.A**
     37         // CHECK1-NEXT: getelementptr inbounds %struct.A*
     38         // CHECK1-NEXT: store i{{.+}} 1
     39         //
     40         // CHECK1: getelementptr inbounds [[T]]* {{.*}}, i32 0, i32 1
     41         // CHECK1-NEXT: load i32**
     42         // CHECK1-NEXT: store i32 1
     43         //
     44         // CHECK1: getelementptr inbounds [[T]]* {{.*}}, i32 0, i32 0
     45         // CHECK1-NEXT: load i32**
     46         // CHECK1-NEXT: store i32 1
     47         //
     48         // CHECK1: getelementptr inbounds [[T]]* {{.*}}, i32 0, i32 4
     49         // CHECK1-NEXT: load i32**
     50         // CHECK1-NEXT: load i32*
     51         // CHECK1-NEXT: getelementptr inbounds [[T]]* {{.*}}, i32 0, i32 3
     52         // CHECK1-NEXT: load i32***
     53         // CHECK1-NEXT: load i32**
     54         // CHECK1-NEXT: store i32
     55         //
     56         // CHECK1: getelementptr inbounds [[T]]* {{.*}}, i32 0, i32 2
     57         // CHECK1-NEXT: load %struct.A**
     58         // CHECK1-NEXT: getelementptr inbounds %struct.A*
     59         // CHECK1-NEXT: store float
     60         //
     61         // CHECK1: getelementptr inbounds [[T]]* {{.*}}, i32 0, i32 2
     62         // CHECK1-NEXT: load %struct.A**
     63         // CHECK1-NEXT: getelementptr inbounds %struct.A*
     64         // CHECK1-NEXT: store i8 99
     65         //
     66         // CHECK1: [[SIZE_ADDR_REF:%.*]] = getelementptr inbounds [[T]]* {{.*}}, i{{.+}} 0, i{{.+}} 5
     67         // CHECK1-DAG: [[SIZE_ADDR:%.*]] = load i{{.+}}** [[SIZE_ADDR_REF]]
     68         // CHECK1-DAG: [[SIZE:%.*]] = load i{{.+}}* [[SIZE_ADDR]]
     69         // CHECK1-DAG: [[PARAM_ARR_IDX:%.*]] = sub nsw i{{.+}} [[SIZE]], 1
     70         // CHECK1-DAG: [[PARAM_ARR_ADDR_REF:%.*]] = getelementptr inbounds [[T]]* {{.*}}, i{{.+}} 0, i{{.+}} 6
     71         // CHECK1-DAG: [[PARAM_ARR_ADDR:%.*]] = load i{{.+}}*** [[PARAM_ARR_ADDR_REF]]
     72         // CHECK1-DAG: [[PARAM_ARR:%.*]] = load i{{.+}}** [[PARAM_ARR_ADDR]]
     73         // CHECK1-DAG: [[PARAM_ARR_SIZE_MINUS_1_ADDR:%.*]] = getelementptr inbounds i{{.+}}* [[PARAM_ARR]], i{{.*}}
     74         // CHECK1: store i{{.+}} 2, i{{.+}}* [[PARAM_ARR_SIZE_MINUS_1_ADDR]]
     75         //
     76         // CHECK1: [[Z_ADDR_REF:%.*]] = getelementptr inbounds [[T]]* {{.*}}, i{{.+}} 0, i{{.+}} 2
     77         // CHECK1-DAG: [[Z_ADDR:%.*]] = load %struct.A** [[Z_ADDR_REF]]
     78         // CHECK1-DAG: [[Z_A_ADDR:%.*]] = getelementptr inbounds %struct.A* [[Z_ADDR]], i{{.+}} 0, i{{.+}} 0
     79         // CHECK1-DAG: [[ARR_IDX_2:%.*]] = load i{{.+}}* [[Z_A_ADDR]]
     80         // CHECK1-DAG: [[ARR_ADDR_REF:%.*]] = getelementptr inbounds [[T]]* {{.*}}, i{{.+}} 0, i{{.+}} 7
     81         // CHECK1-DAG: [[ARR_ADDR:%.*]] = load i{{.+}}** [[ARR_ADDR_REF]]
     82         // CHECK1-DAG: [[ARR_IDX_1:%.*]] = mul {{.*}} 10
     83         // CHECK1-DAG: [[ARR_10_ADDR:%.*]] = getelementptr inbounds i{{.+}}* [[ARR_ADDR]], i{{.*}} [[ARR_IDX_1]]
     84         // CHECK1-DAG: [[ARR_10_Z_A_ADDR:%.*]] = getelementptr inbounds i{{.+}}* [[ARR_10_ADDR]], i{{.*}}
     85         // CHECK1: store i{{.+}} 12, i{{.+}}* [[ARR_10_Z_A_ADDR]]
     86       }
     87     }
     88   }
     89 }
     90 
     91 void test_nest_block() {
     92   __block int x;
     93   int y;
     94   ^{
     95     int z;
     96     x = z;
     97     #pragma clang __debug captured
     98     {
     99       z = y; // OK
    100     }
    101   }();
    102 
    103   // CHECK2: define internal void @{{.*}}test_nest_block_block_invoke
    104   //
    105   // CHECK2: [[Z:%[0-9a-z_]*]] = alloca i32
    106   // CHECK2: alloca %struct.anon{{.*}}
    107   //
    108   // CHECK2: store i32
    109   // CHECK2: store i32* [[Z]]
    110   //
    111   // CHECK2: getelementptr inbounds %struct.anon
    112   // CHECK2-NEXT: getelementptr inbounds
    113   // CHECK2-NEXT: store i32*
    114   //
    115   // CHECK2: call void @__captured_stmt
    116 
    117   int a;
    118   #pragma clang __debug captured
    119   {
    120     __block int b;
    121     int c;
    122     __block int d;
    123     ^{
    124       b = a;
    125       b = c;
    126       b = d;
    127     }();
    128   }
    129 
    130   // CHECK2: alloca %struct.__block_byref_b
    131   // CHECK2-NEXT: [[C:%[0-9a-z_]*]] = alloca i32
    132   // CHECK2-NEXT: alloca %struct.__block_byref_d
    133   //
    134   // CHECK2: bitcast %struct.__block_byref_b*
    135   // CHECK2-NEXT: store i8*
    136   //
    137   // CHECK2: [[CapA:%[0-9a-z_.]*]] = getelementptr inbounds {{.*}}, i32 0, i32 7
    138   //
    139   // CHECK2: getelementptr inbounds %struct.anon{{.*}}, i32 0, i32 0
    140   // CHECK2: load i32**
    141   // CHECK2: load i32*
    142   // CHECK2: store i32 {{.*}}, i32* [[CapA]]
    143   //
    144   // CHECK2: [[CapC:%[0-9a-z_.]*]] = getelementptr inbounds {{.*}}, i32 0, i32 8
    145   // CHECK2-NEXT: [[Val:%[0-9a-z_]*]] = load i32* [[C]]
    146   // CHECK2-NEXT: store i32 [[Val]], i32* [[CapC]]
    147   //
    148   // CHECK2: bitcast %struct.__block_byref_d*
    149   // CHECK2-NEXT: store i8*
    150 }
    151