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

Add default context alltoall API (#350)

Browse Source 
[ROCm/rocshmem commit: fddbe7b15d]
This commit is contained in:
Yiltan authored and GitHub committed 2025-12-10 11:43:15 -05:00
1 parent 972893bab2
commit 258d264ecc
4 files changed
+69 -9
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

projects/rocshmem/docs/api/coll.rst
+1
View File
@@ -85,6 +85,7 @@ These APIs should be called from only one thread/wavefront/workgroup within the
ROSHMEM_ALLTOALL
----------------
.. cpp:function:: __device__ void rocshmem_TYPENAME_alltoall_wg(rocshmem_team_t team, TYPE *dest, const TYPE *source, int nelems)
.. cpp:function:: __device__ void rocshmem_ctx_TYPENAME_alltoall_wg(rocshmem_ctx_t ctx, rocshmem_team_t team, TYPE *dest, const TYPE *source, int nelems)
:param team: The team participating in the collective.