[coll] Prevent race during error check. (#10319)

dmlc · May 28, 2024 · 2de67f0 · 2de67f0
1 parent 7354955
commit 2de67f0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/collective/loop.cc b/src/collective/loop.cc
@@ -185,7 +185,8 @@ void Loop::Process() {
       if (!rc.OK()) {
         set_rc(std::move(rc));
       } else {
-        CHECK(qcopy.empty());
+        std::unique_lock lock{mu_};
+        CHECK(qcopy.empty() || stop_);
       }
     } catch (std::exception const& e) {
       curr_exce_ = std::current_exception();