commit 1bc7c90fcfe0d61ce5f0e84c7e8e1a9127f497b0
parent 3d3a3f0045e248ebe2608ddf55c6fa5ae30e3b93
Author: NunoSempere <nuno.sempere@protonmail.com>
Date: Sat, 18 Nov 2023 22:59:03 +0000
tweak: savepoint; test hypothesis over previous stuff.
Diffstat:
1 file changed, 11 insertions(+), 4 deletions(-)
diff --git a/C/alt/05-refactor-split-array/samples.c b/C/alt/05-refactor-split-array/samples.c
@@ -180,19 +180,26 @@ 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, 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;
+ int n_samples_per_thread = N_SAMPLES / n_threads;
+
+ float** split_results = malloc(n_threads * sizeof(float*));
+ for(int i=0; i<n_threads; i++){
+ split_results[i] = malloc(n_samples_per_thread * sizeof(float));
+ }
+
uint32_t** seeds = malloc(n_threads * sizeof(uint32_t*));
for (uint32_t i = 0; i < n_threads; i++) {
seeds[i] = malloc(sizeof(uint32_t));
*seeds[i] = i + 1; // xorshift can't start with 0
}
-
- #pragma omp parallelz private(i, sample_index, split_array_length)
+
+ int i;
+ #pragma omp parallel private(i)
{
#pragma omp for
for (i = 0; i < n_threads; i++) {