Add default context alltoall API (#350)

[ROCm/rocshmem commit: fddbe7b15d]
Этот коммит содержится в:
Yiltan
2025-12-10 11:43:15 -05:00
коммит произвёл GitHub
родитель 972893bab2
Коммит 258d264ecc
4 изменённых файлов: 69 добавлений и 9 удалений
+1
Просмотреть файл
@@ -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.