[CustomDevice] fix ToCDataType (PaddlePaddle#62562)

hitywt · Mar 11, 2024 · e9cd752 · e9cd752
1 parent 15a404c
commit e9cd752
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/paddle/phi/backends/custom/custom_device.cc b/paddle/phi/backends/custom/custom_device.cc
@@ -592,13 +592,21 @@ class CustomDevice : public DeviceInterface {
   case in:                     \
     return C_DataType::ret
     switch (data_type) {
-      return_result(phi::DataType::FLOAT64, FLOAT64);
-      return_result(phi::DataType::FLOAT32, FLOAT32);
-      return_result(phi::DataType::FLOAT16, FLOAT16);
-      return_result(phi::DataType::INT64, INT64);
-      return_result(phi::DataType::INT32, INT32);
-      return_result(phi::DataType::INT16, INT16);
+      return_result(phi::DataType::BOOL, BOOL);
+      return_result(phi::DataType::UINT8, UINT8);
+      return_result(phi::DataType::UINT16, UINT16);
+      return_result(phi::DataType::UINT32, UINT32);
+      return_result(phi::DataType::UINT64, UINT64);
       return_result(phi::DataType::INT8, INT8);
+      return_result(phi::DataType::INT16, INT16);
+      return_result(phi::DataType::INT32, INT32);
+      return_result(phi::DataType::INT64, INT64);
+      return_result(phi::DataType::FLOAT16, FLOAT16);
+      return_result(phi::DataType::FLOAT32, FLOAT32);
+      return_result(phi::DataType::FLOAT64, FLOAT64);
+      return_result(phi::DataType::BFLOAT16, BFLOAT16);
+      return_result(phi::DataType::COMPLEX64, COMPLEX64);
+      return_result(phi::DataType::COMPLEX128, COMPLEX128);
       default: {
         PADDLE_THROW(phi::errors::Unavailable(
             "DataType is not supported on %s.", Type()));