GPU-comcot/GPUConfig.h at master · ypbing80/GPU-comcot · 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
#ifndef SM_CONFG
    #define SM_CONFG
    #define LOAD_PER_SM 2
    #define NUMSTREAM 4
    extern cudaStream_t EXECstream[NUMSTREAM];
#endif

#ifndef MOMT_KERNEL_CONFG
    #define MOMT_KERNEL_CONFG
    #define BLOCKX 16 // ==> along column axis
    #define EXECX  15 // BLOCKX-1
    #define BLOCKY 16 // ==> along row axis
    #define EXECY  15// BLOCKY-1

     extern dim3 DimBlockMomt;
     extern dim3 DimGridMomt;

     #define BLOCKX_MOMT 512
     extern dim3 DimBlockMomt_MN;
     extern dim3 DimGridMomt_MN;
#endif

#ifndef MASS_KERNEL_CONFG
    #define MASS_KERNEL_CONFG
    #define BLOCKX_MASS 64
     extern dim3 DimBlockMass;
     extern dim3 DimGridMass;
#endif

#ifndef OPENBD_KERNEL_CONFG
    #define OPENBD_KERNEL_CONFG
    #define BLOCKX_OPENBD 64
    extern dim3 DimBlockOpenBD;
    extern dim3 DimGridOpenBD_LR;
    extern dim3 DimGridOpenBD_TB;
#endif

#ifndef MAXAMP_KERNEL_CONFG
    #define MAXAMP_KERNEL_CONFG
    #define MAXAMP_BLOCK 512
    extern size_t GridMaxAmp;
    #define BLOCKX_MAXAMP 8
    #define BLOCKY_MAXAMP 8
    extern dim3 DimBlockMaxAmp;
    extern dim3 DimGridMaxAmp;
#endif