Gromacs
2021-beta2-UNCHECKED
|
#include <cstdio>
#include <string>
#include <vector>
#include "gromacs/gpu_utils/gpu_macros.h"
#include "gromacs/utility/basedefinitions.h"
Declare functions for detection and initialization for GPU devices.
Enumerations | |
enum | GpuApiCallBehavior { Sync, Async } |
Enum which is only used to describe transfer calls at the moment. | |
enum | GpuTaskCompletion { Wait, Check } |
Types of actions associated to waiting or checking the completion of GPU tasks. | |
Functions | |
bool | buildSupportsNonbondedOnGpu (std::string *error) |
Check if GROMACS has been built with GPU support. More... | |
void | startGpuProfiler () |
Starts the GPU profiler if mdrun is being profiled. More... | |
void | resetGpuProfiler () |
Resets the GPU profiler if mdrun is being profiled. More... | |
void | stopGpuProfiler () |
Stops the CUDA profiler if mdrun is being profiled. More... | |
bool | isHostMemoryPinned (const void *h_ptr) |
Tells whether the host buffer was pinned for non-blocking transfers. Only implemented for CUDA. | |
void | setupGpuDevicePeerAccess (const std::vector< int > &gpuIdsToUse, const gmx::MDLogger &mdlog) |
Enable peer access between GPUs where supported. More... | |
bool buildSupportsNonbondedOnGpu | ( | std::string * | error | ) |
Check if GROMACS has been built with GPU support.
[in] | error | Pointer to error string or nullptr. |
void resetGpuProfiler | ( | ) |
Resets the GPU profiler if mdrun is being profiled.
When a profiler run is in progress (based on the presence of the NVPROF_ID env. var.), the profiler data is restet in order to eliminate the data collected from the preceding part fo the run.
This function should typically be called at the mdrun counter reset time.
Note that this is implemented only for the CUDA API.
void setupGpuDevicePeerAccess | ( | const std::vector< int > & | gpuIdsToUse, |
const gmx::MDLogger & | mdlog | ||
) |
Enable peer access between GPUs where supported.
[in] | gpuIdsToUse | List of GPU IDs in use |
[in] | mdlog | Logger object |
void startGpuProfiler | ( | ) |
Starts the GPU profiler if mdrun is being profiled.
When a profiler run is in progress (based on the presence of the NVPROF_ID env. var.), the profiler is started to begin collecting data during the rest of the run (or until stopGpuProfiler is called).
Note that this is implemented only for the CUDA API.
void stopGpuProfiler | ( | ) |
Stops the CUDA profiler if mdrun is being profiled.
This function can be called at cleanup when skipping recording recording subsequent API calls from being traces/profiled is desired, e.g. before uninitialization.
Note that this is implemented only for the CUDA API.