AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #36 from avinashkethineedi/fix/rocshmem-ctx-wg-team-sync

Browse Source 
Fix `rocshmem_ctx_wg_team_sync` API

[ROCm/rocshmem commit: d97e5ba2c8]
This commit is contained in:
Avinash Kethineedi
2025-02-06 13:41:16 -06:00
committed by GitHub
parent 614d5c7c81 fca7471d67
commit 4d919faf55
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocshmem/src/ipc/context_ipc_device_coll.cpp
+2 -1
View File
@@ -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() {