Gromacs
2021-beta2-UNCHECKED
|
#include <gromacs/ewald/pme_force_sender_gpu_impl.h>
Class with interfaces and data for CUDA version of PME Force sending functionality.
Impl class stub.
Public Member Functions | |
Impl (const DeviceStream &pmeStream, MPI_Comm comm, gmx::ArrayRef< PpRanks > ppRanks) | |
Creates PME GPU Force sender object. More... | |
void | sendForceBufferAddressToPpRanks (rvec *d_f) |
sends force buffer address to PP rank More... | |
void | sendFToPpCudaDirect (int ppRank) |
Send PP data to PP rank. More... | |
gmx::PmeForceSenderGpu::Impl::Impl | ( | const DeviceStream & | pmeStream, |
MPI_Comm | comm, | ||
gmx::ArrayRef< PpRanks > | ppRanks | ||
) |
Creates PME GPU Force sender object.
[in] | pmeStream | CUDA stream used for PME computations |
[in] | comm | Communicator used for simulation |
[in] | ppRanks | List of PP ranks |
void gmx::PmeForceSenderGpu::Impl::sendForceBufferAddressToPpRanks | ( | rvec * | d_f | ) |
sends force buffer address to PP rank
[in] | d_f | force buffer in GPU memory |
void gmx::PmeForceSenderGpu::Impl::sendFToPpCudaDirect | ( | int | ppRank | ) |
Send PP data to PP rank.
[in] | ppRank | PP rank to receive data |