compiler-rt/lib/tsan/benchmarks/mini_bench_shared.cpp

   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 }