HRScode/kernel.h at master · guroseternes/HRScode · GitHub

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
70
71
72
73
74
75
#ifndef KERNEL_H_
#define KERNEL_H_
#include <stdio.h>
#include "gpu_ptr.h"
#include <cuda.h>
#include <cfloat>
#include "kernel_arg_structs.h"
//#include "global.h"

//Declare variables

//Block sizes
// For flux function
const int BLOCKDIM_X = 16;
const int BLOCKDIM_Y = 16;

const int SM_BLOCKDIM_Y = BLOCKDIM_Y + 1;
const int TILEDIM_X = BLOCKDIM_X-2;
const int TILEDIM_Y = BLOCKDIM_Y-2;

const int INNERTILEDIM_X = TILEDIM_X-2;
const int INNERTILEDIM_Y = TILEDIM_Y-2;

const int BLOCKDIM_X_RK = 32;
const int BLOCKDIM_Y_RK = 32;

const int BLOCKDIM_BC = 512;

const int TIMETHREADS = 512;

// Constants for the method
const float THETA = 1.5; //minmod limiter parameter
const float GAMMA = 1.4; //gas constant

// Pointer to store the r-values for the CFL condition
//extern float* L_host;
//extern float* L_device;
extern int nElements;

// Pointer to store the dt-value f
//extern float* dt_device;
extern float* dt_host;

// Kernel arguments
extern collBCKernelArgs* BCArgs[3];
extern FluxKernelArgs* fluxArgs[3];
extern RKKernelArgs* RKArgs[3];
extern DtKernelArgs* dtArgs;

// gridBlocks and threadBlocks
extern dim3 gridBC;
extern dim3 blockBC;

extern dim3 gridBlockFlux;
extern dim3 threadBlockFlux;

extern dim3 gridBlockRK;
extern dim3 threadBlockRK;

// Declare functions
extern void init_allocate();

void callRKKernel(dim3 grid, dim3 block, int step, RKKernelArgs* RKarg);

void callDtKernel(int nThreads, DtKernelArgs* DtArg);

void callCollectiveSetBCWall(dim3 grid, dim3 block, const collBCKernelArgs* h_ctx);

void callFluxKernel(dim3 grid, dim3 block, int step, FluxKernelArgs* FluxArgs);

void callCollectiveSetBCOpen(dim3 grid, dim3 block, const collBCKernelArgs* arg);

void callCollectiveSetBCPeriodic(dim3 grid, dim3 block, const collBCKernelArgs* arg);

#endif