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

enable coll and comm

Browse Source
This commit is contained in:
Hui Liu
2023-10-30 15:15:05 -07:00
parent b6b5218245
commit 02f5464fa6
18 changed files with 76 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/cpp/collective/test_allgather.cu
View File

@@ -1,7 +1,7 @@
/**
* Copyright 2023, XGBoost Contributors
*/
#if defined(XGBOOST_USE_NCCL)
#if defined(XGBOOST_USE_NCCL) || defined(XGBOOST_USE_RCCL)
#include <gtest/gtest.h>
#include <thrust/device_vector.h> // for device_vector
#include <thrust/equal.h> // for equal

4
tests/cpp/collective/test_allgather.hip Normal file
View File

@@ -0,0 +1,4 @@
#if defined(XGBOOST_USE_HIP)
#include "test_allgather.cu"
#endif

2
tests/cpp/collective/test_allreduce.cu
View File

@@ -1,7 +1,7 @@
/**
* Copyright 2023, XGBoost Contributors
*/
#if defined(XGBOOST_USE_NCCL)
#if defined(XGBOOST_USE_NCCL) || defined(XGBOOST_USE_RCCL)
#include <gtest/gtest.h>
#include <thrust/host_vector.h> // for host_vector

4
tests/cpp/collective/test_allreduce.hip Normal file
View File

@@ -0,0 +1,4 @@
#if defined(XGBOOST_USE_HIP)
#include "test_allreduce.cu"
#endif