add uint8 custom ccltype (#54671)

YanhuiDua · web-flow · commit 6fc0378afcaf · 2023-06-15T19:22:35.000+08:00
diff --git a/paddle/phi/backends/c_comm_lib.h b/paddle/phi/backends/c_comm_lib.h
@@ -34,7 +34,8 @@ enum CCLDataType {
   CCL_DATA_TYPE_INT64,
   CCL_DATA_TYPE_INT32,
   CCL_DATA_TYPE_INT16,
-  CCL_DATA_TYPE_INT8
+  CCL_DATA_TYPE_INT8,
+  CCL_DATA_TYPE_UINT8
 };
 
 inline CCLDataType ToCCLDataType(phi::DataType type) {
@@ -50,9 +51,12 @@ inline CCLDataType ToCCLDataType(phi::DataType type) {
     return CCL_DATA_TYPE_INT32;
   } else if (type == phi::DataType::INT8) {
     return CCL_DATA_TYPE_INT8;
+  } else if (type == phi::DataType::UINT8) {
+    return CCL_DATA_TYPE_UINT8;
   } else {
     PADDLE_THROW(
-        phi::errors::Unimplemented("This datatype in CCL is not supported."));
+        phi::errors::Unimplemented("This datatype %s in CCL is not supported.",
+                                   phi::DataTypeToString(type)));
   }
 }
 
diff --git a/paddle/phi/backends/custom/custom_device.cc b/paddle/phi/backends/custom/custom_device.cc
@@ -589,6 +589,7 @@ class CustomDevice : public DeviceInterface {
       return_result(CCL_DATA_TYPE_INT32, INT32);
       return_result(CCL_DATA_TYPE_INT16, INT16);
       return_result(CCL_DATA_TYPE_INT8, INT8);
+      return_result(CCL_DATA_TYPE_UINT8, UINT8);
       default: {
         PADDLE_THROW(phi::errors::Unavailable(
             "DataType is not supported on %s.", Type()));
