Added Unit test for nccl send recv (#506)

Added Send Receive test that tests through all pairs

[ROCm/rccl commit: ff54e79799]
This commit is contained in:
akolliasAMD
2022-03-02 15:50:16 -05:00
committed by GitHub
parent a182076a0e
commit 2419a950fe
8 changed files with 135 additions and 9 deletions
+19 -1
View File
@@ -23,7 +23,8 @@ namespace RcclUnitTesting
case ncclCollGather: return DefaultPrepData_Gather(collArgs, false);
case ncclCollScatter: return DefaultPrepData_Scatter(collArgs);
case ncclCollAllToAll: return DefaultPrepData_AllToAll(collArgs);
//case ncclCollSendRecv: return DefaultPrepData_SendRecv(collArgs);
case ncclCollSend: return DefaultPrepData_Send(collArgs);
case ncclCollRecv: return DefaultPrepData_Recv(collArgs);
default:
ERROR("Unknown func type %d\n", collArgs.funcType);
return TEST_FAIL;
@@ -339,4 +340,21 @@ namespace RcclUnitTesting
}
return TEST_SUCCESS;
}
ErrCode DefaultPrepData_Send(CollectiveArgs &collArgs)
{
CHECK_CALL(CheckAllocation(collArgs));
return collArgs.inputGpu.FillPattern(collArgs.dataType,
collArgs.numInputElements,
collArgs.globalRank, true);
}
ErrCode DefaultPrepData_Recv(CollectiveArgs &collArgs)
{
CHECK_CALL(CheckAllocation(collArgs));
return collArgs.expected.FillPattern(collArgs.dataType,
collArgs.numOutputElements,
collArgs.root,
false);
}
}