1*bcaeed49SFangrui Song // RUN: %clangxx_tsan -O1 %s -o %t && %run %t 2>&1 | FileCheck %s
2*bcaeed49SFangrui Song #include "test.h"
3*bcaeed49SFangrui Song
4*bcaeed49SFangrui Song const int kTestCount = 4;
5*bcaeed49SFangrui Song typedef long long T;
6*bcaeed49SFangrui Song T atomics[kTestCount * 2];
7*bcaeed49SFangrui Song
Test(int test,T * p,bool main_thread)8*bcaeed49SFangrui Song void Test(int test, T *p, bool main_thread) {
9*bcaeed49SFangrui Song volatile T sink;
10*bcaeed49SFangrui Song if (test == 0) {
11*bcaeed49SFangrui Song if (main_thread)
12*bcaeed49SFangrui Song __atomic_fetch_add(p, 1, __ATOMIC_RELAXED);
13*bcaeed49SFangrui Song else
14*bcaeed49SFangrui Song __atomic_fetch_add(p, 1, __ATOMIC_RELAXED);
15*bcaeed49SFangrui Song } else if (test == 1) {
16*bcaeed49SFangrui Song if (main_thread)
17*bcaeed49SFangrui Song __atomic_exchange_n(p, 1, __ATOMIC_ACQ_REL);
18*bcaeed49SFangrui Song else
19*bcaeed49SFangrui Song __atomic_exchange_n(p, 1, __ATOMIC_ACQ_REL);
20*bcaeed49SFangrui Song } else if (test == 2) {
21*bcaeed49SFangrui Song if (main_thread)
22*bcaeed49SFangrui Song sink = __atomic_load_n(p, __ATOMIC_SEQ_CST);
23*bcaeed49SFangrui Song else
24*bcaeed49SFangrui Song __atomic_store_n(p, 1, __ATOMIC_SEQ_CST);
25*bcaeed49SFangrui Song } else if (test == 3) {
26*bcaeed49SFangrui Song if (main_thread)
27*bcaeed49SFangrui Song sink = __atomic_load_n(p, __ATOMIC_SEQ_CST);
28*bcaeed49SFangrui Song else
29*bcaeed49SFangrui Song sink = *p;
30*bcaeed49SFangrui Song }
31*bcaeed49SFangrui Song }
32*bcaeed49SFangrui Song
Thread(void * p)33*bcaeed49SFangrui Song void *Thread(void *p) {
34*bcaeed49SFangrui Song for (int i = 0; i < kTestCount; i++) {
35*bcaeed49SFangrui Song Test(i, &atomics[i], false);
36*bcaeed49SFangrui Song }
37*bcaeed49SFangrui Song barrier_wait(&barrier);
38*bcaeed49SFangrui Song barrier_wait(&barrier);
39*bcaeed49SFangrui Song for (int i = 0; i < kTestCount; i++) {
40*bcaeed49SFangrui Song fprintf(stderr, "Test %d reverse\n", i);
41*bcaeed49SFangrui Song Test(i, &atomics[kTestCount + i], false);
42*bcaeed49SFangrui Song }
43*bcaeed49SFangrui Song return 0;
44*bcaeed49SFangrui Song }
45*bcaeed49SFangrui Song
main()46*bcaeed49SFangrui Song int main() {
47*bcaeed49SFangrui Song barrier_init(&barrier, 2);
48*bcaeed49SFangrui Song pthread_t t;
49*bcaeed49SFangrui Song pthread_create(&t, 0, Thread, 0);
50*bcaeed49SFangrui Song barrier_wait(&barrier);
51*bcaeed49SFangrui Song for (int i = 0; i < kTestCount; i++) {
52*bcaeed49SFangrui Song fprintf(stderr, "Test %d\n", i);
53*bcaeed49SFangrui Song Test(i, &atomics[i], true);
54*bcaeed49SFangrui Song }
55*bcaeed49SFangrui Song for (int i = 0; i < kTestCount; i++) {
56*bcaeed49SFangrui Song Test(i, &atomics[kTestCount + i], true);
57*bcaeed49SFangrui Song }
58*bcaeed49SFangrui Song barrier_wait(&barrier);
59*bcaeed49SFangrui Song pthread_join(t, 0);
60*bcaeed49SFangrui Song }
61*bcaeed49SFangrui Song
62*bcaeed49SFangrui Song // CHECK-NOT: ThreadSanitizer: data race
63