Skip to content

Commit 9dfa35f

Browse files
Mamzi Bayatpour  mbayatpour@nvidia.com ()janjust
andcommitted
OSC/UCX: Adding the unpacked rkey counter for dynamic windows
Signed-off-by: Mamzi Bayatpour <mbayatpour@nvidia.com> Co-authored-by: Tomislav Janjusic <tomislavj@nvidia.com>
1 parent 195c3cb commit 9dfa35f

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

ompi/mca/osc/ucx/osc_ucx_comm.c

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -339,13 +339,16 @@ static inline int get_dynamic_win_info(uint64_t remote_addr,
339339

340340
if (mem_rec->rkeys[target] != NULL) {
341341
ucp_rkey_destroy(mem_rec->rkeys[target]);
342+
OPAL_COMMON_UCX_DEBUG_ATOMIC_ADD(opal_common_ucx_unpacked_rkey_counts, -1);
342343
}
343344

344345
void *rkey_buffer = &temp_dynamic_wins[contain].mem_addr;
345346

346347
ret = ucp_ep_rkey_unpack(mem_rec->winfo->endpoints[target], rkey_buffer,
347348
&mem_rec->rkeys[target]);
348349

350+
OPAL_COMMON_UCX_DEBUG_ATOMIC_ADD(opal_common_ucx_unpacked_rkey_counts, 1);
351+
349352
opal_mutex_unlock(&mem_rec->winfo->mutex);
350353

351354
if (ret != UCS_OK) {

0 commit comments

Comments
 (0)