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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
/******************************************************************************
* Copyright (c) Intel Corporation - All rights reserved. *
* This file is part of the LIBXSMM library. *
* *
* For information on the license, see the LICENSE file. *
* Further information: https://github.com/hfp/libxsmm/ *
* SPDX-License-Identifier: BSD-3-Clause *
******************************************************************************/
/* Alexander Heinecke (Intel Corp.), Hans Pabst (Intel Corp.)
******************************************************************************/
#include <libxsmm_timer.h>
#include <libxsmm_sync.h>
#include <math.h>
#if defined(_OPENMP)
# include <omp.h>
#endif
int main(int argc, char* argv[])
{
int num_cores, threads_per_core, num_threads, num_iterations;
libxsmm_timer_tickint start;
libxsmm_barrier* barrier;
if (4 < argc) {
fprintf(stderr, "Usage:\n %s <cores> <threads-per-core> [<iterations>]\n", argv[0]);
return EXIT_SUCCESS;
}
/* parse the command line and set up the test parameters */
#if defined(_OPENMP)
num_cores = (1 < argc ? atoi(argv[1]) : 2);
assert(num_cores >= 1);
threads_per_core = (2 < argc ? atoi(argv[2]) : 2);
assert(threads_per_core >= 1);
#else
threads_per_core = 1;
num_cores = 1;
#endif
num_iterations = (1 < argc ? atoi(argv[3]) : 50000);
assert(num_iterations > 0);
/* create a new barrier */
barrier = libxsmm_barrier_create(num_cores, threads_per_core);
assert(NULL != barrier);
/* each thread must initialize with the barrier */
num_threads = num_cores * threads_per_core;
#if defined(_OPENMP)
# pragma omp parallel num_threads(num_threads)
#endif
{
#if defined(_OPENMP)
const int tid = omp_get_thread_num();
#else
const int tid = 0;
#endif
libxsmm_barrier_init(barrier, tid);
}
start = libxsmm_timer_tick();
#if defined(_OPENMP)
# pragma omp parallel num_threads(num_threads)
#endif
{
#if defined(_OPENMP)
const int tid = omp_get_thread_num();
#else
const int tid = 0;
#endif
int i;
for (i = 0; i < num_iterations; ++i) {
libxsmm_barrier_wait(barrier, tid);
}
}
printf("libxsmm_barrier_wait(): %llu cycles (%d threads)\n",
libxsmm_timer_ncycles(start, libxsmm_timer_tick()) / num_iterations,
num_threads);
libxsmm_barrier_destroy(barrier);
return EXIT_SUCCESS;
}
|
