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

Modify All-To-All doc (#896)

Browse Source 
* Modify All-To-All doc

* Update nccl.h.in

* update unit-tests

---------

Co-authored-by: gilbertlee-amd <44450918+gilbertlee-amd@users.noreply.github.com>
This commit is contained in:
Bertan Dogancay
2023-09-22 09:37:30 -06:00
committed by Pedram Alizadeh
parent 50bc92f1d5
commit c1f57a7041
3 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/nccl.h.in
+2 -1
View File
@@ -639,7 +639,8 @@ ncclResult_t pncclScatter(const void* sendbuff, void* recvbuff,
@details Device (i) send (j)th block of data to device (j) and be placed as (i)th
block. Each block for sending/receiving has *count* elements, which means
that *recvbuff* and *sendbuff* should have a size of nranks*count elements.
In-place operation will happen if sendbuff == recvbuff.
In-place operation is NOT supported. It is the user's responsibility
to ensure that sendbuff and recvbuff are distinct.
@return Result code. See @ref rccl_result_code for more details.
@param[in] sendbuff Data array to send (contains blocks for each other rank)