introduce support for ncclFloat16/half in UT

Bu işleme şunda yer alıyor:
Edgar Gabriel
2022-08-23 19:08:38 +00:00
ebeveyn 3c1b1ec8c8
işleme f6e00dec13
8 değiştirilmiş dosya ile 19 ekleme ve 9 silme
+1 -1
Dosyayı Görüntüle
@@ -13,7 +13,7 @@ namespace RcclUnitTesting
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollGather};
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};
std::vector<int> const roots = {0};
std::vector<int> const numElements = {1048576, 53327, 1024};