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> 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> 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> 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 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