[PHI CAPI] support complex dtype kernel (#52414)

* [PHI CAPI] support complex dtype kernel

* update

paddle/fluid/operators/utils.h

@@ -92,6 +92,11 @@ inline T GetValue(const phi::DenseTensor* x) {
   if (!platform::is_cpu_place(x->place())) {
     phi::DenseTensor cpu_x;
     framework::TensorCopy(*x, platform::CPUPlace(), &cpu_x);
+#if defined(PADDLE_WITH_CUSTOM_DEVICE)
+    platform::DeviceContextPool& pool = platform::DeviceContextPool::Instance();
+    const platform::DeviceContext* dev_ctx = pool.Get(x->place());
+    dev_ctx->Wait();
+#endif
     value = cpu_x.data<T>()[0];
   } else {
     value = x->data<T>()[0];

paddle/phi/backends/device_ext.h

@@ -40,6 +40,8 @@ typedef enum {
   FLOAT32,
   FLOAT64,
   BFLOAT16,
+  COMPLEX64,
+  COMPLEX128,
 } C_DataType;
 
 typedef enum {

paddle/phi/capi/include/data_type.h

@@ -35,7 +35,9 @@ namespace capi {
   _(int8_t, PD_DataType::INT8)                          \
   _(int16_t, PD_DataType::INT16)                        \
   _(int32_t, PD_DataType::INT32)                        \
-  _(int64_t, PD_DataType::INT64)
+  _(int64_t, PD_DataType::INT64)                        \
+  _(phi::dtype::complex<float>, PD_DataType::COMPLEX64) \
+  _(phi::dtype::complex<double>, PD_DataType::COMPLEX128)
 
 template <typename T>
 struct CppTypeToPDType;

paddle/phi/capi/include/type_utils.h

@@ -42,6 +42,8 @@ inline PD_DataType ToPDDataType(::phi::DataType dtype) {
     return_result(UINT16, UINT16);
     return_result(UINT8, UINT8);
     return_result(BOOL, BOOL);
+    return_result(COMPLEX64, COMPLEX64);
+    return_result(COMPLEX128, COMPLEX128);
     default: {
       PADDLE_THROW(
           ::phi::errors::Unavailable("DataType %d is not supported.", dtype));
@@ -69,6 +71,8 @@ inline ::phi::DataType ToPhiDataType(PD_DataType dtype) {
     return_result(UINT16, UINT16);
     return_result(UINT8, UINT8);
     return_result(BOOL, BOOL);
+    return_result(COMPLEX64, COMPLEX64);
+    return_result(COMPLEX128, COMPLEX128);
     default: {
       PADDLE_THROW(
           ::phi::errors::Unavailable("DataType %d is not supported.", dtype));