Use ellpack for prediction only when sparsepage doesn't exist. (#5504)

2020-04-10 12:15:46 +08:00
parent ad826e913f
commit 6671b42dd4
35 changed files with 166 additions and 116 deletions
--- a/src/data/data.cc
+++ b/src/data/data.cc
@@ -293,7 +293,7 @@ void MetaInfo::SetInfo(const char* key, const void* dptr, DataType dtype, size_t

 #if !defined(XGBOOST_USE_CUDA)
 void MetaInfo::SetInfo(const char * c_key, std::string const& interface_str) {
-  LOG(FATAL) << "XGBoost version is not compiled with GPU support";
+  common::AssertGPUSupport();
 }
 #endif  // !defined(XGBOOST_USE_CUDA)