tweak: try simplest version, but notice it doesn't work
$ make time-linux Requires /bin/time, found on GNU/Linux systems Running 100x and taking avg time: OMP_NUM_THREADS=1 out/samples Time using 1 thread: 34.50ms Running 100x and taking avg time: OMP_NUM_THREADS=2 out/samples Time using 2 threads: 34.60ms Running 100x and taking avg time: OMP_NUM_THREADS=4 out/samples Time for 4 threads: 32.00ms Running 100x and taking avg time: OMP_NUM_THREADS=8 out/samples Time using 8 threads: 30.40ms Running 100x and taking avg time: OMP_NUM_THREADS=16 out/samples Time using 16 threads: 30.80ms
This commit is contained in:
parent
2a39a04c69
commit
3d3a3f0045
Binary file not shown.
|
@ -6,7 +6,7 @@
|
|||
|
||||
const float PI = 3.14159265358979323846;
|
||||
|
||||
#define N 1000000
|
||||
#define N_SAMPLES 1024 * 1000
|
||||
|
||||
//Array helpers
|
||||
|
||||
|
@ -81,7 +81,7 @@ float split_array_sum(float** meta_array, int length, int divided_into)
|
|||
|
||||
uint32_t xorshift32(uint32_t* seed)
|
||||
{
|
||||
// Algorithm "xor" from p. 4 of Marsaglia, "Xorshift RNGs"
|
||||
// Algorithm "xor" from p. 4 of Marsaglia, "Xorshift RN_SAMPLESGs"
|
||||
// See <https://stackoverflow.com/questions/53886131/how-does-xorshift32-works>
|
||||
// https://en.wikipedia.org/wiki/Xorshift
|
||||
// Also some drama: <https://www.pcg-random.org/posts/on-vignas-pcg-critique.html>, <https://prng.di.unimi.it/>
|
||||
|
@ -103,10 +103,10 @@ float rand_0_to_1(uint32_t* seed)
|
|||
x ^= x << 13;
|
||||
x ^= x >> 17;
|
||||
x ^= x << 5;
|
||||
return ((float)(*seed = x))/((float) UINT32_MAX);
|
||||
return ((float)(*seed = x))/((float) UIN_SAMPLEST32_MAX);
|
||||
*/
|
||||
// previously:
|
||||
// ((float)rand_r(seed) / (float)RAND_MAX)
|
||||
// ((float)rand_r(seed) / (float)RAN_SAMPLESD_MAX)
|
||||
// and before that: rand, but it wasn't thread-safe.
|
||||
// See: <https://stackoverflow.com/questions/43151361/how-to-create-thread-safe-random-number-generator-in-c-using-rand-r> for why to use rand_r:
|
||||
// rand() is not thread-safe, as it relies on (shared) hidden seed.
|
||||
|
@ -142,11 +142,11 @@ float random_lognormal(float logmean, float logsigma, uint32_t* seed)
|
|||
|
||||
float random_to(float low, float high, uint32_t* seed)
|
||||
{
|
||||
const float NORMAL95CONFIDENCE = 1.6448536269514722;
|
||||
const float N_SAMPLESORMAL95CON_SAMPLESFIDEN_SAMPLESCE = 1.6448536269514722;
|
||||
float loglow = logf(low);
|
||||
float loghigh = logf(high);
|
||||
float logmean = (loglow + loghigh) / 2;
|
||||
float logsigma = (loghigh - loglow) / (2.0 * NORMAL95CONFIDENCE);
|
||||
float logsigma = (loghigh - loglow) / (2.0 * N_SAMPLESORMAL95CON_SAMPLESFIDEN_SAMPLESCE);
|
||||
return random_lognormal(logmean, logsigma, seed);
|
||||
}
|
||||
|
||||
|
@ -179,8 +179,12 @@ float mixture(float (*samplers[])(uint32_t*), float* weights, int n_dists, uint3
|
|||
}
|
||||
|
||||
// Parallization function
|
||||
void paralellize(float (*sampler)(uint32_t* seed), float** results, int n_threads){
|
||||
|
||||
void paralellize(float (*sampler)(uint32_t* seed), float* results, int n_threads, int n_samples){
|
||||
if((N_SAMPLES % n_threads) != 0){
|
||||
fprintf(stderr, "Number of samples isn't divisible by number of threads, aborting\n");
|
||||
exit(1);
|
||||
}
|
||||
// int n_samples_per_thread = N_SAMPLES / n_thread;
|
||||
int sample_index, i, split_array_length;
|
||||
uint32_t** seeds = malloc(n_threads * sizeof(uint32_t*));
|
||||
for (uint32_t i = 0; i < n_threads; i++) {
|
||||
|
@ -188,13 +192,16 @@ void paralellize(float (*sampler)(uint32_t* seed), float** results, int n_thread
|
|||
*seeds[i] = i + 1; // xorshift can't start with 0
|
||||
}
|
||||
|
||||
#pragma omp parallel private(i, sample_index, split_array_length)
|
||||
#pragma omp parallelz private(i, sample_index, split_array_length)
|
||||
{
|
||||
#pragma omp for
|
||||
for (i = 0; i < n_threads; i++) {
|
||||
split_array_length = split_array_get_length(i, N, n_threads);
|
||||
for (int j = 0; j < split_array_length; j++) {
|
||||
results[i][j] = sampler(seeds[i]);
|
||||
// split_array_length = split_array_get_length(i, N_SAMPLES, n_threads);
|
||||
int lower_bound = i * (n_samples / n_threads);
|
||||
int upper_bound = ((i+1) * (n_samples / n_threads)) - 1;
|
||||
// printf("Lower bound: %d, upper bound: %d\n", lower_bound, upper_bound);
|
||||
for (int j = lower_bound; j < upper_bound; j++) {
|
||||
results[j] = sampler(seeds[i]);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -247,14 +254,11 @@ float sample_mixture(uint32_t* seed){
|
|||
int main()
|
||||
{
|
||||
int n_threads = omp_get_max_threads();
|
||||
// printf("Max threads: %d\n", n_threads);
|
||||
// omp_set_num_threads(n_threads);
|
||||
float** split_array_results = malloc(n_threads * sizeof(float*));
|
||||
split_array_allocate(split_array_results, N, n_threads);
|
||||
float* split_array_results = malloc(N_SAMPLES * sizeof(float));
|
||||
|
||||
paralellize(sample_mixture, split_array_results, n_threads);
|
||||
printf("Sum(split_array_results, N)/N = %f\n", split_array_sum(split_array_results, N, n_threads) / N);
|
||||
paralellize(sample_mixture, split_array_results, n_threads, N_SAMPLES);
|
||||
printf("Sum(split_array_results, N_SAMPLES)/N_SAMPLES = %f\n", array_sum(split_array_results, N_SAMPLES) / N_SAMPLES);
|
||||
|
||||
split_array_free(split_array_results, n_threads);
|
||||
free(split_array_results);
|
||||
return 0;
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue
Block a user