OpenArray_Dev/common.hpp at master · wangdong1991/OpenArray_Dev · 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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#ifndef __COMMON_H__
#define __COMMON_H__

#include <array>
#include <assert.h>

#include "otype.hpp"

// define random seed
#define SEED 2333333

// define stencil type
#define STENCIL_STAR 0
#define STENCIL_BOX 1

// define boundary type
#define BOUND_OPEN 0
#define BOUNDARY_OPEN 0
#define BOUND_PERIODIC 1
#define BOUNDARY_PERIODIC 1

// define basic data types
#define DATA_UNKNOWN -1
#define DATA_INT 0
#define DATA_FLOAT 1
#define DATA_DOUBLE 2
#define DATA_BOOL 3

#define NO_STENCIL 0

// define basic data size
#define DATA_SIZE(i) ((int[]{4, 4, 8})[i])

// define shape dimension [x, y, z]
typedef std::array<int, 3> Shape;
typedef std::array<int, 3> int3;

///:for o in ['+','-','*']
int3 operator${o}$(const int3& a, const int3& b);
///:endfor

typedef int DataType;
typedef int DATA_TYPE;


///:mute
///:set i = 0
///:include "NodeType.fypp"
///:endmute
//define node types
enum NodeType{
  ///:for i in range(len(L))
  ///:if i == 0
  ${L[i][0]}$ = 0,
  ///:elif i == len(L) - 1
  ${L[i][0]}$
  ///:else
  ${L[i][0]}$,
  ///:endif
  ///:endfor
};

#define NUM_NODE_TYPES ${len(L)}$

#include <armadillo>


typedef arma::Cube<int>  cube_int;
typedef arma::Cube<float> cube_float;
typedef arma::Cube<double> cube_double;

#define SCALAR_SHAPE Shape({{1,1,1}})

#define WITHOUT_FUSION_KERNEL 0
#define WITH_FUSION_KERNEL 1


#define CSET(A, B)                                    \
  tmp_node_key = gen_node_key(__FILE__, __LINE__);    \
  find_node(tmp_node, tmp_node_key);                  \
  if (is_valid(tmp_node)) {                           \
    A = tmp_node;                                     \
  } else {                                            \
    tmp_node = B;                                     \
    cache_node(tmp_node, tmp_node_key);               \
    A = tmp_node;                                     \
  }

#define THROW_LOGIC_EXCEPTION(msg)              \
  BOOST_THROW_EXCEPTION(std::logic_error(msg))

#define ENSURE_VALID_PTR(A) \
  assert(A && "pointer is null");
  //if(A == NULL) THROW_LOGIC_EXCEPTION("pointer is null.");

#define ENSURE_VALID_PTR_MSG(A, MSG)        \
  assert(A && MSG);
  //if(A == NULL) THROW_LOGIC_EXCEPTION(MSG);

#define MPI_ORDER_START oa::MPI::global()->order_start();
#define MPI_ORDER_END   oa::MPI::global()->order_end();

#define MPI_RANK oa::MPI::global()->rank()
#define MPI_SIZE oa::MPI::global()->size()

extern bool g_cache;
extern bool g_debug;
extern bool transticbegin;

// #define DEBUG

#endif