introduce support for ncclFloat16/half in UT

[ROCm/rccl commit: f6e00dec13]
Αυτή η υποβολή περιλαμβάνεται σε:
Edgar Gabriel
2022-08-23 19:08:38 +00:00
γονέας 1d55fe756c
υποβολή b32b819151
8 αρχεία άλλαξαν με 19 προσθήκες και 9 διαγραφές
@@ -13,10 +13,10 @@ namespace RcclUnitTesting
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollReduceScatter};
std::vector<ncclDataType_t> const dataTypes = {ncclInt8, ncclInt32, ncclInt64};
std::vector<ncclDataType_t> const dataTypes = {ncclInt8, ncclInt32, ncclInt64, ncclFloat16};
std::vector<ncclRedOp_t> const redOps = {ncclSum, ncclProd};
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 = {true};
std::vector<bool> const managedMemList = {false};