128 lines
6.3 KiB
C
128 lines
6.3 KiB
C
#ifndef BSSN_RHS_CUDA_H
|
|
#define BSSN_RHS_CUDA_H
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
enum {
|
|
BSSN_CUDA_STATE_COUNT = 24,
|
|
BSSN_CUDA_MATTER_COUNT = 10
|
|
};
|
|
|
|
int f_compute_rhs_bssn(int *ex, double &T,
|
|
double *X, double *Y, double *Z,
|
|
double *chi, double *trK,
|
|
double *dxx, double *gxy, double *gxz, double *dyy, double *gyz, double *dzz,
|
|
double *Axx, double *Axy, double *Axz, double *Ayy, double *Ayz, double *Azz,
|
|
double *Gamx, double *Gamy, double *Gamz,
|
|
double *Lap, double *betax, double *betay, double *betaz,
|
|
double *dtSfx, double *dtSfy, double *dtSfz,
|
|
double *chi_rhs, double *trK_rhs,
|
|
double *gxx_rhs, double *gxy_rhs, double *gxz_rhs, double *gyy_rhs, double *gyz_rhs, double *gzz_rhs,
|
|
double *Axx_rhs, double *Axy_rhs, double *Axz_rhs, double *Ayy_rhs, double *Ayz_rhs, double *Azz_rhs,
|
|
double *Gamx_rhs, double *Gamy_rhs, double *Gamz_rhs,
|
|
double *Lap_rhs, double *betax_rhs, double *betay_rhs, double *betaz_rhs,
|
|
double *dtSfx_rhs, double *dtSfy_rhs, double *dtSfz_rhs,
|
|
double *rho, double *Sx, double *Sy, double *Sz,
|
|
double *Sxx, double *Sxy, double *Sxz, double *Syy, double *Syz, double *Szz,
|
|
double *Gamxxx, double *Gamxxy, double *Gamxxz, double *Gamxyy, double *Gamxyz, double *Gamxzz,
|
|
double *Gamyxx, double *Gamyxy, double *Gamyxz, double *Gamyyy, double *Gamyyz, double *Gamyzz,
|
|
double *Gamzxx, double *Gamzxy, double *Gamzxz, double *Gamzyy, double *Gamzyz, double *Gamzzz,
|
|
double *Rxx, double *Rxy, double *Rxz, double *Ryy, double *Ryz, double *Rzz,
|
|
double *ham_Res, double *movx_Res, double *movy_Res, double *movz_Res,
|
|
double *Gmx_Res, double *Gmy_Res, double *Gmz_Res,
|
|
int &Symmetry, int &Lev, double &eps, int &co);
|
|
|
|
int bssn_cuda_rk4_substep(void *block_tag,
|
|
int *ex, double *X, double *Y, double *Z,
|
|
double **state_host_in,
|
|
double **state_host_out,
|
|
double **matter_host,
|
|
const double *propspeed,
|
|
const double *soa_flat,
|
|
const double *bbox,
|
|
double &dT,
|
|
double &T,
|
|
int &RK4,
|
|
int &apply_bam_bc,
|
|
int &Symmetry,
|
|
int &Lev,
|
|
double &eps,
|
|
int &co,
|
|
int &use_zero_matter,
|
|
int &keep_resident_state,
|
|
int &apply_enforce_ga,
|
|
double &chitiny);
|
|
|
|
int bssn_cuda_copy_state_region_to_host(void *block_tag,
|
|
int state_index,
|
|
double *host_state,
|
|
int *ex,
|
|
int i0, int j0, int k0,
|
|
int sx, int sy, int sz);
|
|
|
|
int bssn_cuda_copy_state_region_from_host(void *block_tag,
|
|
int state_index,
|
|
double *host_state,
|
|
int *ex,
|
|
int i0, int j0, int k0,
|
|
int sx, int sy, int sz);
|
|
|
|
int bssn_cuda_download_resident_state(void *block_tag,
|
|
int *ex,
|
|
double **state_host_out);
|
|
|
|
int bssn_cuda_download_constraint_outputs(int *ex,
|
|
double **constraint_host_out);
|
|
|
|
int bssn_cuda_pack_state_region_to_host_buffer(void *block_tag,
|
|
int state_index,
|
|
double *host_buffer,
|
|
int *ex,
|
|
int i0, int j0, int k0,
|
|
int sx, int sy, int sz);
|
|
|
|
int bssn_cuda_unpack_state_region_from_host_buffer(void *block_tag,
|
|
int state_index,
|
|
double *host_buffer,
|
|
int *ex,
|
|
int i0, int j0, int k0,
|
|
int sx, int sy, int sz);
|
|
|
|
int bssn_cuda_pack_state_batch_to_host_buffer(void *block_tag,
|
|
int state_count,
|
|
double *host_buffer,
|
|
int *ex,
|
|
int i0, int j0, int k0,
|
|
int sx, int sy, int sz);
|
|
|
|
int bssn_cuda_unpack_state_batch_from_host_buffer(void *block_tag,
|
|
int state_count,
|
|
double *host_buffer,
|
|
int *ex,
|
|
int i0, int j0, int k0,
|
|
int sx, int sy, int sz);
|
|
|
|
int bssn_cuda_download_state_subset(void *block_tag,
|
|
int *ex,
|
|
int subset_count,
|
|
const int *state_indices,
|
|
double **state_host_out);
|
|
|
|
int bssn_cuda_upload_state_subset(void *block_tag,
|
|
int *ex,
|
|
int subset_count,
|
|
const int *state_indices,
|
|
double **state_host_in);
|
|
|
|
int bssn_cuda_has_resident_state(void *block_tag);
|
|
|
|
void bssn_cuda_release_step_ctx(void *block_tag);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif
|