Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix CAPI BuildInfo

Browse Source
This commit is contained in:
amdsc21
2023-03-28 00:14:18 +02:00
parent c50cc424bc
commit 06d9b998ce
6 changed files with 21 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/cpp/c_api/test_c_api.cc
View File

@@ -362,10 +362,13 @@ TEST(CAPI, BuildInfo) {
XGBuildInfo(&out);
auto loaded = Json::Load(StringView{out});
ASSERT_TRUE(get<Object const>(loaded).find("USE_OPENMP") != get<Object const>(loaded).cend());
#if defined(XGBOOST_USE_CUDA)
ASSERT_TRUE(get<Object const>(loaded).find("USE_CUDA") != get<Object const>(loaded).cend());
ASSERT_TRUE(get<Object const>(loaded).find("USE_NCCL") != get<Object const>(loaded).cend());
#elif defined(XGBOOST_USE_HIP)
ASSERT_TRUE(get<Object const>(loaded).find("USE_HIP") != get<Object const>(loaded).cend());
ASSERT_TRUE(get<Object const>(loaded).find("USE_RCCL") != get<Object const>(loaded).cend());
#endif
}
TEST(CAPI, NullPtr) {

8
tests/cpp/collective/test_nccl_device_communicator.cu
View File

@@ -1,13 +1,17 @@
/**
* Copyright 2022-2023, XGBoost contributors
*/
#ifdef XGBOOST_USE_NCCL
#if defined(XGBOOST_USE_NCCL) || defined(XGBOOST_USE_RCCL)
#include <gtest/gtest.h>
#include <string> // for string
#if defined(XGBOOST_USE_NCCL)
#include "../../../src/collective/nccl_device_communicator.cuh"
#elif defined(XGBOOST_USE_RCCL)
#include "../../../src/collective/nccl_device_communicator.hip.h"
#endif
namespace xgboost {
namespace collective {
@@ -33,4 +37,4 @@ TEST(NcclDeviceCommunicatorSimpleTest, SystemError) {
} // namespace collective
} // namespace xgboost
#endif // XGBOOST_USE_NCCL
#endif // XGBOOST_USE_NCCL || XGBOOST_USE_RCCL