Gromacs
2021-beta3-UNCHECKED
|
#include <gromacs/ewald/pme_coordinate_receiver_gpu_impl.h>
Class with interfaces and data for CUDA version of PME coordinate receiving functionality.
Impl class stub.
Public Member Functions | |
Impl (const DeviceStream &pmeStream, MPI_Comm comm, gmx::ArrayRef< PpRanks > ppRanks) | |
Creates PME GPU coordinate receiver object. More... | |
void | sendCoordinateBufferAddressToPpRanks (DeviceBuffer< RVec > d_x) |
send coordinates buffer address to PP rank More... | |
void | launchReceiveCoordinatesFromPpCudaDirect (int ppRank) |
launch receive of coordinate data from PP rank More... | |
void | enqueueWaitReceiveCoordinatesFromPpCudaDirect () |
enqueue wait for coordinate data from PP ranks | |
gmx::PmeCoordinateReceiverGpu::Impl::Impl | ( | const DeviceStream & | pmeStream, |
MPI_Comm | comm, | ||
gmx::ArrayRef< PpRanks > | ppRanks | ||
) |
Creates PME GPU coordinate receiver object.
[in] | pmeStream | CUDA stream used for PME computations |
[in] | comm | Communicator used for simulation |
[in] | ppRanks | List of PP ranks |
void gmx::PmeCoordinateReceiverGpu::Impl::launchReceiveCoordinatesFromPpCudaDirect | ( | int | ppRank | ) |
launch receive of coordinate data from PP rank
[in] | ppRank | PP rank to send data |
void gmx::PmeCoordinateReceiverGpu::Impl::sendCoordinateBufferAddressToPpRanks | ( | DeviceBuffer< RVec > | d_x | ) |
send coordinates buffer address to PP rank
[in] | d_x | coordinates buffer in GPU memory |