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

Update GPU code with dmatrix changes (#3117)

Browse Source
This commit is contained in:
Rory Mitchell 2018-02-17 12:11:48 +13:00 committed by GitHub
parent 10eb05a63a
commit dd82b28e20
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/tree/updater_gpu.cu
View File

@ -651,9 +651,9 @@ class GPUMaker : public TreeUpdater {
fId->reserve(nCols * nRows); fId->reserve(nCols * nRows);
// in case you end up with a DMatrix having no column access // in case you end up with a DMatrix having no column access
// then make sure to enable that before copying the data! // then make sure to enable that before copying the data!
if (!dmat->HaveColAccess()) { if (!dmat->HaveColAccess(true)) {
const std::vector<bool> enable(nCols, true); const std::vector<bool> enable(nCols, true);
dmat->InitColAccess(enable, 1, nRows); dmat->InitColAccess(enable, 1, nRows, true);
} }
dmlc::DataIter<ColBatch>* iter = dmat->ColIterator(); dmlc::DataIter<ColBatch>* iter = dmat->ColIterator();
iter->BeforeFirst(); iter->BeforeFirst();