Gromacs
2021.1
|
#include <gromacs/ewald/pme_force_sender_gpu.h>
Manages sending forces from PME-only ranks to their PP ranks.
Public Member Functions | |
PmeForceSenderGpu (const DeviceStream &pmeStream, MPI_Comm comm, gmx::ArrayRef< PpRanks > ppRanks) | |
Creates PME GPU Force sender object. More... | |
void | sendForceBufferAddressToPpRanks (rvec *d_f) |
Initialization of GPU PME Force sender. More... | |
void | sendFToPpCudaDirect (int ppRank) |
Send PP data to PP rank. More... | |
gmx::PmeForceSenderGpu::PmeForceSenderGpu | ( | const DeviceStream & | pmeStream, |
MPI_Comm | comm, | ||
gmx::ArrayRef< PpRanks > | ppRanks | ||
) |
Creates PME GPU Force sender object.
Constructor stub.
[in] | pmeStream | CUDA stream used for PME computations |
[in] | comm | Communicator used for simulation |
[in] | ppRanks | List of PP ranks |
void gmx::PmeForceSenderGpu::sendForceBufferAddressToPpRanks | ( | rvec * | d_f | ) |
Initialization of GPU PME Force sender.
init PME-PP GPU communication stub
[in] | d_f | force buffer in GPU memory |
void gmx::PmeForceSenderGpu::sendFToPpCudaDirect | ( | int | ppRank | ) |
Send PP data to PP rank.
[in] | ppRank | PP rank to receive data |