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() {