Skip to content

GitLab

Commit 20996c92 authored by Jeff Daily's avatar Jeff Daily
Browse files

partial revert of 61ec4f1a 

Instead of replacing all #ifdef USE_CUDA, just add USE_CUDA define to ROCm build.
parent 1b3deb5f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/treelearner/cuda/cuda_single_gpu_tree_learner.hpp
View file @ 20996c92
...@@ -9,7 +9,7 @@ ...@@ -9,7 +9,7 @@
#include <memory> #include <memory>
#include <vector> #include <vector>
#if defined(USE_CUDA) || defined(USE_ROCM) #ifdef USE_CUDA
#include "cuda_leaf_splits.hpp" #include "cuda_leaf_splits.hpp"
#include "cuda_histogram_constructor.hpp" #include "cuda_histogram_constructor.hpp"
...@@ -161,5 +161,5 @@ class CUDASingleGPUTreeLearner: public SerialTreeLearner { ...@@ -161,5 +161,5 @@ class CUDASingleGPUTreeLearner: public SerialTreeLearner {
} // namespace LightGBM } // namespace LightGBM
#endif // USE_CUDA || USE_ROCM #endif // USE_CUDA
#endif // LIGHTGBM_TREELEARNER_CUDA_CUDA_SINGLE_GPU_TREE_LEARNER_HPP_ #endif // LIGHTGBM_TREELEARNER_CUDA_CUDA_SINGLE_GPU_TREE_LEARNER_HPP_
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