Unverified Commit 88110631 authored by Aleksandar Bojarov's avatar Aleksandar Bojarov Committed by GitHub
Browse files

Fix DEBUG-mode GPU builds (#5778)

Fix for DEBUG mode

This commit fixes issue #5777
parent 814a2f1c
...@@ -94,8 +94,8 @@ void CUDASingleGPUTreeLearner::BeforeTrain() { ...@@ -94,8 +94,8 @@ void CUDASingleGPUTreeLearner::BeforeTrain() {
} }
#ifdef DEBUG #ifdef DEBUG
CopyFromCUDADeviceToHost<score_t>(host_gradients.data(), gradients_, static_cast<size_t>(num_data_), __FILE__, __LINE__); CopyFromCUDADeviceToHost<score_t>(host_gradients_.data(), gradients_, static_cast<size_t>(num_data_), __FILE__, __LINE__);
CopyFromCUDADeviceToHost<score_t>(host_hessians.data(), hessians_, static_cast<size_t>(num_data_), __FILE__, __LINE__); CopyFromCUDADeviceToHost<score_t>(host_hessians_.data(), hessians_, static_cast<size_t>(num_data_), __FILE__, __LINE__);
#endif // DEBUG #endif // DEBUG
const data_size_t* leaf_splits_init_indices = const data_size_t* leaf_splits_init_indices =
......
...@@ -68,7 +68,7 @@ class CUDASingleGPUTreeLearner: public SerialTreeLearner { ...@@ -68,7 +68,7 @@ class CUDASingleGPUTreeLearner: public SerialTreeLearner {
void SelectFeatureByNode(const Tree* tree); void SelectFeatureByNode(const Tree* tree);
#ifdef DEUBG #ifdef DEBUG
void CheckSplitValid( void CheckSplitValid(
const int left_leaf, const int right_leaf, const int left_leaf, const int right_leaf,
const double sum_left_gradients, const double sum_right_gradients); const double sum_left_gradients, const double sum_right_gradients);
......
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