summaryrefslogtreecommitdiff
path: root/libcuda/gpgpu_context.h
blob: 6878d5cf9fae86976e8311d4c678debd70d4094e (plain)
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
#ifndef __gpgpu_context_h__
#define __gpgpu_context_h__
#include <list>
#include <map>
#include <set>
#include <string>
#include "cuda_api_object.h"

class cuobjdumpSection;
class cuobjdumpELFSection;
class cuobjdumpPTXSection;
class symbol_table;
class gpgpu_ptx_sim_arg;
class kernel_info_t;

typedef std::list<gpgpu_ptx_sim_arg> gpgpu_ptx_sim_arg_list_t;

class gpgpu_context {
    public:
	gpgpu_context() {
	    api = new cuda_runtime_api();
	}
	// global list
	std::list<cuobjdumpSection*> cuobjdumpSectionList;
	std::map<int, bool>fatbin_registered;
	std::map<int, std::string> fatbinmap;
	std::map<std::string, symbol_table*> name_symtab;
	std::map<unsigned long long, size_t> g_mallocPtr_Size;
	//maps sm version number to set of filenames
	std::map<unsigned, std::set<std::string> > version_filename;
	std::map<void *,void **> pinned_memory; //support for pinned memories added
	std::map<void *, size_t> pinned_memory_size;
	// objects pointers for each file
	cuda_runtime_api* api;
	// member function list
	void cuobjdumpInit();
	void cuobjdumpParseBinary(unsigned int handle);
	void extract_code_using_cuobjdump();
	std::list<cuobjdumpSection*> pruneSectionList(CUctx_st *context);
	std::list<cuobjdumpSection*> mergeMatchingSections(std::string identifier);
	std::list<cuobjdumpSection*> mergeSections();
	cuobjdumpELFSection* findELFSection(const std::string identifier);
	cuobjdumpPTXSection* findPTXSection(const std::string identifier);
	void extract_ptx_files_using_cuobjdump(CUctx_st *context);
	void cuobjdumpRegisterFatBinary(unsigned int handle, const char* filename, CUctx_st *context);
	kernel_info_t *gpgpu_cuda_ptx_sim_init_grid( const char *kernel_key,
		gpgpu_ptx_sim_arg_list_t args,
		struct dim3 gridDim,
		struct dim3 blockDim,
		struct CUctx_st* context );

};
#endif /* __gpgpu_context_h__ */