test_random.cpp

Go to the documentation of this file.

/*
# =============================================================================
# Copyright (c) 2016 - 2024 Blue Brain Project/EPFL
#
# See top-level LICENSE file for details.
# =============================================================================.
*/
#include <catch2/catch_test_macros.hpp>
#include <set>
#include "coreneuron/utils/randoms/nrnran123.h"
 
using namespace coreneuron;
 
TEST_CASE("random123 smoke test") {
    const int SEED_KEY = 1;
    const int NUM_STREAMS = 20;
    const int NUM_SAMPLES = 1000;
    nrnran123_State* rand_streams[NUM_STREAMS];
 
    const int res_size = NUM_SAMPLES * NUM_STREAMS;
    double res[res_size];
 
    for (int i = 0; i < NUM_STREAMS; i++) {
        rand_streams[i] = nrnran123_newstream(SEED_KEY, i);
        nrnran123_setseq(rand_streams[i], 0, 0);
    }
 
    nrn_pragma_omp(target teams distribute parallel for map(tofrom: res[0:res_size]) map(to: rand_streams[0:NUM_STREAMS]))
    nrn_pragma_acc(parallel loop copy(res [0:res_size]) copyin(rand_streams [0:NUM_STREAMS]))
    for (int i = 0; i < NUM_STREAMS; i++) {
        for (int j = 0; j < NUM_SAMPLES; j++) {
            double val = nrnran123_dblpick(rand_streams[i]);
            res[i * NUM_SAMPLES + j] = val;
        }
    }
 
    // there should be no duplicates
 
    std::set<double> check_set;
 
    for (int i = 0; i < NUM_STREAMS * NUM_SAMPLES; i++) {
        double d = res[i];
        size_t old_size = check_set.size();
        check_set.insert(res[i]);
        size_t new_size = check_set.size();
 
        if (old_size == new_size) {
            std::cerr << "Duplicate found! i = " << i << ", d = " << d << std::endl;
        }
    }
    REQUIRE(check_set.size() == NUM_SAMPLES * NUM_STREAMS);
}