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

fix ../tree/gpu_hist/evaluate_splits.hip bugs, size 64

Browse Source
This commit is contained in:
amdsc21 2023-03-14 06:17:21 +01:00
parent a2bab03205
commit 364df7db0f
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/tree/gpu_hist/evaluate_splits.cu
View File

@ -11,6 +11,7 @@
#include "../../common/device_helpers.cuh"
#elif defined(XGBOOST_USE_HIP)
#include "../../common/device_helpers.hip.h"
#include <hip/hip_cooperative_groups.h>
#endif
#include "../../data/ellpack_page.cuh"
@ -96,7 +97,11 @@ class EvaluateSplitAgent {
param(shared_inputs.param), evaluator(evaluator),
missing(parent_sum - ReduceFeature()) {
static_assert(
#if defined(XGBOOST_USE_HIP)
kBlockSize == WAVEFRONT_SIZE,
#elif defined(XGBOOST_USE_CUDA)
kBlockSize == 32,
#endif
"This kernel relies on the assumption block_size == warp_size");
// There should be no missing value gradients for a dense matrix
KERNEL_CHECK(!shared_inputs.is_dense || missing.GetQuantisedHess() == 0);
@ -388,7 +393,11 @@ void GPUHistEvaluator::LaunchEvaluateSplits(
combined_num_features, DeviceSplitCandidate());
// One block for each feature
#if defined(XGBOOST_USE_HIP)
uint32_t constexpr kBlockThreads = WAVEFRONT_SIZE;
#elif defined(XGBOOST_USE_CUDA)
uint32_t constexpr kBlockThreads = 32;
#endif
dh::LaunchKernel {static_cast<uint32_t>(combined_num_features), kBlockThreads,
0}(
EvaluateSplitsKernel<kBlockThreads>, max_active_features, d_inputs,