test_map.hpp

#include <unordered_map>

using namespace std;

#define MB_1	(1024 * 1024)

#define	KEY_INVALID		0
#define DIE(assertion, call_description) \
	do {	\
		if (assertion) {	\
		fprintf(stderr, "(%s, %d): ",	\
		__FILE__, __LINE__);	\
		perror(call_description);	\
		exit(1);	\
	}	\
} while (0)

class GpuAllocator
{
    long long int allocMax;
    long long int allocCurrent;

	std::unordered_map<void*, int> allocMap;

public:
    GpuAllocator(long long int allocMax);

    cudaError_t _cudaMalloc( void** devPtr, size_t size );
    cudaError_t _cudaMallocManaged( void** devPtr, size_t size );
    cudaError_t _cudaFree( void* devPtr );

    float _loadFactor( void );
    long long int _used( void );
};

extern GpuAllocator *glbGpuAllocator;