Executor¶
Overview¶
NeoFOAM uses the MPI+X approach for parallelism, where X is the execution space used for device parallelism. The Executor class uses Kokkos and provides an interface for memory management and specifics were to execute the operations:
CPUExecutor: run on the CPU with MPI
OMPExecutor: run on the CPU with OpenMP and MPI
GPUExecutor: run on the GPU with MPI
Design¶
One of the design goals is the ability to easily switch between different execution spaces at run time, i.e. enabling the ability to switch between CPU and GPU execution without having to recompile NeoFOAM. This is achieved by passing the executor as an argument to the container types like Field as shown below.
NeoFOAM::GPUExecutor gpuExec {};
NeoFOAM::CPUExecutor cpuExec {};
NeoFOAM::Field<NeoFOAM::scalar> GPUField(gpuExec, 10);
NeoFOAM::Field<NeoFOAM::scalar> CPUField(cpuExec, 10);
The Executor is a std::variant
using executor = std::variant<OMPExecutor, GPUExecutor, CPUExecutor>;
and allows to switch between the different strategies for memory allocation and execution at runtime. We use std::visit to switch between the different strategies:
NeoFOAM::CPUExecutor exec{};
std::visit([&](const auto& exec)
{ Functor(exec); },
exec);
that are provided by a functor
struct Functor
{
void operator()(const CPUExecutor& exec)
{
std::cout << "CPUExecutor" << std::endl;
}
void operator()(const OMPExecutor& exec)
{
std::cout << "OMPExecutor" << std::endl;
}
void operator()(const GPUExecutor& exec)
{
std::cout << "GPUExecutor" << std::endl;
}
};
The visit pattern with the above functor would print different messages depending on the executor type. To extend the library with the additional features the above functor design should be used for the different implementations.
One can check that two operators are ‘of the same type’, i.e. execute in the same execution space using the equality operators == and !=.