File tree Expand file tree Collapse file tree 2 files changed +9
-13
lines changed Expand file tree Collapse file tree 2 files changed +9
-13
lines changed Original file line number Diff line number Diff line change 1
- /* !
2
- * Copyright 2017-2019 XGBoost contributors
1
+ /* *
2
+ * Copyright 2017-2024, XGBoost contributors
3
3
*/
4
4
#pragma once
5
- #include < thrust/random.h>
6
- #include < cstdio>
7
- #include < cub/cub.cuh>
8
- #include < stdexcept>
9
- #include < string>
10
- #include < vector>
11
- #include " ../common/categorical.h"
12
- #include " ../common/device_helpers.cuh"
13
- #include " ../common/random.h"
5
+ #include < limits> // for numeric_limits
6
+ #include < ostream> // for ostream
7
+
14
8
#include " gpu_hist/histogram.cuh"
15
9
#include " param.h"
10
+ #include " xgboost/base.h"
16
11
17
12
namespace xgboost ::tree {
18
13
struct GPUTrainingParam {
@@ -54,8 +49,8 @@ enum DefaultDirection {
54
49
};
55
50
56
51
struct DeviceSplitCandidate {
57
- float loss_chg {-FLT_MAX };
58
- DefaultDirection dir {kLeftDir };
52
+ float loss_chg{-std::numeric_limits< float >:: max () };
53
+ DefaultDirection dir{kLeftDir };
59
54
int findex {-1 };
60
55
float fvalue {0 };
61
56
// categorical split, either it's the split category for OHE or the threshold for partition-based
Original file line number Diff line number Diff line change 19
19
#include " ../common/cuda_context.cuh" // CUDAContext
20
20
#include " ../common/device_helpers.cuh"
21
21
#include " ../common/hist_util.h"
22
+ #include " ../common/random.h" // for ColumnSampler, GlobalRandom
22
23
#include " ../common/timer.h"
23
24
#include " ../data/ellpack_page.cuh"
24
25
#include " ../data/ellpack_page.h"
You can’t perform that action at this time.
0 commit comments