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
113
114
115
116
117
118
119
120
121
122
123
124
125
|
// start of prelude.cu
#define SCALAR_FUN_ATTR __device__ static inline
#define FUTHARK_FUN_ATTR __device__ static
#define FUTHARK_F64_ENABLED
#if defined(__CUDACC_RTC__) || defined(__HIPCC_RTC__)
typedef char int8_t;
typedef short int16_t;
typedef int int32_t;
typedef long long int64_t;
typedef unsigned char uint8_t;
typedef unsigned short uint16_t;
typedef unsigned int uint32_t;
typedef unsigned long long uint64_t;
#else
// This is for the benefit of offline compilation with clang.
typedef signed char int8_t;
typedef short int16_t;
typedef int int32_t;
typedef long int64_t;
typedef unsigned char uint8_t;
typedef unsigned short uint16_t;
typedef unsigned int uint32_t;
typedef unsigned long uint64_t;
#endif
#if defined(__CUDACC_RTC__)
typedef uint64_t uintptr_t;
#endif
#if defined(__HIPCC_RTC__)
typedef uint64_t uintptr_t;
#endif
#define __global
#define __local
#define __private
#define __constant
#define __write_only
#define __read_only
static inline __device__ int get_tblock_id(int d) {
switch (d) {
case 0: return blockIdx.x;
case 1: return blockIdx.y;
case 2: return blockIdx.z;
default: return 0;
}
}
static inline __device__ int get_num_tblocks(int d) {
switch(d) {
case 0: return gridDim.x;
case 1: return gridDim.y;
case 2: return gridDim.z;
default: return 0;
}
}
static inline __device__ int get_global_id(int d) {
switch (d) {
case 0: return threadIdx.x + blockIdx.x * blockDim.x;
case 1: return threadIdx.y + blockIdx.y * blockDim.y;
case 2: return threadIdx.z + blockIdx.z * blockDim.z;
default: return 0;
}
}
static inline __device__ int get_local_id(int d) {
switch (d) {
case 0: return threadIdx.x;
case 1: return threadIdx.y;
case 2: return threadIdx.z;
default: return 0;
}
}
static inline __device__ int get_local_size(int d) {
switch (d) {
case 0: return blockDim.x;
case 1: return blockDim.y;
case 2: return blockDim.z;
default: return 0;
}
}
static inline __device__ int get_global_size(int d) {
switch (d) {
case 0: return gridDim.x * blockDim.x;
case 1: return gridDim.y * blockDim.y;
case 2: return gridDim.z * blockDim.z;
default: return 0;
}
}
#define CLK_LOCAL_MEM_FENCE 1
#define CLK_GLOBAL_MEM_FENCE 2
static inline __device__ void barrier(int x) {
__syncthreads();
}
static inline __device__ void mem_fence_local() {
__threadfence_block();
}
static inline __device__ void mem_fence_global() {
__threadfence();
}
static inline __device__ void barrier_local() {
__syncthreads();
}
#if defined(__CUDACC_RTC__) || defined(__HIPCC_RTC__)
#define NAN (0.0/0.0)
#define INFINITY (1.0/0.0)
#endif
extern volatile __shared__ unsigned char shared_mem[];
#define SHARED_MEM_PARAM
#define FUTHARK_KERNEL extern "C" __global__ __launch_bounds__(MAX_THREADS_PER_BLOCK)
#define FUTHARK_KERNEL_SIZED(a,b,c) extern "C" __global__ __launch_bounds__(a*b*c)
// End of prelude.cu
|
