Skip to content

Add fence_nb to flux pmix #8380

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Jan 19, 2021
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 20 additions & 0 deletions opal/mca/pmix/flux/pmix_flux.c
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,8 @@ static int flux_abort(int flag, const char msg[],
opal_list_t *procs);
static int flux_commit(void);
static int flux_fence(opal_list_t *procs, int collect_data);
static int flux_fencenb(opal_list_t *procs, int collect_data,
opal_pmix_op_cbfunc_t cbfunc, void *cbdata);
static int flux_put(opal_pmix_scope_t scope,
opal_value_t *kv);
static int flux_get(const opal_process_name_t *id,
Expand All @@ -65,6 +67,7 @@ const opal_pmix_base_module_t opal_pmix_flux_module = {
.abort = flux_abort,
.commit = flux_commit,
.fence = flux_fence,
.fence_nb = flux_fencenb,
.put = flux_put,
.get = flux_get,
.publish = flux_publish,
Expand Down Expand Up @@ -675,6 +678,23 @@ static int flux_commit(void)
return OPAL_SUCCESS;
}

static int flux_fencenb(opal_list_t *procs, int collect_data,
opal_pmix_op_cbfunc_t cbfunc, void *cbdata)
{
int rc;
int status;
if (PMI_SUCCESS != (rc = PMI_Barrier())) {
OPAL_PMI_ERROR(rc, "PMI_Barrier");
return OPAL_ERROR;
}
if (NULL == cbfunc) {
return OPAL_ERROR;
} else {
cbfunc(status, cbdata);
}
return OPAL_SUCCESS;
}

static int flux_fence(opal_list_t *procs, int collect_data)
{
int rc;
Expand Down