1 #if defined(NAMD_CUDA) || defined(NAMD_HIP)
3 #include <hip/hip_runtime.h>
13 #define CUDA_PME_CHARGES_PROTOTYPE \
14 void cuda_pme_charges( \
15 const float *coeffs, \
16 float * const *q_arr, int *f_arr, int *fz_arr, \
17 float *a_data, int n_atoms, \
18 int K1, int K2, int K3, \
19 int order, cudaStream_t stream)
24 #define CUDA_PME_CHARGES_BATCHED_PROTOTYPE \
25 void cuda_pme_charges_batched( \
26 const float *coeffs, \
27 float * const *q_arr, int *f_arr, int *fz_arr, \
28 float **a_data_ptr, int* n_atoms_ptr, \
29 int* K1_ptr, int* K2_ptr, int* K3_ptr, \
30 int order, int numPatches, int n_max_atoms, cudaStream_t stream)
35 #define CUDA_PME_FORCES_PROTOTYPE \
36 void cuda_pme_forces( \
37 const float *coeffs, \
38 float * const *q_arr, \
39 float * const *afn, int dimy, int maxn, \
41 int K1, int K2, int K3, \
42 int order, cudaStream_t stream)
#define CUDA_PME_FORCES_PROTOTYPE
#define CUDA_PME_CHARGES_BATCHED_PROTOTYPE
#define CUDA_PME_CHARGES_PROTOTYPE
void cuda_init_bspline_coeffs(float **c, float **dc, int order)