NavierStokes3DInitialize_GPU.c

Go to the documentation of this file.

#include <arrayfunctions_gpu.h>
#include <physicalmodels/navierstokes3d.h>
#include <hypar.h>

int gpuNavierStokes3DInitialize(
    void *s,  
    void *m   
)
{
    HyPar          *solver   = (HyPar*)          s;
    NavierStokes3D *physics  = (NavierStokes3D*) solver->physics;

    int *dim = solver->dim_local;
    int ghosts = solver->ghosts;
    int d, size = 1; for (d = 0; d <_MODEL_NDIMS_; d++) size *= (dim[d] + 2*ghosts);

    gpuMalloc((void**)&physics->gpu_Q, size*_MODEL_NVARS_*sizeof(double));
    gpuMalloc((void**)&physics->gpu_QDerivX, size*_MODEL_NVARS_*sizeof(double));
    gpuMalloc((void**)&physics->gpu_QDerivY, size*_MODEL_NVARS_*sizeof(double));
    gpuMalloc((void**)&physics->gpu_QDerivZ, size*_MODEL_NVARS_*sizeof(double));
    gpuMalloc((void**)&physics->gpu_FViscous, size*_MODEL_NVARS_*sizeof(double));
    gpuMalloc((void**)&physics->gpu_FDeriv, size*_MODEL_NVARS_*sizeof(double));
    gpuMalloc((void**)&physics->gpu_grav_field_f, size*sizeof(double));
    gpuMalloc((void**)&physics->gpu_grav_field_g, size*sizeof(double));
    gpuMalloc((void**)&physics->gpu_fast_jac, _MODEL_NDIMS_*size*_MODEL_NVARS_*_MODEL_NVARS_*sizeof(double));
    gpuMalloc((void**)&physics->gpu_solution, size*_MODEL_NVARS_*sizeof(double));
    gpuMemset(physics->gpu_Q, 0, size*_MODEL_NVARS_*sizeof(double));
    gpuMemset(physics->gpu_QDerivX, 0, size*_MODEL_NVARS_*sizeof(double));
    gpuMemset(physics->gpu_QDerivY, 0, size*_MODEL_NVARS_*sizeof(double));
    gpuMemset(physics->gpu_QDerivZ, 0, size*_MODEL_NVARS_*sizeof(double));
    gpuMemset(physics->gpu_FViscous, 0, size*_MODEL_NVARS_*sizeof(double));
    gpuMemset(physics->gpu_FDeriv, 0, size*_MODEL_NVARS_*sizeof(double));
    gpuMemcpy(physics->gpu_grav_field_f, physics->grav_field_f, size*sizeof(double), gpuMemcpyHostToDevice);
    gpuMemcpy(physics->gpu_grav_field_g, physics->grav_field_g, size*sizeof(double), gpuMemcpyHostToDevice);
    gpuMemset(physics->gpu_fast_jac, 0, _MODEL_NDIMS_*size*_MODEL_NVARS_*_MODEL_NVARS_*sizeof(double));
    gpuMemset(physics->gpu_solution, 0, size*_MODEL_NVARS_*sizeof(double));

    return (0);
}