From c5b548c39887a9417ce52b4066568dd7375e1e47 Mon Sep 17 00:00:00 2001 From: avinashkethineedi Date: Wed, 5 Feb 2025 18:54:49 +0000 Subject: [PATCH] Fix `rocshmem_ctx_wg_team_sync` API - Updated `rocshmem_ctx_wg_team_sync` to utilize a team-specific memory buffer for synchronization --- src/ipc/context_ipc_device_coll.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/ipc/context_ipc_device_coll.cpp b/src/ipc/context_ipc_device_coll.cpp index 5ee574ba55..337b9106ca 100644 --- a/src/ipc/context_ipc_device_coll.cpp +++ b/src/ipc/context_ipc_device_coll.cpp @@ -105,8 +105,9 @@ __device__ void IPCContext::sync(rocshmem_team_t team) { int pe_start = team_obj->tinfo_wrt_world->pe_start; int pe_stride = team_obj->tinfo_wrt_world->stride; int pe_size = team_obj->num_pes; + long *p_sync = team_obj->barrier_pSync; - internal_sync(pe, pe_start, pe_stride, pe_size, barrier_sync); + internal_sync(pe, pe_start, pe_stride, pe_size, p_sync); } __device__ void IPCContext::sync_all() {