introduce support for ncclFloat16/half in UT

[ROCm/rccl commit: f6e00dec13]
Cette révision appartient à :
Edgar Gabriel
2022-08-23 19:08:38 +00:00
Parent 1d55fe756c
révision b32b819151
8 fichiers modifiés avec 19 ajouts et 9 suppressions
+1 -1
Voir le fichier
@@ -13,7 +13,7 @@ namespace RcclUnitTesting
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollAllReduce};
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 = {ncclSum, ncclMin};
std::vector<int> const roots = {0};
std::vector<int> const numElements = {1048576, 53327, 1024};