From a27d9510d1b4fdb38e527788cacfc3ff28146886 Mon Sep 17 00:00:00 2001 From: Nikita Titov Date: Mon, 11 Jan 2021 18:28:41 +0300 Subject: [PATCH 1/3] Update cuda_tree_learner.cpp --- src/treelearner/cuda_tree_learner.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/treelearner/cuda_tree_learner.cpp b/src/treelearner/cuda_tree_learner.cpp index 16569eef257f..78735c07c6e5 100644 --- a/src/treelearner/cuda_tree_learner.cpp +++ b/src/treelearner/cuda_tree_learner.cpp @@ -534,8 +534,8 @@ void CUDATreeLearner::InitGPU(int num_gpu) { copyDenseFeature(); } -Tree* CUDATreeLearner::Train(const score_t* gradients, const score_t *hessians) { - Tree *ret = SerialTreeLearner::Train(gradients, hessians); +Tree* CUDATreeLearner::Train(const score_t* gradients, const score_t *hessians, bool is_first_tree) { + Tree *ret = SerialTreeLearner::Train(gradients, hessians, is_first_tree); return ret; } From 88e95d6bba47629637bdbb9015a5a1f1e52f6473 Mon Sep 17 00:00:00 2001 From: Nikita Titov Date: Mon, 11 Jan 2021 18:29:43 +0300 Subject: [PATCH 2/3] Update cuda_tree_learner.h --- src/treelearner/cuda_tree_learner.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/treelearner/cuda_tree_learner.h b/src/treelearner/cuda_tree_learner.h index 442c2f53ea01..41e1591a257c 100644 --- a/src/treelearner/cuda_tree_learner.h +++ b/src/treelearner/cuda_tree_learner.h @@ -45,7 +45,7 @@ class CUDATreeLearner: public SerialTreeLearner { ~CUDATreeLearner(); void Init(const Dataset* train_data, bool is_constant_hessian) override; void ResetTrainingDataInner(const Dataset* train_data, bool is_constant_hessian, bool reset_multi_val_bin) override; - Tree* Train(const score_t* gradients, const score_t *hessians); + Tree* Train(const score_t* gradients, const score_t *hessians, bool is_first_tree); void SetBaggingData(const Dataset* subset, const data_size_t* used_indices, data_size_t num_data) override { SerialTreeLearner::SetBaggingData(subset, used_indices, num_data); if (subset == nullptr && used_indices != nullptr) { From 0cf8181e16ee3b991b690df37c6f1fe40bec39a4 Mon Sep 17 00:00:00 2001 From: Nikita Titov Date: Mon, 11 Jan 2021 19:59:21 +0300 Subject: [PATCH 3/3] Update cuda.yml --- .github/workflows/cuda.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/cuda.yml b/.github/workflows/cuda.yml index 71dbdda19ea9..db3c763b8951 100644 --- a/.github/workflows/cuda.yml +++ b/.github/workflows/cuda.yml @@ -47,6 +47,7 @@ jobs: export ROOT_DOCKER_FOLDER=/LightGBM cat > docker.env <