Adding FP16 cases to unit tests(#1093)

Signed-off-by: Tim Hu <timhu102@amd.com>

[ROCm/rccl commit: 0d06b0f1de]
此提交包含在:
Tim authored and GitHub committed 2024-02-26 12:08:04 -05:00
1 父節點 874998033f
當前提交 826d20495f
6 file changed
+63 -14

No files matched your search

+1 -1
查看文件
@@ -16,7 +16,7 @@ namespace RcclUnitTesting
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollAllToAll};
std::vector<ncclDataType_t> const dataTypes = {ncclFloat32};
std::vector<ncclDataType_t> const dataTypes = {ncclFloat16, ncclFloat32};
std::vector<ncclRedOp_t> const redOps = {ncclSum};
std::vector<int> const roots = {0};
std::vector<int> const numElements = {1048576, 1024};