#include <gromacs/ewald/pme_pp_comm_gpu.h>
Manages communication related to GPU buffers between this PME rank and its PP rank.
|
class | Impl |
| Class with interfaces and data for CUDA version of PME-PP Communication. More...
|
|
gmx::PmePpCommGpu::PmePpCommGpu |
( |
MPI_Comm |
comm, |
|
|
int |
pmeRank |
|
) |
| |
Creates PME-PP GPU communication object.
- Parameters
-
[in] | comm | Communicator used for simulation |
[in] | pmeRank | Rank of PME task |
void gmx::PmePpCommGpu::receiveForceFromPmeCudaDirect |
( |
void * |
recvPtr, |
|
|
int |
recvSize, |
|
|
bool |
recvPmeForceToGpu |
|
) |
| |
Pull data from PME GPU directly using CUDA Memory copy.
- Parameters
-
[out] | recvPtr | Buffer to receive PME force data |
[in] | recvSize | Number of elements to receive |
[in] | recvPmeForceToGpu | Whether receive is to GPU, otherwise CPU |
void gmx::PmePpCommGpu::reinit |
( |
int |
size | ) |
|
Perform steps required when buffer size changes.
- Parameters
-
[in] | size | Number of elements in buffer |
void gmx::PmePpCommGpu::sendCoordinatesToPmeCudaDirect |
( |
void * |
sendPtr, |
|
|
int |
sendSize, |
|
|
bool |
sendPmeCoordinatesFromGpu, |
|
|
GpuEventSynchronizer * |
coordinatesReadyOnDeviceEvent |
|
) |
| |
Push coordinates buffer directly to GPU memory on PME task.
- Parameters
-
[in] | sendPtr | Buffer with coordinate data |
[in] | sendSize | Number of elements to send |
[in] | sendPmeCoordinatesFromGpu | Whether send is from GPU, otherwise CPU |
[in] | coordinatesReadyOnDeviceEvent | Event recorded when coordinates are available on device |
The documentation for this class was generated from the following file: