Adding FP16 cases to unit tests(#1093)

Signed-off-by: Tim Hu <timhu102@amd.com>
2024-02-26 12:08:04 -05:00
parent 74f9e5db64
commit 0d06b0f1de
6 changed files with 63 additions and 14 deletions
@@ -12,7 +12,7 @@ namespace RcclUnitTesting
    TestBed testBed;

    // Configuration
-    std::vector<ncclDataType_t> const& dataTypes       = {ncclInt32, ncclFloat64};
+    std::vector<ncclDataType_t> const& dataTypes       = {ncclInt32, ncclFloat16, ncclFloat64};
    std::vector<int>            const  numElements     = {1048576, 53327, 1024, 0};
    bool                        const  inPlace         = false;
    bool                        const  useManagedMem   = false;