introduce support for ncclFloat16/half in UT

[ROCm/rccl commit: f6e00dec13]
이 커밋은 다음에 포함됨:
Edgar Gabriel가 2022-08-24 15:28:24 +00:00를 커밋했습니다
1 부모 1d55fe756c
커밋 b32b819151
8 파일이 변경됨
+19 -9

검색과 일치하는 파일 없음

+2 -2
파일 보기
@@ -13,10 +13,10 @@ namespace RcclUnitTesting
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollReduceScatter};
std::vector<ncclDataType_t> const dataTypes = {ncclFloat32, ncclFloat64, ncclBfloat16};
std::vector<ncclDataType_t> const dataTypes = {ncclFloat16, ncclFloat32, ncclFloat64, ncclBfloat16};
std::vector<ncclRedOp_t> const redOps = {ncclMin, ncclMax, ncclAvg};
std::vector<int> const roots = {0};
std::vector<int> const numElements = {1048576, 53327, 1024};
std::vector<int> const numElements = {1048576, 53327, 5461, 1024};
std::vector<bool> const inPlaceList = {false};
std::vector<bool> const managedMemList = {false};