portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-10-31 20:14:32 +00:00
Code Releases Activity

Diagnostics on memory alloc fail

Browse Source
This commit is contained in:
paboyle
2018-06-27 21:26:20 +01:00
parent dd891f5e3b
commit 7f45222924
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/allocator/AlignedAllocator.h
View File

@@ -170,8 +170,10 @@ public:
// Unified (managed) memory // Unified (managed) memory
//////////////////////////////////// ////////////////////////////////////
if ( ptr == (_Tp *) NULL ) { if ( ptr == (_Tp *) NULL ) {
if( cudaMallocManaged((void **)&ptr,bytes) != cudaSuccess ) { auto err = cudaMallocManaged((void **)&ptr,bytes);
if( err != cudaSuccess ) {
ptr = (_Tp *) NULL; ptr = (_Tp *) NULL;
std::cerr << " cudaMallocManaged failed for " << bytes<<" bytes " <<cudaGetErrorString(err)<< std::endl;
assert(0); assert(0);
} }
} }