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

Use quantised gradients in gpu_hist histograms (#8246)

Browse Source
This commit is contained in:
Rory Mitchell
2022-09-26 17:35:35 +02:00
committed by GitHub
parent 4056974e37
commit 8f77677193
14 changed files with 394 additions and 336 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/tree/gpu_hist/evaluate_splits.cuh
View File

@@ -10,6 +10,7 @@
#include "../split_evaluator.h"
#include "../updater_gpu_common.cuh"
#include "expand_entry.cuh"
#include "histogram.cuh"
namespace xgboost {
namespace common {
@@ -24,12 +25,13 @@ struct EvaluateSplitInputs {
int depth;
GradientPairPrecise parent_sum;
common::Span<const bst_feature_t> feature_set;
common::Span<const GradientPairPrecise> gradient_histogram;
common::Span<const GradientPairInt64> gradient_histogram;
};
// Inputs necessary for all nodes
struct EvaluateSplitSharedInputs {
GPUTrainingParam param;
GradientQuantizer rounding;
common::Span<FeatureType const> feature_types;
common::Span<const uint32_t> feature_segments;
common::Span<const float> feature_values;