portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-05 11:45:56 +01:00
Code Releases Activity

pull develop

Browse Source
This commit is contained in:
Christoph Lehner 2020-05-19 08:56:18 -04:00
commit 8358ee38c4
2 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
Grid/allocator/GridMemoryManager.cc
View File

@ -12,6 +12,7 @@ std::vector<_grid_mem_range> _grid_mem_avail;
std::map<void*,_grid_mem_range> _grid_mem_alloc;
void gridMemoryInit() {
#ifdef GRID_NVCC
size_t free,total;
cudaMemGetInfo(&free,&total);
@ -35,10 +36,11 @@ void gridMemoryInit() {
std::cout << GridLogMessage << "gridMemoryInit: " << sz << " bytes" << std::endl;
_grid_mem_avail.push_back( { 0, _grid_mem_pages } );
#endif
}
void gridMallocManaged(void** pp, size_t sz) {
#ifdef GRID_NVCC
if (_grid_mem_avail.empty())
gridMemoryInit();
@ -57,10 +59,13 @@ void gridMallocManaged(void** pp, size_t sz) {
*pp = (char*)_grid_mem_base + _GRID_MEM_PAGE_SIZE*_grid_mem_avail[m].page_start;
_grid_mem_alloc[*pp] = { _grid_mem_avail[m].page_start, _grid_mem_avail[m].page_start + pages };
_grid_mem_avail[m].page_start += pages;
#else
*pp = malloc(sz);
#endif
}
void gridFree(void* p) {
#ifdef GRID_NVCC
if (_grid_mem_avail.empty())
gridMemoryInit();
@ -85,10 +90,13 @@ void gridFree(void* p) {
_grid_mem_avail.push_back( alloc );
}
_grid_mem_alloc.erase(p);
#else
free(p);
#endif
}
void gridAcceleratorPrefetch(void* p, size_t sz) {
#ifdef GRID_NVCC
auto & alloc = _grid_mem_alloc[p];
if (alloc.page_start == alloc.page_end) // pinned to host
return;
@ -96,10 +104,11 @@ void gridAcceleratorPrefetch(void* p, size_t sz) {
int target;
cudaGetDevice(&target);
cudaMemPrefetchAsync(p,sz,target);
#endif
}
void gridMemGetInfo(size_t* pfree, size_t* ptotal) {
#ifdef GRID_NVCC
if (_grid_mem_avail.empty())
gridMemoryInit();
@ -107,10 +116,14 @@ void gridMemGetInfo(size_t* pfree, size_t* ptotal) {
*pfree = 0;
for (auto & a : _grid_mem_avail)
*pfree += (a.page_end - a.page_start) * _GRID_MEM_PAGE_SIZE;
#else
*pfree = 0;
*ptotal = 0;
#endif
}
void gridMoveToHost(void** pp) {
#ifdef GRID_NVCC
if (_grid_mem_avail.empty())
gridMemoryInit();
@ -126,6 +139,7 @@ void gridMoveToHost(void** pp) {
gridFree(*pp);
*pp = pn;
_grid_mem_alloc[pn] = { 0,0 };
#endif
}
NAMESPACE_END(Grid);

5
Grid/serialisation/MacroMagic.h
View File

@ -114,7 +114,8 @@ THE SOFTWARE.
#define GRID_MACRO_WRITE_MEMBER(A,B) ::Grid::write(WR,#B,obj. B);
#define GRID_SERIALIZABLE_CLASS_MEMBERS(cname,...)\
std::string SerialisableClassName(void) const {return std::string(#cname);} \
static inline std::string SerialisableClassName(void) {return std::string(#cname);} \
static constexpr bool isEnum = false; \
GRID_MACRO_EVAL(GRID_MACRO_MAP(GRID_MACRO_MEMBER,__VA_ARGS__))\
template <typename T>\
static inline void write(Writer<T> &WR,const std::string &s, const cname &obj){ \
@ -162,6 +163,8 @@ public:\
public:\
accelerator name(void) : value_(undefname) {}; \
accelerator name(int value): value_(value) {}; \
static inline std::string SerialisableClassName(void) {return std::string(#name);}\
static constexpr bool isEnum = true; \
template <typename T>\
static inline void write(::Grid::Writer<T> &WR,const std::string &s, const name &obj) \
{\