Merge pull request #462 from InfiniTensor/issue/434-cambricon

issue/434 - added bf16 support for Cambricon MLU

src/infiniccl/cambricon/infiniccl_cambricon.cc

@@ -25,6 +25,8 @@ inline cnclDataType_t getCnclDtype(infiniDtype_t datatype) {
         return cnclFloat32;
     case INFINI_DTYPE_F16:
         return cnclFloat16;
+    case INFINI_DTYPE_BF16:
+        return cnclBfloat16;
     default:
         std::cerr << "Unsupported data type: " << datatype << std::endl;
         std::abort();
@@ -89,9 +91,7 @@ infiniStatus_t allReduce(
     infinicclComm_t comm,
     infinirtStream_t stream) {
 
-    if (datatype != INFINI_DTYPE_F32 && datatype != INFINI_DTYPE_F16) {
-        return INFINI_STATUS_BAD_PARAM;
-    }
+    CHECK_DTYPE(datatype, INFINI_DTYPE_F32, INFINI_DTYPE_F16, INFINI_DTYPE_BF16);
 
     CHECK_CNCL(cnclAllReduce(sendbuf, recvbuf, count, getCnclDtype(datatype),
                              getCnclRedOp(op), getCnclComm(comm),
@@ -99,4 +99,5 @@ infiniStatus_t allReduce(
 
     return INFINI_STATUS_SUCCESS;
 }
+
 } // namespace infiniccl::cambricon