Home | History | Annotate | Download | only in CodeGen
      1 // RUN: %clang_cc1 %s -emit-llvm -o - -triple=i686-apple-darwin9 | FileCheck %s
      2 
      3 // Also test serialization of atomic operations here, to avoid duplicating the
      4 // test.
      5 // RUN: %clang_cc1 %s -emit-pch -o %t -triple=i686-apple-darwin9
      6 // RUN: %clang_cc1 %s -include-pch %t -triple=i686-apple-darwin9 -emit-llvm -o - | FileCheck %s
      7 #ifndef ALREADY_INCLUDED
      8 #define ALREADY_INCLUDED
      9 
     10 // Basic IRGen tests for __c11_atomic_* and GNU __atomic_*
     11 
     12 typedef enum memory_order {
     13   memory_order_relaxed, memory_order_consume, memory_order_acquire,
     14   memory_order_release, memory_order_acq_rel, memory_order_seq_cst
     15 } memory_order;
     16 
     17 int fi1(_Atomic(int) *i) {
     18   // CHECK-LABEL: @fi1
     19   // CHECK: load atomic i32* {{.*}} seq_cst
     20   return __c11_atomic_load(i, memory_order_seq_cst);
     21 }
     22 
     23 int fi1a(int *i) {
     24   // CHECK-LABEL: @fi1a
     25   // CHECK: load atomic i32* {{.*}} seq_cst
     26   int v;
     27   __atomic_load(i, &v, memory_order_seq_cst);
     28   return v;
     29 }
     30 
     31 int fi1b(int *i) {
     32   // CHECK-LABEL: @fi1b
     33   // CHECK: load atomic i32* {{.*}} seq_cst
     34   return __atomic_load_n(i, memory_order_seq_cst);
     35 }
     36 
     37 void fi2(_Atomic(int) *i) {
     38   // CHECK-LABEL: @fi2
     39   // CHECK: store atomic i32 {{.*}} seq_cst
     40   __c11_atomic_store(i, 1, memory_order_seq_cst);
     41 }
     42 
     43 void fi2a(int *i) {
     44   // CHECK-LABEL: @fi2a
     45   // CHECK: store atomic i32 {{.*}} seq_cst
     46   int v = 1;
     47   __atomic_store(i, &v, memory_order_seq_cst);
     48 }
     49 
     50 void fi2b(int *i) {
     51   // CHECK-LABEL: @fi2b
     52   // CHECK: store atomic i32 {{.*}} seq_cst
     53   __atomic_store_n(i, 1, memory_order_seq_cst);
     54 }
     55 
     56 int fi3(_Atomic(int) *i) {
     57   // CHECK-LABEL: @fi3
     58   // CHECK: atomicrmw and
     59   // CHECK-NOT: and
     60   return __c11_atomic_fetch_and(i, 1, memory_order_seq_cst);
     61 }
     62 
     63 int fi3a(int *i) {
     64   // CHECK-LABEL: @fi3a
     65   // CHECK: atomicrmw xor
     66   // CHECK-NOT: xor
     67   return __atomic_fetch_xor(i, 1, memory_order_seq_cst);
     68 }
     69 
     70 int fi3b(int *i) {
     71   // CHECK-LABEL: @fi3b
     72   // CHECK: atomicrmw add
     73   // CHECK: add
     74   return __atomic_add_fetch(i, 1, memory_order_seq_cst);
     75 }
     76 
     77 int fi3c(int *i) {
     78   // CHECK-LABEL: @fi3c
     79   // CHECK: atomicrmw nand
     80   // CHECK-NOT: and
     81   return __atomic_fetch_nand(i, 1, memory_order_seq_cst);
     82 }
     83 
     84 int fi3d(int *i) {
     85   // CHECK-LABEL: @fi3d
     86   // CHECK: atomicrmw nand
     87   // CHECK: and
     88   // CHECK: xor
     89   return __atomic_nand_fetch(i, 1, memory_order_seq_cst);
     90 }
     91 
     92 _Bool fi4(_Atomic(int) *i) {
     93   // CHECK-LABEL: @fi4
     94   // CHECK: [[PAIR:%[.0-9A-Z_a-z]+]] = cmpxchg i32* [[PTR:%[.0-9A-Z_a-z]+]], i32 [[EXPECTED:%[.0-9A-Z_a-z]+]], i32 [[DESIRED:%[.0-9A-Z_a-z]+]]
     95   // CHECK: [[OLD:%[.0-9A-Z_a-z]+]] = extractvalue { i32, i1 } [[PAIR]], 0
     96   // CHECK: [[CMP:%[.0-9A-Z_a-z]+]] = extractvalue { i32, i1 } [[PAIR]], 1
     97   // CHECK: br i1 [[CMP]], label %[[STORE_EXPECTED:[.0-9A-Z_a-z]+]], label %[[CONTINUE:[.0-9A-Z_a-z]+]]
     98   // CHECK: store i32 [[OLD]]
     99   int cmp = 0;
    100   return __c11_atomic_compare_exchange_strong(i, &cmp, 1, memory_order_acquire, memory_order_acquire);
    101 }
    102 
    103 _Bool fi4a(int *i) {
    104   // CHECK-LABEL: @fi4
    105   // CHECK: [[PAIR:%[.0-9A-Z_a-z]+]] = cmpxchg i32* [[PTR:%[.0-9A-Z_a-z]+]], i32 [[EXPECTED:%[.0-9A-Z_a-z]+]], i32 [[DESIRED:%[.0-9A-Z_a-z]+]]
    106   // CHECK: [[OLD:%[.0-9A-Z_a-z]+]] = extractvalue { i32, i1 } [[PAIR]], 0
    107   // CHECK: [[CMP:%[.0-9A-Z_a-z]+]] = extractvalue { i32, i1 } [[PAIR]], 1
    108   // CHECK: br i1 [[CMP]], label %[[STORE_EXPECTED:[.0-9A-Z_a-z]+]], label %[[CONTINUE:[.0-9A-Z_a-z]+]]
    109   // CHECK: store i32 [[OLD]]
    110   int cmp = 0;
    111   int desired = 1;
    112   return __atomic_compare_exchange(i, &cmp, &desired, 0, memory_order_acquire, memory_order_acquire);
    113 }
    114 
    115 _Bool fi4b(int *i) {
    116   // CHECK-LABEL: @fi4
    117   // CHECK: [[PAIR:%[.0-9A-Z_a-z]+]] = cmpxchg weak i32* [[PTR:%[.0-9A-Z_a-z]+]], i32 [[EXPECTED:%[.0-9A-Z_a-z]+]], i32 [[DESIRED:%[.0-9A-Z_a-z]+]]
    118   // CHECK: [[OLD:%[.0-9A-Z_a-z]+]] = extractvalue { i32, i1 } [[PAIR]], 0
    119   // CHECK: [[CMP:%[.0-9A-Z_a-z]+]] = extractvalue { i32, i1 } [[PAIR]], 1
    120   // CHECK: br i1 [[CMP]], label %[[STORE_EXPECTED:[.0-9A-Z_a-z]+]], label %[[CONTINUE:[.0-9A-Z_a-z]+]]
    121   // CHECK: store i32 [[OLD]]
    122   int cmp = 0;
    123   return __atomic_compare_exchange_n(i, &cmp, 1, 1, memory_order_acquire, memory_order_acquire);
    124 }
    125 
    126 float ff1(_Atomic(float) *d) {
    127   // CHECK-LABEL: @ff1
    128   // CHECK: load atomic i32* {{.*}} monotonic
    129   return __c11_atomic_load(d, memory_order_relaxed);
    130 }
    131 
    132 void ff2(_Atomic(float) *d) {
    133   // CHECK-LABEL: @ff2
    134   // CHECK: store atomic i32 {{.*}} release
    135   __c11_atomic_store(d, 1, memory_order_release);
    136 }
    137 
    138 float ff3(_Atomic(float) *d) {
    139   return __c11_atomic_exchange(d, 2, memory_order_seq_cst);
    140 }
    141 
    142 int* fp1(_Atomic(int*) *p) {
    143   // CHECK-LABEL: @fp1
    144   // CHECK: load atomic i32* {{.*}} seq_cst
    145   return __c11_atomic_load(p, memory_order_seq_cst);
    146 }
    147 
    148 int* fp2(_Atomic(int*) *p) {
    149   // CHECK-LABEL: @fp2
    150   // CHECK: store i32 4
    151   // CHECK: atomicrmw add {{.*}} monotonic
    152   return __c11_atomic_fetch_add(p, 1, memory_order_relaxed);
    153 }
    154 
    155 int *fp2a(int **p) {
    156   // CHECK-LABEL: @fp2a
    157   // CHECK: store i32 4
    158   // CHECK: atomicrmw sub {{.*}} monotonic
    159   // Note, the GNU builtins do not multiply by sizeof(T)!
    160   return __atomic_fetch_sub(p, 4, memory_order_relaxed);
    161 }
    162 
    163 _Complex float fc(_Atomic(_Complex float) *c) {
    164   // CHECK-LABEL: @fc
    165   // CHECK: atomicrmw xchg i64*
    166   return __c11_atomic_exchange(c, 2, memory_order_seq_cst);
    167 }
    168 
    169 typedef struct X { int x; } X;
    170 X fs(_Atomic(X) *c) {
    171   // CHECK-LABEL: @fs
    172   // CHECK: atomicrmw xchg i32*
    173   return __c11_atomic_exchange(c, (X){2}, memory_order_seq_cst);
    174 }
    175 
    176 X fsa(X *c, X *d) {
    177   // CHECK-LABEL: @fsa
    178   // CHECK: atomicrmw xchg i32*
    179   X ret;
    180   __atomic_exchange(c, d, &ret, memory_order_seq_cst);
    181   return ret;
    182 }
    183 
    184 _Bool fsb(_Bool *c) {
    185   // CHECK-LABEL: @fsb
    186   // CHECK: atomicrmw xchg i8*
    187   return __atomic_exchange_n(c, 1, memory_order_seq_cst);
    188 }
    189 
    190 char flag1;
    191 volatile char flag2;
    192 void test_and_set() {
    193   // CHECK: atomicrmw xchg i8* @flag1, i8 1 seq_cst
    194   __atomic_test_and_set(&flag1, memory_order_seq_cst);
    195   // CHECK: atomicrmw volatile xchg i8* @flag2, i8 1 acquire
    196   __atomic_test_and_set(&flag2, memory_order_acquire);
    197   // CHECK: store atomic volatile i8 0, i8* @flag2 release
    198   __atomic_clear(&flag2, memory_order_release);
    199   // CHECK: store atomic i8 0, i8* @flag1 seq_cst
    200   __atomic_clear(&flag1, memory_order_seq_cst);
    201 }
    202 
    203 struct Sixteen {
    204   char c[16];
    205 } sixteen;
    206 struct Seventeen {
    207   char c[17];
    208 } seventeen;
    209 
    210 int lock_free(struct Incomplete *incomplete) {
    211   // CHECK-LABEL: @lock_free
    212 
    213   // CHECK: call i32 @__atomic_is_lock_free(i32 3, i8* null)
    214   __c11_atomic_is_lock_free(3);
    215 
    216   // CHECK: call i32 @__atomic_is_lock_free(i32 16, i8* {{.*}}@sixteen{{.*}})
    217   __atomic_is_lock_free(16, &sixteen);
    218 
    219   // CHECK: call i32 @__atomic_is_lock_free(i32 17, i8* {{.*}}@seventeen{{.*}})
    220   __atomic_is_lock_free(17, &seventeen);
    221 
    222   // CHECK: call i32 @__atomic_is_lock_free(i32 4, {{.*}})
    223   __atomic_is_lock_free(4, incomplete);
    224 
    225   char cs[20];
    226   // CHECK: call i32 @__atomic_is_lock_free(i32 4, {{.*}})
    227   __atomic_is_lock_free(4, cs+1);
    228 
    229   // CHECK-NOT: call
    230   __atomic_always_lock_free(3, 0);
    231   __atomic_always_lock_free(16, 0);
    232   __atomic_always_lock_free(17, 0);
    233   __atomic_always_lock_free(16, &sixteen);
    234   __atomic_always_lock_free(17, &seventeen);
    235 
    236   int n;
    237   __atomic_is_lock_free(4, &n);
    238 
    239   // CHECK: ret i32 1
    240   return __c11_atomic_is_lock_free(sizeof(_Atomic(int)));
    241 }
    242 
    243 // Tests for atomic operations on big values.  These should call the functions
    244 // defined here:
    245 // http://gcc.gnu.org/wiki/Atomic/GCCMM/LIbrary#The_Library_interface
    246 
    247 struct foo {
    248   int big[128];
    249 };
    250 struct bar {
    251   char c[3];
    252 };
    253 
    254 struct bar smallThing, thing1, thing2;
    255 struct foo bigThing;
    256 _Atomic(struct foo) bigAtomic;
    257 
    258 void structAtomicStore() {
    259   // CHECK-LABEL: @structAtomicStore
    260   struct foo f = {0};
    261   struct bar b = {0};
    262   __atomic_store(&smallThing, &b, 5);
    263   // CHECK: call void @__atomic_store(i32 3, i8* {{.*}} @smallThing
    264 
    265   __atomic_store(&bigThing, &f, 5);
    266   // CHECK: call void @__atomic_store(i32 512, i8* {{.*}} @bigThing
    267 }
    268 void structAtomicLoad() {
    269   // CHECK-LABEL: @structAtomicLoad
    270   struct bar b;
    271   __atomic_load(&smallThing, &b, 5);
    272   // CHECK: call void @__atomic_load(i32 3, i8* {{.*}} @smallThing
    273 
    274   struct foo f = {0};
    275   __atomic_load(&bigThing, &f, 5);
    276   // CHECK: call void @__atomic_load(i32 512, i8* {{.*}} @bigThing
    277 }
    278 struct foo structAtomicExchange() {
    279   // CHECK-LABEL: @structAtomicExchange
    280   struct foo f = {0};
    281   struct foo old;
    282   __atomic_exchange(&f, &bigThing, &old, 5);
    283   // CHECK: call void @__atomic_exchange(i32 512, {{.*}}, i8* bitcast ({{.*}} @bigThing to i8*),
    284 
    285   return __c11_atomic_exchange(&bigAtomic, f, 5);
    286   // CHECK: call void @__atomic_exchange(i32 512, i8* bitcast ({{.*}} @bigAtomic to i8*),
    287 }
    288 int structAtomicCmpExchange() {
    289   // CHECK-LABEL: @structAtomicCmpExchange
    290   _Bool x = __atomic_compare_exchange(&smallThing, &thing1, &thing2, 1, 5, 5);
    291   // CHECK: call zeroext i1 @__atomic_compare_exchange(i32 3, {{.*}} @smallThing{{.*}} @thing1{{.*}} @thing2
    292 
    293   struct foo f = {0};
    294   struct foo g = {0};
    295   g.big[12] = 12;
    296   return x & __c11_atomic_compare_exchange_strong(&bigAtomic, &f, g, 5, 5);
    297   // CHECK: call zeroext i1 @__atomic_compare_exchange(i32 512, i8* bitcast ({{.*}} @bigAtomic to i8*),
    298 }
    299 
    300 // Check that no atomic operations are used in any initialisation of _Atomic
    301 // types.
    302 _Atomic(int) atomic_init_i = 42;
    303 
    304 // CHECK-LABEL: @atomic_init_foo
    305 void atomic_init_foo()
    306 {
    307   // CHECK-NOT: }
    308   // CHECK-NOT: atomic
    309   // CHECK: store
    310   _Atomic(int) j = 12;
    311 
    312   // CHECK-NOT: }
    313   // CHECK-NOT: atomic
    314   // CHECK: store
    315   __c11_atomic_init(&j, 42);
    316 
    317   // CHECK-NOT: atomic
    318   // CHECK: }
    319 }
    320 
    321 // CHECK-LABEL: @failureOrder
    322 void failureOrder(_Atomic(int) *ptr, int *ptr2) {
    323   __c11_atomic_compare_exchange_strong(ptr, ptr2, 43, memory_order_acquire, memory_order_relaxed);
    324   // CHECK: cmpxchg i32* {{%[0-9A-Za-z._]+}}, i32 {{%[0-9A-Za-z._]+}}, i32 {{%[0-9A-Za-z_.]+}} acquire monotonic
    325 
    326   __c11_atomic_compare_exchange_weak(ptr, ptr2, 43, memory_order_seq_cst, memory_order_acquire);
    327   // CHECK: cmpxchg weak i32* {{%[0-9A-Za-z._]+}}, i32 {{%[0-9A-Za-z._]+}}, i32 {{%[0-9A-Za-z_.]+}} seq_cst acquire
    328 
    329   // Unknown ordering: conservatively pick strongest valid option (for now!).
    330   __atomic_compare_exchange(ptr2, ptr2, ptr2, 0, memory_order_acq_rel, *ptr2);
    331   // CHECK: cmpxchg i32* {{%[0-9A-Za-z._]+}}, i32 {{%[0-9A-Za-z._]+}}, i32 {{%[0-9A-Za-z_.]+}} acq_rel acquire
    332 
    333   // Undefined behaviour: don't really care what that last ordering is so leave
    334   // it out:
    335   __atomic_compare_exchange_n(ptr2, ptr2, 43, 1, memory_order_seq_cst, 42);
    336   // CHECK: cmpxchg weak i32* {{%[0-9A-Za-z._]+}}, i32 {{%[0-9A-Za-z._]+}}, i32 {{%[0-9A-Za-z_.]+}} seq_cst
    337 }
    338 
    339 // CHECK-LABEL: @generalFailureOrder
    340 void generalFailureOrder(_Atomic(int) *ptr, int *ptr2, int success, int fail) {
    341   __c11_atomic_compare_exchange_strong(ptr, ptr2, 42, success, fail);
    342   // CHECK: switch i32 {{.*}}, label %[[MONOTONIC:[0-9a-zA-Z._]+]] [
    343   // CHECK-NEXT: i32 1, label %[[ACQUIRE:[0-9a-zA-Z._]+]]
    344   // CHECK-NEXT: i32 2, label %[[ACQUIRE]]
    345   // CHECK-NEXT: i32 3, label %[[RELEASE:[0-9a-zA-Z._]+]]
    346   // CHECK-NEXT: i32 4, label %[[ACQREL:[0-9a-zA-Z._]+]]
    347   // CHECK-NEXT: i32 5, label %[[SEQCST:[0-9a-zA-Z._]+]]
    348 
    349   // CHECK: [[MONOTONIC]]
    350   // CHECK: switch {{.*}}, label %[[MONOTONIC_MONOTONIC:[0-9a-zA-Z._]+]] [
    351   // CHECK-NEXT: ]
    352 
    353   // CHECK: [[ACQUIRE]]
    354   // CHECK: switch {{.*}}, label %[[ACQUIRE_MONOTONIC:[0-9a-zA-Z._]+]] [
    355   // CHECK-NEXT: i32 1, label %[[ACQUIRE_ACQUIRE:[0-9a-zA-Z._]+]]
    356   // CHECK-NEXT: i32 2, label %[[ACQUIRE_ACQUIRE:[0-9a-zA-Z._]+]]
    357   // CHECK-NEXT: ]
    358 
    359   // CHECK: [[RELEASE]]
    360   // CHECK: switch {{.*}}, label %[[RELEASE_MONOTONIC:[0-9a-zA-Z._]+]] [
    361   // CHECK-NEXT: ]
    362 
    363   // CHECK: [[ACQREL]]
    364   // CHECK: switch {{.*}}, label %[[ACQREL_MONOTONIC:[0-9a-zA-Z._]+]] [
    365   // CHECK-NEXT: i32 1, label %[[ACQREL_ACQUIRE:[0-9a-zA-Z._]+]]
    366   // CHECK-NEXT: i32 2, label %[[ACQREL_ACQUIRE:[0-9a-zA-Z._]+]]
    367   // CHECK-NEXT: ]
    368 
    369   // CHECK: [[SEQCST]]
    370   // CHECK: switch {{.*}}, label %[[SEQCST_MONOTONIC:[0-9a-zA-Z._]+]] [
    371   // CHECK-NEXT: i32 1, label %[[SEQCST_ACQUIRE:[0-9a-zA-Z._]+]]
    372   // CHECK-NEXT: i32 2, label %[[SEQCST_ACQUIRE:[0-9a-zA-Z._]+]]
    373   // CHECK-NEXT: i32 5, label %[[SEQCST_SEQCST:[0-9a-zA-Z._]+]]
    374   // CHECK-NEXT: ]
    375 
    376   // CHECK: [[MONOTONIC_MONOTONIC]]
    377   // CHECK: cmpxchg {{.*}} monotonic monotonic
    378   // CHECK: br
    379 
    380   // CHECK: [[ACQUIRE_MONOTONIC]]
    381   // CHECK: cmpxchg {{.*}} acquire monotonic
    382   // CHECK: br
    383 
    384   // CHECK: [[ACQUIRE_ACQUIRE]]
    385   // CHECK: cmpxchg {{.*}} acquire acquire
    386   // CHECK: br
    387 
    388   // CHECK: [[ACQREL_MONOTONIC]]
    389   // CHECK: cmpxchg {{.*}} acq_rel monotonic
    390   // CHECK: br
    391 
    392   // CHECK: [[ACQREL_ACQUIRE]]
    393   // CHECK: cmpxchg {{.*}} acq_rel acquire
    394   // CHECK: br
    395 
    396   // CHECK: [[SEQCST_MONOTONIC]]
    397   // CHECK: cmpxchg {{.*}} seq_cst monotonic
    398   // CHECK: br
    399 
    400   // CHECK: [[SEQCST_ACQUIRE]]
    401   // CHECK: cmpxchg {{.*}} seq_cst acquire
    402   // CHECK: br
    403 
    404   // CHECK: [[SEQCST_SEQCST]]
    405   // CHECK: cmpxchg {{.*}} seq_cst seq_cst
    406   // CHECK: br
    407 }
    408 
    409 void generalWeakness(int *ptr, int *ptr2, _Bool weak) {
    410   __atomic_compare_exchange_n(ptr, ptr2, 42, weak, memory_order_seq_cst, memory_order_seq_cst);
    411   // CHECK: switch i1 {{.*}}, label %[[WEAK:[0-9a-zA-Z._]+]] [
    412   // CHECK-NEXT: i1 false, label %[[STRONG:[0-9a-zA-Z._]+]]
    413 
    414   // CHECK: [[STRONG]]
    415   // CHECK-NOT: br
    416   // CHECK: cmpxchg {{.*}} seq_cst seq_cst
    417   // CHECK: br
    418 
    419   // CHECK: [[WEAK]]
    420   // CHECK-NOT: br
    421   // CHECK: cmpxchg weak {{.*}} seq_cst seq_cst
    422   // CHECK: br
    423 }
    424 
    425 // Having checked the flow in the previous two cases, we'll trust clang to
    426 // combine them sanely.
    427 void EMIT_ALL_THE_THINGS(int *ptr, int *ptr2, int new, _Bool weak, int success, int fail) {
    428   __atomic_compare_exchange(ptr, ptr2, &new, weak, success, fail);
    429 
    430   // CHECK: = cmpxchg {{.*}} monotonic monotonic
    431   // CHECK: = cmpxchg weak {{.*}} monotonic monotonic
    432   // CHECK: = cmpxchg {{.*}} acquire monotonic
    433   // CHECK: = cmpxchg {{.*}} acquire acquire
    434   // CHECK: = cmpxchg weak {{.*}} acquire monotonic
    435   // CHECK: = cmpxchg weak {{.*}} acquire acquire
    436   // CHECK: = cmpxchg {{.*}} release monotonic
    437   // CHECK: = cmpxchg weak {{.*}} release monotonic
    438   // CHECK: = cmpxchg {{.*}} acq_rel monotonic
    439   // CHECK: = cmpxchg {{.*}} acq_rel acquire
    440   // CHECK: = cmpxchg weak {{.*}} acq_rel monotonic
    441   // CHECK: = cmpxchg weak {{.*}} acq_rel acquire
    442   // CHECK: = cmpxchg {{.*}} seq_cst monotonic
    443   // CHECK: = cmpxchg {{.*}} seq_cst acquire
    444   // CHECK: = cmpxchg {{.*}} seq_cst seq_cst
    445   // CHECK: = cmpxchg weak {{.*}} seq_cst monotonic
    446   // CHECK: = cmpxchg weak {{.*}} seq_cst acquire
    447   // CHECK: = cmpxchg weak {{.*}} seq_cst seq_cst
    448 }
    449 
    450 #endif
    451