TileSpGEMM/common.h at main · YuyaoNiu/TileSpGEMM · 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
#include <time.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>

#include <sys/time.h>

#include "utils.h"

#ifndef MAT_VAL_TYPE
#define MAT_VAL_TYPE double
#endif

#ifndef MAT_PTR_TYPE
#define MAT_PTR_TYPE int
#endif

//#ifndef SubNum
//#define SubNum  3
//#endif


#define WARP_SIZE 32
#define WARP_PER_BLOCK 4

#define HALFWARP_SIZE WARP_SIZE/2
#define HALFWARP_PER_BLOCK 2*WARP_PER_BLOCK

#ifndef BLOCK_SIZE
#define BLOCK_SIZE  16
#endif

#ifndef NTHREADS_MAX
#define NTHREADS_MAX 1
#endif


#define METHOD_SERIAL               1
#define METHOD_LOCK                 2
#define METHOD_LOCK_AND_DEPENDENCY  3

#define HASH_SCALE 107