From 0080c970751cdac3f6b205553ef99a0138187599 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Thu, 21 Sep 2023 01:09:36 +0800 Subject: [PATCH] Workaround poll on macos. (#9596) --- rabit/src/allreduce_base.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/rabit/src/allreduce_base.cc b/rabit/src/allreduce_base.cc index 04246b5a1..416801ee2 100644 --- a/rabit/src/allreduce_base.cc +++ b/rabit/src/allreduce_base.cc @@ -549,7 +549,7 @@ AllreduceBase::TryAllreduceTree(void *sendrecvbuf_, break; } // select must return - auto poll_res = watcher.Poll(timeout_sec); + auto poll_res = watcher.Poll(timeout_sec, false); // fail on macos if (!poll_res.OK()) { LOG(FATAL) << poll_res.Report(); } @@ -814,7 +814,7 @@ AllreduceBase::TryAllgatherRing(void *sendrecvbuf_, size_t total_size, break; } - auto poll_res = watcher.Poll(timeout_sec); + auto poll_res = watcher.Poll(timeout_sec, false); // fail on macos if (!poll_res.OK()) { LOG(FATAL) << poll_res.Report(); } @@ -919,7 +919,7 @@ AllreduceBase::TryReduceScatterRing(void *sendrecvbuf_, if (finished) { break; } - auto poll_res = watcher.Poll(timeout_sec); + auto poll_res = watcher.Poll(timeout_sec, false); // fail on macos if (!poll_res.OK()) { LOG(FATAL) << poll_res.Report(); }