KOKO\Mimi
29 lines
884 B
C++
29 lines
884 B
C++
#include "fesa/assembly/parallel_for.h"
|
|
|
|
#include <oneapi/tbb/parallel_for.h>
|
|
|
|
namespace fesa {
|
|
|
|
void SerialParallelFor::Execute(
|
|
std::size_t count, const std::function<void(std::size_t)>& body) const {
|
|
for (std::size_t index = 0; index < count; ++index) {
|
|
body(index);
|
|
}
|
|
}
|
|
|
|
void TbbParallelFor::Execute(
|
|
std::size_t count, const std::function<void(std::size_t)>& body) const {
|
|
if (count == 0U) {
|
|
return;
|
|
}
|
|
|
|
// Use oneTBB's caller-scoped scheduler policy. This adapter does not set
|
|
// process-wide concurrency or override the later MKL/TBB oversubscription
|
|
// policy. A body exception cancels sibling tasks and is rethrown; work
|
|
// already running during cancellation may still finish its indexed slot.
|
|
oneapi::tbb::parallel_for(std::size_t{0}, count,
|
|
[&body](std::size_t index) { body(index); });
|
|
}
|
|
|
|
} // namespace fesa
|