finish common.cu

2023-03-10 05:19:41 +01:00
parent 8fd2af1c8b
commit 91a5ef762e
3 changed files with 17 additions and 1 deletions
--- a/src/common/common.h
+++ b/src/common/common.h
@@ -156,7 +156,7 @@ int AllVisibleGPUs();
 inline void AssertGPUSupport() {
 #if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
    LOG(FATAL) << "XGBoost version not compiled with GPU support.";
-#endif  // XGBOOST_USE_CUDA
+#endif  // XGBOOST_USE_CUDA && XGBOOST_USE_HIP
 }

 inline void AssertOneAPISupport() {