| 96 | } |
| 97 | |
| 98 | void GroupInfo::clear() { |
| 99 | std::unique_lock<std::mutex> lk{m_group_mtx}; |
| 100 | m_count--; |
| 101 | if (m_count == 0) { |
| 102 | m_opr_infos.clear(); |
| 103 | m_rank_map.clear(); |
| 104 | m_root_rank = -1; |
| 105 | m_nr_expected_devs = 0; |
| 106 | m_nr_registered_devs = 0; |
| 107 | m_output_shape.invalidate(); |
| 108 | m_clear_cv.notify_all(); |
| 109 | } else { |
| 110 | m_clear_cv.wait(lk, [&] { return m_count == 0; }); |
| 111 | } |
| 112 | } |
| 113 | |
| 114 | /* ================= GroupManager ================= */ |
| 115 |