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

没有与搜索条件相匹配的文件

+1 -1
查看文件
@@ -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};