1 /* Copyright (c) 2012-2014. The SimGrid Team.
2 * All rights reserved. */
4 /* This program is free software; you can redistribute it and/or modify it
5 * under the terms of the license (GNU LGPL) which comes with this package. */
10 #include <xbt/parmap.h>
11 #include <xbt/sysdep.h>
12 #include "src/internal_config.h" /* HAVE_FUTEX_H */
13 #include "simgrid/simix.h"
14 #include "xbt/xbt_os_time.h"
16 #define MODES_DEFAULT 0x7
18 #define ARRAY_SIZE 10007
21 void (*fun_to_apply)(void *);
23 static const char *parmap_mode_name(e_xbt_parmap_mode_t mode)
27 case XBT_PARMAP_POSIX:
28 snprintf(name, sizeof name, "POSIX");
30 case XBT_PARMAP_FUTEX:
31 snprintf(name, sizeof name, "FUTEX");
33 case XBT_PARMAP_BUSY_WAIT:
34 snprintf(name, sizeof name, "BUSY_WAIT");
36 case XBT_PARMAP_DEFAULT:
37 snprintf(name, sizeof name, "DEFAULT");
40 snprintf(name, sizeof name, "UNKNOWN(%d)", (int)mode);
46 static int parmap_skip_mode(e_xbt_parmap_mode_t mode)
50 case XBT_PARMAP_FUTEX:
51 printf("not available\n");
59 static unsigned fibonacci(unsigned n)
64 return fibonacci(n - 1) + fibonacci(n - 2);
67 static void fun_small_comp(void *arg)
73 static void fun_big_comp(void *arg)
76 *u = fibonacci(*u % FIBO_MAX);
79 static void array_new(unsigned **a, xbt_dynar_t *data)
82 *a = xbt_malloc(ARRAY_SIZE * sizeof **a);
83 *data = xbt_dynar_new(sizeof *a, NULL);
84 xbt_dynar_shrink(*data, ARRAY_SIZE);
85 for (i = 0 ; i < ARRAY_SIZE ; i++) {
87 xbt_dynar_push_as(*data, void*, &(*a)[i]);
91 static void bench_parmap_full(int nthreads, e_xbt_parmap_mode_t mode)
97 double start_time, elapsed_time;
99 printf("** mode = %-15s ", parmap_mode_name(mode));
102 if (parmap_skip_mode(mode))
105 array_new(&a, &data);
108 start_time = xbt_os_time();
110 parmap = xbt_parmap_new(nthreads, mode);
111 xbt_parmap_apply(parmap, fun_to_apply, data);
112 xbt_parmap_destroy(parmap);
113 elapsed_time = xbt_os_time() - start_time;
115 } while (elapsed_time < TIMEOUT);
117 printf("ran %d times in %g seconds (%g/s)\n", i, elapsed_time, i / elapsed_time);
119 xbt_dynar_free(&data);
123 static void bench_parmap_apply(int nthreads, e_xbt_parmap_mode_t mode)
127 double start_time, elapsed_time;
129 printf("** mode = %-15s ", parmap_mode_name(mode));
132 if (parmap_skip_mode(mode))
135 array_new(&a, &data);
137 xbt_parmap_t parmap = xbt_parmap_new(nthreads, mode);
139 start_time = xbt_os_time();
141 xbt_parmap_apply(parmap, fun_to_apply, data);
142 elapsed_time = xbt_os_time() - start_time;
144 } while (elapsed_time < TIMEOUT);
145 xbt_parmap_destroy(parmap);
147 printf("ran %d times in %g seconds (%g/s)\n",
148 i, elapsed_time, i / elapsed_time);
150 xbt_dynar_free(&data);
154 static void bench_all_modes(void (*bench_fun)(int, e_xbt_parmap_mode_t),
155 int nthreads, unsigned modes)
157 e_xbt_parmap_mode_t all_modes[] = {XBT_PARMAP_POSIX, XBT_PARMAP_FUTEX, XBT_PARMAP_BUSY_WAIT, XBT_PARMAP_DEFAULT};
159 for (unsigned i = 0 ; i < sizeof all_modes / sizeof all_modes[0] ; i++) {
161 bench_fun(nthreads, all_modes[i]);
165 int main(int argc, char *argv[])
168 unsigned modes = MODES_DEFAULT;
170 SIMIX_global_init(&argc, argv);
172 if (argc != 2 && argc != 3) {
173 fprintf(stderr, "Usage: %s nthreads [modes]\n"
174 " nthreads - number of working threads\n"
175 " modes - bitmask of modes to test\n",
179 nthreads = atoi(argv[1]);
181 fprintf(stderr, "ERROR: invalid thread count: %d\n", nthreads);
185 modes = strtol(argv[2], NULL, 0);
187 printf("Parmap benchmark with %d workers (modes = %#x)...\n\n", nthreads, modes);
189 fun_to_apply = fun_small_comp;
191 printf("Benchmark for parmap create+apply+destroy (small comp):\n");
192 bench_all_modes(bench_parmap_full, nthreads, modes);
195 printf("Benchmark for parmap apply only (small comp):\n");
196 bench_all_modes(bench_parmap_apply, nthreads, modes);
199 fun_to_apply = fun_big_comp;
201 printf("Benchmark for parmap create+apply+destroy (big comp):\n");
202 bench_all_modes(bench_parmap_full, nthreads, modes);
205 printf("Benchmark for parmap apply only (big comp):\n");
206 bench_all_modes(bench_parmap_apply, nthreads, modes);