Added Unit test for nccl send recv (#506)

Added Send Receive test that tests through all pairs
Этот коммит содержится в:
akolliasAMD
2022-03-02 15:50:16 -05:00
коммит произвёл GitHub
родитель 29ad0f5fbe
Коммит ff54e79799
8 изменённых файлов: 135 добавлений и 9 удалений
+19 -1
Просмотреть файл
@@ -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);
}
}