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

Fix error message. (#6176)

Browse Source
This commit is contained in:
Jiaming Yuan 2020-09-29 11:18:25 +08:00 committed by GitHub
parent dda9e1e487
commit 52c0b3f100
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/predictor/gpu_predictor.cu
View File

@ -559,8 +559,7 @@ class GPUPredictor : public xgboost::Predictor {
bool approximate, int condition, bool approximate, int condition,
unsigned condition_feature) override { unsigned condition_feature) override {
if (approximate) { if (approximate) {
LOG(FATAL) << "[Internal error]: " << __func__ LOG(FATAL) << "Approximated contribution is not implemented in GPU Predictor.";
<< " approximate is not implemented in GPU Predictor.";
} }
dh::safe_cuda(cudaSetDevice(generic_param_->gpu_id)); dh::safe_cuda(cudaSetDevice(generic_param_->gpu_id));