1 // Mini-benchmark for tsan: non-shared memory writes. 2 #include <pthread.h> 3 #include <stdio.h> 4 #include <stdlib.h> 5 #include <assert.h> 6 7 int len; 8 int *a; 9 const int kNumIter = 1000; 10 11 __attribute__((noinline)) 12 void Run(int idx) { 13 for (int i = 0, n = len; i < n; i++) 14 a[i + idx * n] = i; 15 } 16 17 void *Thread(void *arg) { 18 long idx = (long)arg; 19 printf("Thread %ld started\n", idx); 20 for (int i = 0; i < kNumIter; i++) 21 Run(idx); 22 printf("Thread %ld done\n", idx); 23 return 0; 24 } 25 26 int main(int argc, char **argv) { 27 int n_threads = 0; 28 if (argc != 3) { 29 n_threads = 4; 30 len = 1000000; 31 } else { 32 n_threads = atoi(argv[1]); 33 assert(n_threads > 0 && n_threads <= 32); 34 len = atoi(argv[2]); 35 } 36 printf("%s: n_threads=%d len=%d iter=%d\n", 37 __FILE__, n_threads, len, kNumIter); 38 a = new int[n_threads * len]; 39 pthread_t *t = new pthread_t[n_threads]; 40 for (int i = 0; i < n_threads; i++) { 41 pthread_create(&t[i], 0, Thread, (void*)i); 42 } 43 for (int i = 0; i < n_threads; i++) { 44 pthread_join(t[i], 0); 45 } 46 delete [] t; 47 delete [] a; 48 return 0; 49 } 50
