"tests/git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "ef0b2d8287c21a1960b4e212f58e828f0ef264c4"
Unverified Commit 846b512d authored by Nikita Titov's avatar Nikita Titov Committed by GitHub
Browse files

fix compilation warnings in CUDA treelearner (#3889)

* remove unused private field

* mask Train as override

* remove unused private field
parent 846e10c7
...@@ -45,7 +45,7 @@ class CUDATreeLearner: public SerialTreeLearner { ...@@ -45,7 +45,7 @@ class CUDATreeLearner: public SerialTreeLearner {
~CUDATreeLearner(); ~CUDATreeLearner();
void Init(const Dataset* train_data, bool is_constant_hessian) override; 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; 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, bool is_first_tree); Tree* Train(const score_t* gradients, const score_t *hessians, bool is_first_tree) override;
void SetBaggingData(const Dataset* subset, const data_size_t* used_indices, data_size_t num_data) override { void SetBaggingData(const Dataset* subset, const data_size_t* used_indices, data_size_t num_data) override {
SerialTreeLearner::SetBaggingData(subset, used_indices, num_data); SerialTreeLearner::SetBaggingData(subset, used_indices, num_data);
if (subset == nullptr && used_indices != nullptr) { if (subset == nullptr && used_indices != nullptr) {
...@@ -198,12 +198,8 @@ class CUDATreeLearner: public SerialTreeLearner { ...@@ -198,12 +198,8 @@ class CUDATreeLearner: public SerialTreeLearner {
std::vector<uint8_t*> device_features_; std::vector<uint8_t*> device_features_;
/*! \brief GPU memory object holding the ordered gradient */ /*! \brief GPU memory object holding the ordered gradient */
std::vector<score_t*> device_gradients_; std::vector<score_t*> device_gradients_;
/*! \brief Pointer to pinned memory of ordered gradient */
void * ptr_pinned_gradients_ = nullptr;
/*! \brief GPU memory object holding the ordered hessian */ /*! \brief GPU memory object holding the ordered hessian */
std::vector<score_t*> device_hessians_; std::vector<score_t*> device_hessians_;
/*! \brief Pointer to pinned memory of ordered hessian */
void * ptr_pinned_hessians_ = nullptr;
/*! \brief A vector of feature mask. 1 = feature used, 0 = feature not used */ /*! \brief A vector of feature mask. 1 = feature used, 0 = feature not used */
std::vector<char> feature_masks_; std::vector<char> feature_masks_;
/*! \brief GPU memory object holding the feature masks */ /*! \brief GPU memory object holding the feature masks */
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment