From 3376af879169558cc4cad8f9c5ef092e0aced927 Mon Sep 17 00:00:00 2001 From: fis Date: Tue, 16 Nov 2021 19:39:34 +0800 Subject: [PATCH] Polish. --- src/predictor/gpu_predictor.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/predictor/gpu_predictor.cu b/src/predictor/gpu_predictor.cu index 025a1c495c52..71724f95236c 100644 --- a/src/predictor/gpu_predictor.cu +++ b/src/predictor/gpu_predictor.cu @@ -854,7 +854,7 @@ class GPUPredictor : public xgboost::Predictor { dh::tend(phis)); } // Add the base margin term to last column - p_fmat->Info().base_margin_.Data()->SetDevice(generic_param_->gpu_id); + p_fmat->Info().base_margin_.SetDevice(generic_param_->gpu_id); const auto margin = p_fmat->Info().base_margin_.Data()->ConstDeviceSpan(); float base_score = model.learner_model_param->base_score; dh::LaunchN( @@ -913,7 +913,7 @@ class GPUPredictor : public xgboost::Predictor { dh::tend(phis)); } // Add the base margin term to last column - p_fmat->Info().base_margin_.Data()->SetDevice(generic_param_->gpu_id); + p_fmat->Info().base_margin_.SetDevice(generic_param_->gpu_id); const auto margin = p_fmat->Info().base_margin_.Data()->ConstDeviceSpan(); float base_score = model.learner_model_param->base_score; size_t n_features = model.learner_model_param->num_feature;