From be5dde9da895700cd30418e16c11135f7b71d237 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 29 Jul 2020 13:07:36 +0800 Subject: [PATCH] fix(mgb/opr-mm): fix group manager sync error GitOrigin-RevId: 344aae534c02cff8450884396a6cf9c0331ce6d0 --- src/opr-mm/impl/group_manager.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/opr-mm/impl/group_manager.cpp b/src/opr-mm/impl/group_manager.cpp index c5ba66c1..02d262dc 100644 --- a/src/opr-mm/impl/group_manager.cpp +++ b/src/opr-mm/impl/group_manager.cpp @@ -71,7 +71,6 @@ void GroupInfo::add_opr(const std::string& key, size_t nr_expected_devices, } m_opr_infos.push_back({comp_node_hash, is_root, rank}); m_nr_registered_devs++; - m_count++; if (m_nr_registered_devs > nr_expected_devices) { mgb_log_error( "too many opr registered with key %s, expected %zu, actual %u", @@ -84,6 +83,7 @@ void GroupInfo::add_opr(const std::string& key, size_t nr_expected_devices, if (m_nr_expected_devs == m_nr_registered_devs) { sort_opr_infos(); gen_infos_from_opr_infos(); + m_count = m_nr_registered_devs; m_register_cv.notify_all(); } else { m_register_cv.wait(lk,