StarPU Handbook
starpu_rand.h
Go to the documentation of this file.
1 /* StarPU --- Runtime system for heterogeneous multicore architectures.
2  *
3  * Copyright (C) 2012 Inria
4  * Copyright (C) 2012-2013,2015,2017 CNRS
5  * Copyright (C) 2015-2016 Université de Bordeaux
6  *
7  * StarPU is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU Lesser General Public License as published by
9  * the Free Software Foundation; either version 2.1 of the License, or (at
10  * your option) any later version.
11  *
12  * StarPU is distributed in the hope that it will be useful, but
13  * WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
15  *
16  * See the GNU Lesser General Public License in COPYING.LGPL for more details.
17  */
18 
19 #ifndef __STARPU_RAND_H__
20 #define __STARPU_RAND_H__
21 
22 #include <stdlib.h>
23 #include <starpu_config.h>
24 
25 #ifdef __cplusplus
26 extern "C"
27 {
28 #endif
29 
30 #ifdef STARPU_SIMGRID
31 /* In simgrid mode, force using seed 0 by default to get reproducible behavior by default */
32 #define starpu_seed(seed) starpu_get_env_number_default("STARPU_RAND_SEED", 0)
33 #else
34 #define starpu_seed(seed) starpu_get_env_number_default("STARPU_RAND_SEED", (seed))
35 #endif
36 
37 #ifdef STARPU_USE_DRAND48
38 # define starpu_srand48(seed) srand48(starpu_seed(seed))
39 # define starpu_drand48() drand48()
40 # define starpu_lrand48() lrand48()
41 # define starpu_erand48(xsubi) erand48(xsubi)
42 # ifdef STARPU_USE_ERAND48_R
43 typedef struct drand48_data starpu_drand48_data;
44 # define starpu_srand48_r(seed, buffer) srand48_r(starpu_seed(seed), buffer)
45 # define starpu_drand48_r(buffer, result) drand48_r(buffer, result)
46 # define starpu_lrand48_r(buffer, result) lrand48_r(buffer, result)
47 # define starpu_erand48_r(xsubi, buffer, result) erand48_r(xsubi, buffer, result)
48 #else
49 typedef int starpu_drand48_data;
50 # define starpu_srand48_r(seed, buffer) srand48(starpu_seed(seed))
51 # define starpu_drand48_r(buffer, result) do {*(result) = drand48(); } while (0)
52 # define starpu_lrand48_r(buffer, result) do {*(result) = lrand48(); } while (0)
53 # define starpu_erand48_r(xsubi, buffer, result) do {(void) buffer; *(result) = erand48(xsubi); } while (0)
54 # endif
55 #else
56 typedef int starpu_drand48_data;
57 # define starpu_srand48(seed) srand(starpu_seed(seed))
58 # define starpu_drand48() (double)(rand()) / RAND_MAX
59 # define starpu_lrand48() rand()
60 # define starpu_erand48(xsubi) starpu_drand48()
61 # define starpu_srand48_r(seed, buffer) srand(starpu_seed(seed))
62 # define starpu_erand48_r(xsubi, buffer, result) do {(void) xsubi; (void) buffer; *(result) = ((double)(rand()) / RAND_MAX);} while (0)
63 #endif
64 
65 #ifdef __cplusplus
66 }
67 #endif
68 
69 #endif /* __STARPU_RAND_H__ */