diff --git a/src/data/iterative_dmatrix.cc b/src/data/iterative_dmatrix.cc index 8eb1c2034..671b5c87c 100644 --- a/src/data/iterative_dmatrix.cc +++ b/src/data/iterative_dmatrix.cc @@ -356,7 +356,7 @@ BatchSet IterativeDMatrix::GetExtBatches(Context const* ctx, return BatchSet(begin_iter); } -#if !defined(XGBOOST_USE_CUDA) +#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP) inline void IterativeDMatrix::InitFromCUDA(Context const*, BatchParam const&, DataIterHandle, float, std::shared_ptr) { // silent the warning about unused variables. @@ -376,5 +376,5 @@ inline BatchSet IterativeDMatrix::GetEllpackBatches(Context const* inline void GetCutsFromEllpack(EllpackPage const&, common::HistogramCuts*) { common::AssertGPUSupport(); } -#endif // !defined(XGBOOST_USE_CUDA) +#endif // !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP) } // namespace xgboost::data