Gromacs
2022.2
|
Functions | |
std::vector< GpuTaskAssignment > | buildTaskAssignment (const GpuTasksOnRanks &gpuTasksOnRanksOfThisNode, ArrayRef< const int > gpuIds) |
Build data structure of types of GPU tasks on a rank, together with the mapped GPU device IDs, for all GPU tasks on all the ranks of this node. More... | |
bool | isAnyGpuSharedBetweenRanks (ArrayRef< const GpuTaskAssignment > gpuTaskAssignments) |
Return whether a GPU device is shared between any ranks. More... | |
size_t | countGpuTasksOnThisNode (const GpuTasksOnRanks &gpuTasksOnRanksOfThisNode) |
Counts all the GPU tasks on this node. | |
int | countOverAllRanks (MPI_Comm comm, int countOnThisRank) |
Return on each rank the total count over all ranks of all simulations. | |
void | barrierOverAllRanks (MPI_Comm comm) |
Barrier over all rank in comm . | |
std::vector<GpuTaskAssignment> gmx::anonymous_namespace{taskassignment.cpp}::buildTaskAssignment | ( | const GpuTasksOnRanks & | gpuTasksOnRanksOfThisNode, |
ArrayRef< const int > | gpuIds | ||
) |
Build data structure of types of GPU tasks on a rank, together with the mapped GPU device IDs, for all GPU tasks on all the ranks of this node.
[in] | gpuTasksOnRanksOfThisNode | For each rank on this node, the set of tasks that are eligible to run on GPUs. |
[in] | gpuIds | The user-supplied GPU IDs. |
bool gmx::anonymous_namespace{taskassignment.cpp}::isAnyGpuSharedBetweenRanks | ( | ArrayRef< const GpuTaskAssignment > | gpuTaskAssignments | ) |
Return whether a GPU device is shared between any ranks.
Sharing GPUs among multiple ranks is possible via either user or automated selection.