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: c5b548c398]
Этот коммит содержится в:
avinashkethineedi
2025-02-05 18:54:49 +00:00
родитель 2214d21491
Коммит fca7471d67
+2 -1
Просмотреть файл
@@ -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() {