PaddlePaddle
diff --git a/‎paddle/fluid/distributed/collective/bkcl_tools.cc
Copy file name to clipboard
+24Lines changed: 24 additions & 0 deletions b/‎paddle/fluid/distributed/collective/bkcl_tools.cc
Copy file name to clipboard
+24Lines changed: 24 additions & 0 deletions
diff --git a/‎paddle/fluid/distributed/collective/bkcl_tools.h
Copy file name to clipboard
+2Lines changed: 2 additions & 0 deletions b/‎paddle/fluid/distributed/collective/bkcl_tools.h
Copy file name to clipboard
+2Lines changed: 2 additions & 0 deletions
@@ -41,5 +41,29 @@ std::string SerializeBKCLUniqueId(const BKCLUniqueId& bkclID) {
   return oss.str();
 }
 
+std::string BKCLDTypeToString(BKCLDataType dtype) {
+#define PD_BKCL_DTYPE_TO_STR(__bkcl_dtype, __str_dtype) \
+  if (dtype == __bkcl_dtype) return __str_dtype;
+  PD_BKCL_DTYPE_TO_STR(BKCL_FLOAT, "float32");
+  PD_BKCL_DTYPE_TO_STR(BKCL_FLOAT16, "float16");
+  PD_BKCL_DTYPE_TO_STR(BKCL_BFLOAT16, "bfloat16");
+  PD_BKCL_DTYPE_TO_STR(BKCL_FLOAT64, "float64");
+  PD_BKCL_DTYPE_TO_STR(BKCL_UINT8, "uint8");
+  PD_BKCL_DTYPE_TO_STR(BKCL_INT32, "int32");
+  PD_BKCL_DTYPE_TO_STR(BKCL_INT64, "int64");
+
+#undef PD_BKCL_DTYPE_TO_STR
+  PADDLE_THROW(phi::errors::InvalidArgument(
+      "This datatype %d in nccl is not supported.", static_cast<int>(dtype)));
+}
+
+std::string BKCLRedTypeToString(BKCLOp op) {
+  if (op == BKCL_ADD) return "SUM";
+  if (op == BKCL_PRODUCT) return "PROD";
+  if (op == BKCL_MIN) return "MIN";
+  if (op == BKCL_MAX) return "MAX";
+  return "UDF_" + std::to_string(op);
+}
+
 }  //  namespace distributed
 }  //  namespace paddle
@@ -102,6 +102,8 @@ class XPUEventManager {
 
 BKCLOp ToBKCLRedType(ReduceOp reduction);
 std::string SerializeBKCLUniqueId(const BKCLUniqueId& bkclId);
+std::string BKCLDTypeToString(BKCLDataType dtype);
+std::string BKCLRedTypeToString(BKCLOp op);
 
 }  // namespace distributed
 }  // namespace paddle