#include <cstddef>
#include <string>
#include <vector>
#include "gromacs/utility/arrayref.h"

Include dependency graph for usergpuids.h:

This graph shows which files directly or indirectly include this file:

Description

Declares routines for handling user-specified GPU IDs.

Author: Mark Abraham mark..nosp@m.j.ab.nosp@m.raham.nosp@m.@gma.nosp@m.il.co.nosp@m.m

Functions
std::vector< int >	gmx::parseUserGpuIdString (const std::string &gpuIdString)
	Parse a GPU ID specifier string into a container describing device IDs exposed to the run. More...

std::vector< int >	gmx::makeGpuIdsToUse (const gmx_gpu_info_t &gpuInfo, const std::string &gpuIdsAvailableString)
	Implement GPU ID selection by returning the available GPU IDs on this physical node that are compatible. More...

std::vector< int >	gmx::parseUserTaskAssignmentString (const std::string &gpuIdString)
	Parse a GPU ID specifier string into a container describing device ID to task mapping. More...

std::vector< int >	gmx::makeGpuIds (ArrayRef< const int > compatibleGpus, size_t numGpuTasks)
	Make a vector containing `numGpuTasks` IDs of the IDs found in `compatibleGpus`. More...

std::string	gmx::makeGpuIdString (const std::vector< int > &gpuIds, int totalNumberOfTasks)
	Convert a container of GPU deviced IDs to a string that can be used by gmx tune_pme as input to mdrun -gputasks. More...

void	gmx::checkUserGpuIds (const gmx_gpu_info_t &gpu_info, const std::vector< int > &compatibleGpus, const std::vector< int > &gpuIds)
	Check that all user-selected GPUs are compatible. More...

Functions