introduce support for ncclFloat16/half in UT

2022-08-23 19:08:38 +00:00
parent 3c1b1ec8c8
commit f6e00dec13
8 changed files with 19 additions and 9 deletions
@@ -47,7 +47,7 @@ namespace RcclUnitTesting
  {
    TestBed testBed;
    // Configuration
-    std::vector<ncclDataType_t> const& dataTypes       = {ncclInt32, ncclFloat64};
+    std::vector<ncclDataType_t> const& dataTypes       = {ncclInt32, ncclFloat64, ncclFloat16};
    std::vector<int>            const  numElements     = {1048576, 53327, 1024};
    bool                        const  inPlace         = false;
    bool                        const  useManagedMem   = false;