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

finish evaluate_splits.cu

Browse Source
This commit is contained in:
amdsc21
2023-03-09 22:15:10 +01:00
parent 1e09c21456
commit f55243fda0
7 changed files with 73 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/common/cuda_pinned_allocator.h
View File

@@ -74,7 +74,7 @@ class pinned_allocator {
pointer result(nullptr);
#if defined(XGBOOST_USE_HIP)
dh::safe_cuda(hipMallocHost(reinterpret_cast<void**>(&result), cnt * sizeof(value_type)));
dh::safe_cuda(hipHostMalloc(reinterpret_cast<void**>(&result), cnt * sizeof(value_type)));
#else
dh::safe_cuda(cudaMallocHost(reinterpret_cast<void**>(&result), cnt * sizeof(value_type)));
#endif
@@ -84,7 +84,7 @@ class pinned_allocator {
inline void deallocate(pointer p, size_type) {
#if defined(XGBOOST_USE_HIP)
dh::safe_cuda(hipFreeHost(p));
dh::safe_cuda(hipHostFree(p));
#else
dh::safe_cuda(cudaFreeHost(p));
#endif

4
src/common/transform.h
View File

@@ -17,8 +17,10 @@
#include "xgboost/host_device_vector.h"
#include "xgboost/span.h"
#if defined (__CUDACC__) || defined(__HIP_PLATFORM_AMD__)
#if defined (__CUDACC__)
#include "device_helpers.cuh"
#elif defined(__HIP_PLATFORM_AMD__)
#include "device_helpers.hip.h"
#endif // defined (__CUDACC__) || defined(__HIP_PLATFORM_AMD__)
namespace xgboost {