1*68d75effSDimitry Andric // Synthetic benchmark for __tsan_read/write{1,2,4,8}.
2*68d75effSDimitry Andric // As compared to mini_bench_local/shared.cc this benchmark passes through
3*68d75effSDimitry Andric // deduplication logic (ContainsSameAccess).
4*68d75effSDimitry Andric // First argument is access size (1, 2, 4, 8). Second optional arg switches
5*68d75effSDimitry Andric // from writes to reads.
6*68d75effSDimitry Andric
7*68d75effSDimitry Andric #include <pthread.h>
8*68d75effSDimitry Andric #include <stdlib.h>
9*68d75effSDimitry Andric #include <stdio.h>
10*68d75effSDimitry Andric #include <unistd.h>
11*68d75effSDimitry Andric #include <linux/futex.h>
12*68d75effSDimitry Andric #include <sys/syscall.h>
13*68d75effSDimitry Andric #include <sys/time.h>
14*68d75effSDimitry Andric
15*68d75effSDimitry Andric template<typename T, bool write>
thread(void * arg)16*68d75effSDimitry Andric void* thread(void *arg) {
17*68d75effSDimitry Andric const int kSize = 2 << 10;
18*68d75effSDimitry Andric static volatile long data[kSize];
19*68d75effSDimitry Andric static volatile long turn;
20*68d75effSDimitry Andric const int kRepeat = 1 << 17;
21*68d75effSDimitry Andric const int id = !!arg;
22*68d75effSDimitry Andric for (int i = 0; i < kRepeat; i++) {
23*68d75effSDimitry Andric for (;;) {
24*68d75effSDimitry Andric int t = __atomic_load_n(&turn, __ATOMIC_ACQUIRE);
25*68d75effSDimitry Andric if (t == id)
26*68d75effSDimitry Andric break;
27*68d75effSDimitry Andric syscall(SYS_futex, &turn, FUTEX_WAIT, t, 0, 0, 0);
28*68d75effSDimitry Andric }
29*68d75effSDimitry Andric for (int j = 0; j < kSize; j++) {
30*68d75effSDimitry Andric if (write) {
31*68d75effSDimitry Andric ((volatile T*)&data[j])[0] = 1;
32*68d75effSDimitry Andric ((volatile T*)&data[j])[sizeof(T) == 8 ? 0 : 1] = 1;
33*68d75effSDimitry Andric } else {
34*68d75effSDimitry Andric T v0 = ((volatile T*)&data[j])[0];
35*68d75effSDimitry Andric T v1 = ((volatile T*)&data[j])[sizeof(T) == 8 ? 0 : 1];
36*68d75effSDimitry Andric (void)v0;
37*68d75effSDimitry Andric (void)v1;
38*68d75effSDimitry Andric }
39*68d75effSDimitry Andric }
40*68d75effSDimitry Andric __atomic_store_n(&turn, 1 - id, __ATOMIC_RELEASE);
41*68d75effSDimitry Andric syscall(SYS_futex, &turn, FUTEX_WAKE, 0, 0, 0, 0);
42*68d75effSDimitry Andric }
43*68d75effSDimitry Andric return 0;
44*68d75effSDimitry Andric }
45*68d75effSDimitry Andric
46*68d75effSDimitry Andric template<typename T, bool write>
test()47*68d75effSDimitry Andric void test() {
48*68d75effSDimitry Andric pthread_t th;
49*68d75effSDimitry Andric pthread_create(&th, 0, thread<T, write>, (void*)1);
50*68d75effSDimitry Andric thread<T, write>(0);
51*68d75effSDimitry Andric pthread_join(th, 0);
52*68d75effSDimitry Andric }
53*68d75effSDimitry Andric
54*68d75effSDimitry Andric template<bool write>
testw(int size)55*68d75effSDimitry Andric void testw(int size) {
56*68d75effSDimitry Andric switch (size) {
57*68d75effSDimitry Andric case 1: return test<char, write>();
58*68d75effSDimitry Andric case 2: return test<short, write>();
59*68d75effSDimitry Andric case 4: return test<int, write>();
60*68d75effSDimitry Andric case 8: return test<long long, write>();
61*68d75effSDimitry Andric }
62*68d75effSDimitry Andric }
63*68d75effSDimitry Andric
main(int argc,char ** argv)64*68d75effSDimitry Andric int main(int argc, char** argv) {
65*68d75effSDimitry Andric int size = 8;
66*68d75effSDimitry Andric bool write = true;
67*68d75effSDimitry Andric if (argc > 1) {
68*68d75effSDimitry Andric size = atoi(argv[1]);
69*68d75effSDimitry Andric if (size != 1 && size != 2 && size != 4 && size != 8)
70*68d75effSDimitry Andric size = 8;
71*68d75effSDimitry Andric }
72*68d75effSDimitry Andric if (argc > 2)
73*68d75effSDimitry Andric write = false;
74*68d75effSDimitry Andric printf("%s%d\n", write ? "write" : "read", size);
75*68d75effSDimitry Andric if (write)
76*68d75effSDimitry Andric testw<true>(size);
77*68d75effSDimitry Andric else
78*68d75effSDimitry Andric testw<false>(size);
79*68d75effSDimitry Andric return 0;
80*68d75effSDimitry Andric }
81