support fp16 with ibm ddl (#1465)

Signed-off-by: Ahmed Elnaggar <ahmed.elnaggar@tum.de>
horovod · Oct 24, 2019 · bbf09d7 · bbf09d7
1 parent bb2134b
commit bbf09d7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/horovod/common/ops/ddl_operations.cc b/horovod/common/ops/ddl_operations.cc
@@ -23,6 +23,8 @@ DDL_Type GetDDLDataType(const std::shared_ptr<Tensor> tensor) {
   switch (tensor->dtype()) {
     case HOROVOD_FLOAT32:
       return DDL_TYPE_FLOAT;
+    case HOROVOD_FLOAT16:
+      return DDL_TYPE_HALF;
     default:
       throw std::logic_error("Type " + DataType_Name(tensor->dtype()) +
                              " is not supported in DDL mode.");