Gromacs
2019-rc1
|
#include "gmxpre.h"
#include "usergpuids.h"
#include <cctype>
#include <algorithm>
#include <sstream>
#include <string>
#include <vector>
#include "gromacs/gpu_utils/gpu_utils.h"
#include "gromacs/hardware/hw_info.h"
#include "gromacs/utility/exceptions.h"
#include "gromacs/utility/stringutil.h"
Defines routines for handling user-specified GPU IDs.
Functions | |
std::vector< int > | gmx::parseUserGpuIds (const std::string &gpuIdString) |
Parse a GPU ID string into a container describing the task types and associated device IDs. 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... | |