fix isnan

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

View File

@@ -15,13 +15,10 @@
#include "../filesystem.h" // dmlc::TemporaryDirectory
#include "../helpers.h"
#ifdef __CUDACC__
#if defined(__CUDACC__) || defined(__HIP_PLATFORM_AMD__)
#include <xgboost/json.h>
#include "../../../src/data/device_adapter.cuh"
#elif defined(__HIP_PLATFORM_AMD__)
#include <xgboost/json.h>
#include "../../../src/data/device_adapter.hip.h"
#endif // __CUDACC__
#endif // __CUDACC__, __HIP_PLATFORM_AMD__
// Some helper functions used to test both GPU and CPU algorithms
//