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
|
#ifndef __cuda_device_runtime_h__
#define __cuda_device_runtime_h__
// Jin: cuda_device_runtime.h
// Defines CUDA device runtime APIs for CDP support
class device_launch_config_t {
public:
device_launch_config_t() {}
device_launch_config_t(dim3 _grid_dim, dim3 _block_dim,
unsigned int _shared_mem, function_info* _entry)
: grid_dim(_grid_dim),
block_dim(_block_dim),
shared_mem(_shared_mem),
entry(_entry) {}
dim3 grid_dim;
dim3 block_dim;
unsigned int shared_mem;
function_info* entry;
};
class device_launch_operation_t {
public:
device_launch_operation_t() {}
device_launch_operation_t(kernel_info_t* _grid, CUstream_st* _stream)
: grid(_grid), stream(_stream) {}
kernel_info_t* grid; // a new child grid
CUstream_st* stream;
};
class gpgpu_context;
class cuda_device_runtime {
public:
cuda_device_runtime(gpgpu_context* ctx) {
g_total_param_size = 0;
g_max_total_param_size = 0;
gpgpu_ctx = ctx;
}
unsigned long long g_total_param_size;
std::map<void*, device_launch_config_t> g_cuda_device_launch_param_map;
std::list<device_launch_operation_t> g_cuda_device_launch_op;
unsigned g_kernel_launch_latency;
unsigned g_TB_launch_latency;
unsigned long long g_max_total_param_size;
bool g_cdp_enabled;
// backward pointer
class gpgpu_context* gpgpu_ctx;
#if (CUDART_VERSION >= 5000)
#pragma once
void gpgpusim_cuda_launchDeviceV2(const ptx_instruction* pI,
ptx_thread_info* thread,
const function_info* target_func);
void gpgpusim_cuda_streamCreateWithFlags(const ptx_instruction* pI,
ptx_thread_info* thread,
const function_info* target_func);
void gpgpusim_cuda_getParameterBufferV2(const ptx_instruction* pI,
ptx_thread_info* thread,
const function_info* target_func);
void launch_all_device_kernels();
void launch_one_device_kernel();
#endif
};
#endif /* __cuda_device_runtime_h__ */
|
