static void _reset_coll_ufwd(pmixp_coll_t *coll)
{
/* upward status */
coll->contrib_children = 0;
coll->contrib_local = false;
memset(coll->contrib_chld, 0,
sizeof(coll->contrib_chld[0]) * coll->chldrn_cnt);
coll->serv_offs = pmixp_server_buf_reset(coll->ufwd_buf);
if (SLURM_SUCCESS != _pack_coll_info(coll, coll->ufwd_buf)) {
PMIXP_ERROR("Cannot pack ranges to message header!");
}
coll->ufwd_offset = get_buf_offset(coll->ufwd_buf);
coll->ufwd_status = PMIXP_COLL_SND_NONE;
}
static void _reset_coll_dfwd(pmixp_coll_t *coll)
{
/* downwards status */
(void)pmixp_server_buf_reset(coll->dfwd_buf);
if (SLURM_SUCCESS != _pack_coll_info(coll, coll->dfwd_buf)) {
PMIXP_ERROR("Cannot pack ranges to message header!");
}
coll->dfwd_cb_cnt = 0;
coll->dfwd_cb_wait = 0;
coll->dfwd_status = PMIXP_COLL_SND_NONE;
coll->contrib_prnt = false;
/* Save the toal service offset */
coll->dfwd_offset = get_buf_offset(coll->dfwd_buf);
}
static void _ring_sent_cb(int rc, pmixp_p2p_ctx_t ctx, void *_cbdata)
{
pmixp_coll_ring_cbdata_t *cbdata = (pmixp_coll_ring_cbdata_t*)_cbdata;
pmixp_coll_ring_ctx_t *coll_ctx = cbdata->coll_ctx;
pmixp_coll_t *coll = cbdata->coll;
Buf buf = cbdata->buf;
pmixp_coll_sanity_check(coll);
if (PMIXP_P2P_REGULAR == ctx) {
/* lock the collective */
slurm_mutex_lock(&coll->lock);
}
#ifdef PMIXP_COLL_DEBUG
PMIXP_DEBUG("%p: called %d", coll_ctx, coll_ctx->seq);
#endif
if (cbdata->seq != coll_ctx->seq) {
/* it seems like this collective was reset since the time
* we initiated this send.
* Just exit to avoid data corruption.
*/
PMIXP_DEBUG("%p: collective was reset!", coll_ctx);
goto exit;
}
coll_ctx->forward_cnt++;
_progress_coll_ring(coll_ctx);
exit:
pmixp_server_buf_reset(buf);
list_push(coll->state.ring.fwrd_buf_pool, buf);
if (PMIXP_P2P_REGULAR == ctx) {
/* unlock the collective */
slurm_mutex_unlock(&coll->lock);
}
xfree(cbdata);
}
