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

finish fit_stump.cu

Browse Source
This commit is contained in:
amdsc21 2023-03-10 00:46:29 +01:00
parent 1530c03f7d
commit 1c58ff61d1
4 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/tree/constraints.cu
View File

@ -15,7 +15,7 @@
#include "constraints.cuh" #include "constraints.cuh"
#include "param.h" #include "param.h"
#if defined(XGBOOST_USE_hip.CUDA) #if defined(XGBOOST_USE_CUDA)
#include "../common/device_helpers.cuh" #include "../common/device_helpers.cuh"
#elif defined(XGBOOST_USE_HIP) #elif defined(XGBOOST_USE_HIP)
#include "../common/device_helpers.hip.h" #include "../common/device_helpers.hip.h"

4
src/tree/fit_stump.cc
View File

@ -56,12 +56,12 @@ namespace cuda_impl {
void FitStump(Context const* ctx, linalg::TensorView<GradientPair const, 2> gpair, void FitStump(Context const* ctx, linalg::TensorView<GradientPair const, 2> gpair,
linalg::VectorView<float> out); linalg::VectorView<float> out);
#if !defined(XGBOOST_USE_CUDA) #if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
inline void FitStump(Context const*, linalg::TensorView<GradientPair const, 2>, inline void FitStump(Context const*, linalg::TensorView<GradientPair const, 2>,
linalg::VectorView<float>) { linalg::VectorView<float>) {
common::AssertGPUSupport(); common::AssertGPUSupport();
} }
#endif // !defined(XGBOOST_USE_CUDA) #endif // !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_C
} // namespace cuda_impl } // namespace cuda_impl
void FitStump(Context const* ctx, HostDeviceVector<GradientPair> const& gpair, void FitStump(Context const* ctx, HostDeviceVector<GradientPair> const& gpair,

12
src/tree/fit_stump.cu
View File

@ -12,7 +12,13 @@
#include <cstddef> // std::size_t #include <cstddef> // std::size_t
#include "../collective/device_communicator.cuh" // DeviceCommunicator #include "../collective/device_communicator.cuh" // DeviceCommunicator
#if defined(XGBOOST_USE_CUDA)
#include "../common/device_helpers.cuh" // dh::MakeTransformIterator #include "../common/device_helpers.cuh" // dh::MakeTransformIterator
#elif defined(XGBOOST_USE_HIP)
#include "../common/device_helpers.hip.h" // dh::MakeTransformIterator
#endif
#include "fit_stump.h" #include "fit_stump.h"
#include "xgboost/base.h" // GradientPairPrecise, GradientPair, XGBOOST_DEVICE #include "xgboost/base.h" // GradientPairPrecise, GradientPair, XGBOOST_DEVICE
#include "xgboost/context.h" // Context #include "xgboost/context.h" // Context
@ -45,7 +51,13 @@ void FitStump(Context const* ctx, linalg::TensorView<GradientPair const, 2> gpai
CHECK(d_sum.CContiguous()); CHECK(d_sum.CContiguous());
dh::XGBCachingDeviceAllocator<char> alloc; dh::XGBCachingDeviceAllocator<char> alloc;
#if defined(XGBOOST_USE_CUDA)
auto policy = thrust::cuda::par(alloc); auto policy = thrust::cuda::par(alloc);
#elif defined(XGBOOST_USE_HIP)
auto policy = thrust::hip::par(alloc);
#endif
thrust::reduce_by_key(policy, key_it, key_it + gpair.Size(), grad_it, thrust::reduce_by_key(policy, key_it, key_it + gpair.Size(), grad_it,
thrust::make_discard_iterator(), dh::tbegin(d_sum.Values())); thrust::make_discard_iterator(), dh::tbegin(d_sum.Values()));

4
src/tree/fit_stump.hip
View File

@ -0,0 +1,4 @@
#if defined(XGBOOST_USE_HIP)
#include "fit_stump.cu"
#endif