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

Update collective APIs naming (#77)

Browse Source 
* Update the naming convention for collective APIs to ensure consistency across the interface.

* Move all collective API declarations to rocshmem_COLL.hpp

* The following APIs were updated as part of this change:
  - `barrier`
  - `barrier_all`
  - `sync`
  - `sync_all`
  - `all_to_all`
  - `broadcast`
  - `fcollect`
  - `all_reduce`

* Update header file generation code for collective APIs

[ROCm/rocshmem commit: 68421895d6]
This commit is contained in:
Avinash Kethineedi
2025-04-10 12:14:47 -05:00
committed by GitHub
parent 5b22ddd1ff
commit 41d5d739e2
15 changed files with 334 additions and 334 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocshmem/tests/functional_tests/team_broadcast_tester.cpp
+1 -1
View File
@@ -34,7 +34,7 @@ __device__ void wg_team_broadcast(rocshmem_ctx_t ctx, rocshmem_team_t team,
__device__ void wg_team_broadcast<T>( \
rocshmem_ctx_t ctx, rocshmem_team_t team, T * dest, const T *source, \
int nelem, int pe_root) { \
rocshmem_ctx_##TNAME##_wg_broadcast(ctx, team, dest, source, nelem, \
rocshmem_ctx_##TNAME##_broadcast_wg(ctx, team, dest, source, nelem, \
pe_root); \
}