Gromacs
2019
|
#include "gmxpre.h"
#include "findallgputasks.h"
#include "config.h"
#include <numeric>
#include <vector>
#include "gromacs/utility/exceptions.h"
#include "gromacs/utility/gmxassert.h"
#include "gromacs/utility/gmxmpi.h"
#include "gromacs/utility/physicalnodecommunicator.h"
Defines routine for collecting all GPU tasks found on ranks of a node.
Functions | |
std::vector< int > | gmx::anonymous_namespace{findallgputasks.cpp}::allgather (const int &input, int numRanks, MPI_Comm communicator) |
Helper function to prepare to all-gather the vector of non-bonded tasks on this node. | |
std::vector< int > | gmx::anonymous_namespace{findallgputasks.cpp}::computeDisplacements (ArrayRef< const int > extentOnEachRank, int numRanks) |
Helper function to compute allgatherv displacements. | |
std::vector< GpuTask > | gmx::anonymous_namespace{findallgputasks.cpp}::allgatherv (ArrayRef< const GpuTask > input, ArrayRef< const int > extentOnEachRank, ArrayRef< const int > displacementForEachRank, MPI_Comm communicator) |
Helper function to all-gather the vector of all GPU tasks on ranks of this node. | |
GpuTasksOnRanks | gmx::findAllGpuTasksOnThisNode (ArrayRef< const GpuTask > gpuTasksOnThisRank, const PhysicalNodeCommunicator &physicalNodeComm) |
Returns container of all tasks on all ranks of this node that are eligible for GPU execution. More... | |
Variables | |
constexpr bool | gmx::anonymous_namespace{findallgputasks.cpp}::g_usingMpi = GMX_MPI |
Constant used to help minimize preprocessing of code. | |