Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add gpu error check

Browse Source
This commit is contained in:
Hendrik Groove 2024-10-20 17:34:51 +02:00
parent fd95be5f20
commit ab41cd26a6
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/common/gpu_error_check.h Normal file
View File

@ -0,0 +1,15 @@
#pragma once
#include <iostream>
#include <hip/hip_runtime.h>
#define GPU_CHECK(ans) { gpuAssert((ans), __FILE__, __LINE__); }
inline void gpuAssert(hipError_t code, const char *file, int line, bool abort=true)
{
if (code != hipSuccess)
{
std::cerr << "GPU Error: " << hipGetErrorString(code) << " " << file << " " << line << std::endl;
if (abort) exit(code);
}
}
#define GPU_CHECK_LAST() { gpuAssert(hipGetLastError(), __FILE__, __LINE__); }