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
@@ -95,8 +95,7 @@ namespace RcclUnitTesting
      dataTypes.push_back(ncclUint32);
      dataTypes.push_back(ncclInt64);
      dataTypes.push_back(ncclUint64);
-      // Half-precision floats disabled due to lack of host-side support
-      // dataTypes.push_back(ncclFloat16);
+      dataTypes.push_back(ncclFloat16);
      dataTypes.push_back(ncclFloat32);
      dataTypes.push_back(ncclFloat64);
      dataTypes.push_back(ncclBfloat16);