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

[coll] Increase timeout for allgather test. (#9777)

Browse Source
This commit is contained in:
Jiaming Yuan 2023-11-09 05:26:40 +08:00 committed by GitHub
parent 44099f585d
commit 6fd4a30667
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -47,7 +47,7 @@ class Worker : public WorkerForTest {
std::size_t n = 8192; // n_bytes = 8192 * sizeof(int) std::size_t n = 8192; // n_bytes = 8192 * sizeof(int)
std::vector<std::int32_t> data(comm_.World() * n, 0); std::vector<std::int32_t> data(comm_.World() * n, 0);
auto s_data = common::Span{data.data(), data.size()}; auto s_data = common::Span<std::int32_t>{data};
auto seg = s_data.subspan(comm_.Rank() * n, n); auto seg = s_data.subspan(comm_.Rank() * n, n);
std::iota(seg.begin(), seg.end(), comm_.Rank()); std::iota(seg.begin(), seg.end(), comm_.Rank());

3
tests/cpp/collective/test_worker.h
View File

@ -92,11 +92,12 @@ class TrackerTest : public SocketTest {
template <typename WorkerFn> template <typename WorkerFn>
void TestDistributed(std::int32_t n_workers, WorkerFn worker_fn) { void TestDistributed(std::int32_t n_workers, WorkerFn worker_fn) {
std::chrono::seconds timeout{1}; std::chrono::seconds timeout{2};
std::string host; std::string host;
auto rc = GetHostAddress(&host); auto rc = GetHostAddress(&host);
ASSERT_TRUE(rc.OK()) << rc.Report(); ASSERT_TRUE(rc.OK()) << rc.Report();
LOG(INFO) << "Using " << n_workers << " workers for test.";
RabitTracker tracker{StringView{host}, n_workers, 0, timeout}; RabitTracker tracker{StringView{host}, n_workers, 0, timeout};
auto fut = tracker.Run(); auto fut = tracker.Run();