xref: /llvm-project/compiler-rt/test/tsan/bench_threads.cpp (revision b332134921b42796c6b46453eaf2affdc09e3154)
1*64b45399SDmitry Vyukov // RUN: %clangxx_tsan %s -o %t
2*64b45399SDmitry Vyukov // RUN: %run %t 2>&1 | FileCheck %s
3*64b45399SDmitry Vyukov 
4*64b45399SDmitry Vyukov // bench.h needs pthread barriers which are not available on OS X
5*64b45399SDmitry Vyukov // UNSUPPORTED: darwin
6*64b45399SDmitry Vyukov 
7*64b45399SDmitry Vyukov #include "bench.h"
8*64b45399SDmitry Vyukov 
nop_thread(void * arg)9*64b45399SDmitry Vyukov void *nop_thread(void *arg) {
10*64b45399SDmitry Vyukov   pthread_setname_np(pthread_self(), "nop_thread");
11*64b45399SDmitry Vyukov   return nullptr;
12*64b45399SDmitry Vyukov }
13*64b45399SDmitry Vyukov 
thread(int tid)14*64b45399SDmitry Vyukov void thread(int tid) {
15*64b45399SDmitry Vyukov   for (int i = 0; i < bench_niter; i++) {
16*64b45399SDmitry Vyukov     pthread_t th;
17*64b45399SDmitry Vyukov     pthread_create(&th, nullptr, nop_thread, nullptr);
18*64b45399SDmitry Vyukov     pthread_join(th, nullptr);
19*64b45399SDmitry Vyukov   }
20*64b45399SDmitry Vyukov }
21*64b45399SDmitry Vyukov 
bench()22*64b45399SDmitry Vyukov void bench() {
23*64b45399SDmitry Vyukov   // Benchmark thread creation/joining in presence of a large number
24*64b45399SDmitry Vyukov   // of threads (both alive and already joined).
25*64b45399SDmitry Vyukov   printf("starting transient threads...\n");
26*64b45399SDmitry Vyukov   for (int i = 0; i < 200; i++) {
27*64b45399SDmitry Vyukov     const int kBatch = 100;
28*64b45399SDmitry Vyukov     pthread_t th[kBatch];
29*64b45399SDmitry Vyukov     for (int j = 0; j < kBatch; j++)
30*64b45399SDmitry Vyukov       pthread_create(&th[j], nullptr, nop_thread, nullptr);
31*64b45399SDmitry Vyukov     for (int j = 0; j < kBatch; j++)
32*64b45399SDmitry Vyukov       pthread_join(th[j], nullptr);
33*64b45399SDmitry Vyukov   }
34*64b45399SDmitry Vyukov   printf("starting persistent threads...\n");
35*64b45399SDmitry Vyukov   const int kLiveThreads = 2000;
36*64b45399SDmitry Vyukov   pthread_t th[kLiveThreads];
37*64b45399SDmitry Vyukov   for (int j = 0; j < kLiveThreads; j++)
38*64b45399SDmitry Vyukov     pthread_create(&th[j], nullptr, nop_thread, nullptr);
39*64b45399SDmitry Vyukov   printf("starting benchmark threads...\n");
40*64b45399SDmitry Vyukov   start_thread_group(bench_nthread, thread);
41*64b45399SDmitry Vyukov   for (int j = 0; j < kLiveThreads; j++)
42*64b45399SDmitry Vyukov     pthread_join(th[j], nullptr);
43*64b45399SDmitry Vyukov }
44*64b45399SDmitry Vyukov 
45*64b45399SDmitry Vyukov // CHECK: DONE
46