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

Fix gpu devices. (#3693)

Browse Source 
* Fix gpu_set normalized and unnormalized.
* Fix DeviceSpan.
This commit is contained in:
trivialfis
2018-09-19 17:39:42 +12:00
committed by Rory Mitchell
parent 0f99cdfe0e
commit 9119f9e369
13 changed files with 199 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/tree/updater_gpu.cu
View File

@@ -4,7 +4,7 @@
#include <xgboost/tree_updater.h>
#include <utility>
#include <vector>
#include "../common/gpu_set.h"
#include "../common/common.h"
#include "param.h"
#include "updater_gpu_common.cuh"