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

[GPU-Plugin] Make node_idx type 32 bit for hist algo. Set default n_gpus to 1. (#2445)

Browse Source
This commit is contained in:
Rory Mitchell
2017-06-23 18:26:45 +12:00
committed by GitHub
parent 169c983b5f
commit 0e48f87529
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
plugin/updater_gpu/src/gpu_hist_builder.cuh
View File

@@ -121,8 +121,8 @@ class GPUHistBuilder {
std::vector<dh::dvec<float>> fidx_min_map;
std::vector<dh::dvec<int>> feature_segments;
std::vector<dh::dvec<bst_float>> prediction_cache;
std::vector<dh::dvec<NodeIdT>> position;
std::vector<dh::dvec<NodeIdT>> position_tmp;
std::vector<dh::dvec<int>> position;
std::vector<dh::dvec<int>> position_tmp;
std::vector<DeviceGMat> device_matrix;
std::vector<dh::dvec<gpu_gpair>> device_gpair;
std::vector<dh::dvec<int>> gidx_feature_map;