Skip to content

Commit

Permalink
spml/ucx: PR fixes
Browse files Browse the repository at this point in the history
Signed-off-by: Michal Shalev <mshalev.nvidia.com>
  • Loading branch information
michal-shalev committed Nov 6, 2024
1 parent 78d0a63 commit 32a86a4
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions oshmem/mca/spml/ucx/spml_ucx.c
Original file line number Diff line number Diff line change
Expand Up @@ -692,7 +692,7 @@ int mca_spml_ucx_add_procs(oshmem_group_t* group, size_t nprocs)
}
}

indices = malloc(nprocs * sizeof(int));
indices = malloc(nprocs * sizeof(*indices));
if (!indices) {
goto error;
}
Expand All @@ -719,7 +719,7 @@ int mca_spml_ucx_add_procs(oshmem_group_t* group, size_t nprocs)
err = ucp_ep_create(mca_spml_ucx_ctx_default.ucp_worker[0], &ep_params,
&mca_spml_ucx_ctx_default.ucp_peers[indices[proc_index]].ucp_conn);
if (UCS_OK != err) {
SPML_UCX_ERROR("ucp_ep_create(proc=%zu/%zu) failed: %s", n, nprocs,
SPML_UCX_ERROR("ucp_ep_create(proc=%d/%zu) failed: %s", proc_index, nprocs,
ucs_status_string(err));
goto error2;
}
Expand Down

0 comments on commit 32a86a4

Please sign in to comment.