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

fix isnan

Browse Source
This commit is contained in:
amdsc21
2023-03-11 02:39:58 +01:00
parent e5b6219a84
commit b4dbe7a649
6 changed files with 9 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/common/math.h
View File

@@ -155,16 +155,20 @@ bool CheckNAN(double v);
#else
XGBOOST_DEVICE bool inline CheckNAN(float x) {
#if defined(__CUDA_ARCH__) || defined(__HIP_PLATFORM_AMD__)
#if defined(__CUDA_ARCH__)
return isnan(x);
#elif defined(__HIP_PLATFORM_AMD__)
return __builtin_isnan(x);
#else
return std::isnan(x);
#endif // defined(__CUDA_ARCH__) || defined(__HIP_PLATFORM_AMD__)
}
XGBOOST_DEVICE bool inline CheckNAN(double x) {
#if defined(__CUDA_ARCH__) || defined(__HIP_PLATFORM_AMD__)
#if defined(__CUDA_ARCH__)
return isnan(x);
#elif defined(__HIP_PLATFORM_AMD__)
return __builtin_isnan(x);
#else
return std::isnan(x);
#endif // defined(__CUDA_ARCH__) || defined(__HIP_PLATFORM_AMD__)