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

Added Unit test for nccl send recv (#506)

Browse Source 
Added Send Receive test that tests through all pairs
This commit is contained in:
akolliasAMD
2022-03-02 15:50:16 -05:00
committed by GitHub
parent 29ad0f5fbe
commit ff54e79799
8 changed files with 135 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/common/PrepDataFuncs.hpp
+2 -1
View File
@@ -22,5 +22,6 @@ namespace RcclUnitTesting
ErrCode DefaultPrepData_ReduceScatter(CollectiveArgs &collArgs);
ErrCode DefaultPrepData_Scatter(CollectiveArgs &collArgs);
ErrCode DefaultPrepData_AllToAll(CollectiveArgs &collArgs);
ErrCode DefaultPrepData_SendRecv(CollectiveArgs &collArgs);
ErrCode DefaultPrepData_Send(CollectiveArgs &collArgs);
ErrCode DefaultPrepData_Recv(CollectiveArgs &collArgs);
}