Adding FP16 cases to unit tests(#1093)

Signed-off-by: Tim Hu <timhu102@amd.com> [ROCm/rccl commit: 0d06b0f1de]
2024-02-26 12:08:04 -05:00
parent 874998033f
commit 826d20495f
6 changed files with 63 additions and 14 deletions
@@ -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};