Adding FP16 cases to unit tests(#1093)

Signed-off-by: Tim Hu <timhu102@amd.com>
이 커밋은 다음에 포함됨:
Tim
2024-02-26 12:08:04 -05:00
커밋한 사람 GitHub
부모 74f9e5db64
커밋 0d06b0f1de
6개의 변경된 파일과 63개의 추가작업 그리고 14개의 파일을 삭제
+1 -1
파일 보기
@@ -13,7 +13,7 @@ namespace RcclUnitTesting
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollAllGather};
std::vector<ncclDataType_t> const dataTypes = {ncclFloat32};
std::vector<ncclDataType_t> const dataTypes = {ncclFloat16, ncclFloat32};
std::vector<ncclRedOp_t> const redOps = {ncclSum};
std::vector<int> const roots = {0};
std::vector<int> const numElements = {1048576, 500};