finish common.cu

2023-03-10 05:20:29 +01:00
parent 91a5ef762e
commit 54b076b40f
1 changed files with 2 additions and 2 deletions
--- a/src/common/common.cc
+++ b/src/common/common.cc
@@ -23,11 +23,11 @@ GlobalRandomEngine& GlobalRandom() {
  return RandomThreadLocalStore::Get()->engine;
 }

-#if !defined(XGBOOST_USE_CUDA)
+#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
 int AllVisibleGPUs() {
  return 0;
 }
-#endif  // !defined(XGBOOST_USE_CUDA)
+#endif  // !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)

 }  // namespace common
 }  // namespace xgboost