[XPU] update xccl version to 1.1.6.1 (PaddlePaddle#58807)

* [XPU] update xccl version to 1.1.6.1 * add datatype bf16 and bool for xccl
danleifeng · Nov 9, 2023 · 81a824d · 81a824d
1 parent 3d0cb2f
commit 81a824d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/cmake/external/xpu.cmake b/cmake/external/xpu.cmake
@@ -26,7 +26,7 @@ set(XPU_XPTI_LIB_NAME "libxpti.so")
 if(NOT DEFINED XPU_BASE_DATE)
   set(XPU_BASE_DATE "20231103")
 endif()
-set(XPU_XCCL_BASE_VERSION "1.0.53.6")
+set(XPU_XCCL_BASE_VERSION "1.1.6.1")
 if(NOT DEFINED XPU_XFT_BASE_VERSION)
   set(XPU_XFT_BASE_VERSION "20230602")
 endif()

diff --git a/paddle/fluid/platform/device/xpu/bkcl_helper.h b/paddle/fluid/platform/device/xpu/bkcl_helper.h
@@ -53,11 +53,16 @@ inline BKCLDataType ToBKCLDataType(framework::proto::VarType::Type type) {
     return BKCL_FLOAT64;
   } else if (type == framework::proto::VarType::FP16) {
     return BKCL_FLOAT16;
+  } else if (type == framework::proto::VarType::BF16) {
+    return BKCL_BFLOAT16;
   } else if (type == framework::proto::VarType::UINT8) {
     return BKCL_UINT8;
+  } else if (type == framework::proto::VarType::BOOL) {
+    return BKCL_UINT8;
   } else {
     PADDLE_THROW(platform::errors::Unimplemented(
-        "BKCL currently only support FP32, INT64, INT32, FP64, FP16 and UINT8, "
+        "BKCL currently only support FP32, INT64, INT32, FP64, FP16, BF16, "
+        "UINT8 and BOOL, "
         "other data types are not supported."));
   }
 }