#define GMX_PME_PP_COMM_GPU_IMPL_H
#include "gromacs/ewald/pme_pp_comm_gpu.h"
-#include "gromacs/gpu_utils/devicebuffer_datatype.h"
#include "gromacs/gpu_utils/gpueventsynchronizer.cuh"
#include "gromacs/math/vectypes.h"
#include "gromacs/utility/gmxmpi.h"
/*! \brief
* Return pointer to buffer used for staging PME force on GPU
*/
- DeviceBuffer<gmx::RVec> getGpuForceStagingPtr();
+ void* getGpuForceStagingPtr();
/*! \brief
* Return pointer to event recorded when forces are ready
//! Handle for CUDA stream used for the communication operations in this class
const DeviceStream& pmePpCommStream_;
//! Remote location of PME coordinate data buffer
- DeviceBuffer<gmx::RVec> remotePmeXBuffer_ = nullptr;
+ void* remotePmeXBuffer_ = nullptr;
//! Remote location of PME force data buffer
- DeviceBuffer<gmx::RVec> remotePmeFBuffer_ = nullptr;
+ void* remotePmeFBuffer_ = nullptr;
//! communicator for simulation
MPI_Comm comm_;
//! Rank of PME task
int pmeRank_ = -1;
//! Buffer for staging PME force on GPU
- DeviceBuffer<gmx::RVec> d_pmeForces_ = nullptr;
+ rvec* d_pmeForces_ = nullptr;
//! number of atoms in PME force staging array
int d_pmeForcesSize_ = -1;
//! number of atoms allocated in recvbuf array