forked from N-BodyShop/changa
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathEwaldCUDA.h
70 lines (53 loc) · 1.6 KB
/
EwaldCUDA.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
#ifndef _EWALD_CUDA_H_
#define _EWALD_CUDA_H_
#include "HostCUDA.h"
/* defines for Hybrid API buffer indices */
#define PARTICLE_TABLE 0
#define EWALD_READ_ONLY_DATA 1
#define EWALD_TABLE 2
#define BUFFERS_PER_CHARE 3
#define NEWH 80
#define BLOCK_SIZE 128
#define NUM_GRAVITY_BUFS 10
/** @brief Data for the Ewald h loop in the CUDA kernel
*/
typedef struct {
float hx, hy, hz;
float hCfac, hSfac;
} EwtData;
/** @brief CUDA version of complete MultipoleMoments for Ewald
*/
typedef struct {
float xx, xy, xz, yy, yz, zz;
float totalMass;
float cmx, cmy, cmz;
} MultipoleMomentsData;
/** @brief Parameters and data for Ewald in the CUDA kernel
*/
typedef struct {
MultipoleMomentsData mm;
int n, nReps, nEwReps, nEwhLoop;
float L, fEwCut, alpha, alpha2, k1, ka, fEwCut2, fInner2;
} EwaldReadOnlyData;
/** @brief Particle data for the CUDA Ewald kernels
*/
typedef struct {
float position_x,position_y,position_z;
float acceleration_x, acceleration_y, acceleration_z;
float potential;
} GravityParticleData;
typedef struct {
GravityParticleData *p;
EwtData *ewt;
EwaldReadOnlyData *cachedData;
} EwaldData;
void EwaldHostMemorySetup(EwaldData *h_idata, int nParticles, int nEwhLoop);
void EwaldHostMemoryFree(EwaldData *h_idata);
#ifdef CUDA_INSTRUMENT_WRS
void EwaldHost(EwaldData *h_idata, void *cb, int myIndex, char phase);
#else
void EwaldHost(EwaldData *h_idata, void *cb, int myIndex);
#endif
__global__ void EwaldTopKernel(GravityParticleData *particleTable);
__global__ void EwaldBottomKernel(GravityParticleData *particleTable);
#endif