| 226 | } |
| 227 | |
| 228 | void init_io_random(int seed, int num_io_RNGs) |
| 229 | { |
| 230 | int seed_base = seed; |
| 231 | if (seed == -1) { |
| 232 | // Seed with a random value. |
| 233 | std::random_device rd; |
| 234 | seed_base = rd(); |
| 235 | } |
| 236 | |
| 237 | ::io_generators.resize(num_io_RNGs); |
| 238 | for (int i = 0; i < num_io_RNGs; i++) { |
| 239 | auto& io_rng = ::io_generators[i]; |
| 240 | io_rng.generator.seed(hash_combine(seed_base, i)); |
| 241 | io_rng.fast_generator.seed(hash_combine(seed_base, i)); |
| 242 | io_rng.active_thread_id.store(std::thread::id()); |
| 243 | } |
| 244 | ::io_generators_inited = true; |
| 245 | } |
| 246 | |
| 247 | } // namespace lbann |
no test coverage detected