1 // Mini-benchmark for tsan: shared memory reads. 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 if (a[i] != i) abort(); 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[len]; 39 for (int i = 0, n = len; i < n; i++) 40 a[i] = i; 41 pthread_t *t = new pthread_t[n_threads]; 42 for (int i = 0; i < n_threads; i++) { 43 pthread_create(&t[i], 0, Thread, (void*)i); 44 } 45 for (int i = 0; i < n_threads; i++) { 46 pthread_join(t[i], 0); 47 } 48 delete [] t; 49 delete [] a; 50 return 0; 51 } 52
