reference, declarationdefinition
definition → references, declarations, derived classes, virtual overrides
reference to multiple definitions → definitions
unreferenced
    1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
// Mini-benchmark for tsan: non-shared memory writes.
#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include <assert.h>

int len;
int *a;
const int kNumIter = 1000;

__attribute__((noinline))
void Run(int idx) {
  for (int i = 0, n = len; i < n; i++)
    a[i + idx * n] = i;
}

void *Thread(void *arg) {
  long idx = (long)arg;
  printf("Thread %ld started\n", idx);
  for (int i = 0; i < kNumIter; i++)
    Run(idx);
  printf("Thread %ld done\n", idx);
  return 0;
}

int main(int argc, char **argv) {
  int n_threads = 0;
  if (argc != 3) {
    n_threads = 4;
    len = 1000000;
  } else {
    n_threads = atoi(argv[1]);
    assert(n_threads > 0 && n_threads <= 32);
    len = atoi(argv[2]);
  }
  printf("%s: n_threads=%d len=%d iter=%d\n",
         __FILE__, n_threads, len, kNumIter);
  a = new int[n_threads * len];
  pthread_t *t = new pthread_t[n_threads];
  for (int i = 0; i < n_threads; i++) {
    pthread_create(&t[i], 0, Thread, (void*)i);
  }
  for (int i = 0; i < n_threads; i++) {
    pthread_join(t[i], 0);
  }
  delete [] t;
  delete [] a;
  return 0;
}