From fca7471d67a775857a23f34c31cf0660e81aa859 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 [ROCm/rocshmem commit: c5b548c39887a9417ce52b4066568dd7375e1e47] --- projects/rocshmem/src/ipc/context_ipc_device_coll.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/projects/rocshmem/src/ipc/context_ipc_device_coll.cpp b/projects/rocshmem/src/ipc/context_ipc_device_coll.cpp index 5ee574ba55..337b9106ca 100644 --- a/projects/rocshmem/src/ipc/context_ipc_device_coll.cpp +++ b/projects/rocshmem/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() {