parallel_for_queued.cpp

Go to the documentation of this file.

#ifndef NO_MULTITHREADING
#include "log.hpp"
#include "thread.hpp"
#include "thread_pool.hpp"
#include <atomic>
#include <condition_variable>
#include <functional>
#include <mutex>
#include <queue>
#include <thread>
#include <vector>
 
namespace bb {
void parallel_for_queued(size_t num_iterations, const std::function<void(size_t)>& func)
{
    static ThreadPool pool(get_num_cpus());
 
    // info("wait for pool enter");
    pool.wait();
    for (size_t i = 0; i < num_iterations; ++i) {
        // info("enqueing iteration ", i);
        pool.enqueue([=]() { func(i); });
    }
    // info("wait for pool exit");
    pool.wait();
    // info("pool finished work");
}
} // namespace bb
#endif