Skip to content

Commit

Permalink
fix compilation warnings in CUDA treelearner (#3889)
Browse files Browse the repository at this point in the history
* remove unused private field

* mask Train as override

* remove unused private field
  • Loading branch information
StrikerRUS authored Feb 9, 2021
1 parent 846e10c commit 846b512
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions src/treelearner/cuda_tree_learner.h
Original file line number Diff line number Diff line change
Expand Up @@ -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, 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 {
SerialTreeLearner::SetBaggingData(subset, used_indices, num_data);
if (subset == nullptr && used_indices != nullptr) {
Expand Down Expand Up @@ -198,12 +198,8 @@ class CUDATreeLearner: public SerialTreeLearner {
std::vector<uint8_t*> device_features_;
/*! \brief GPU memory object holding the ordered gradient */
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 */
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 */
std::vector<char> feature_masks_;
/*! \brief GPU memory object holding the feature masks */
Expand Down

0 comments on commit 846b512

Please sign in to comment.