[CustomKernel] phi capi add inference support (#44268)

paddle/fluid/inference/CMakeLists.txt

@@ -40,6 +40,10 @@ get_property(phi_modules GLOBAL PROPERTY PHI_MODULES)
 get_property(phi_kernels GLOBAL PROPERTY PHI_KERNELS)
 set(utils_modules stringpiece pretty_log string_helper benchmark)
 
+if(WITH_CUSTOM_DEVICE)
+  set(fluid_modules ${fluid_modules} phi_capi)
+endif()
+
 add_subdirectory(api)
 
 # Create static inference library if needed

paddle/fluid/inference/api/CMakeLists.txt

@@ -55,6 +55,9 @@ set(paddle_inference_api_deps
 if(WITH_CRYPTO)
   list(APPEND paddle_inference_api_deps paddle_crypto)
 endif()
+if(WITH_CUSTOM_DEVICE)
+  set(paddle_inference_api_deps ${paddle_inference_api_deps} phi_capi)
+endif()
 
 cc_library(
   paddle_inference_api

paddle/fluid/inference/api/api.cc

@@ -156,3 +156,7 @@ std::shared_ptr<framework::Cipher> MakeCipher(const std::string &config_file) {
 #endif
 
 }  // namespace paddle
+
+#ifdef PADDLE_WITH_CUSTOM_DEVICE
+#include "paddle/phi/capi/capi.h"
+#endif

paddle/fluid/inference/paddle_inference_custom_device.map

@@ -5,6 +5,7 @@
 		*profile*;
 		*phi*;
 		*FLAGS_*;
+		PD_*;
 	local:
 		*;
 };

paddle/phi/capi/include/c_tensor.h

@@ -24,7 +24,7 @@ extern "C" {
 
 typedef struct PD_Tensor PD_Tensor;
 
-PD_DataType PD_TensorGetDataType(const PD_Tensor *tensor, PD_Status *status);
+PD_DataType PD_TensorGetPDDataType(const PD_Tensor *tensor, PD_Status *status);
 
 PD_DataLayout PD_TensorGetDataLayout(const PD_Tensor *tensor,
                                      PD_Status *status);

paddle/phi/capi/include/wrapper_base.h

@@ -128,7 +128,7 @@ class DenseTensor : public WrapperBase<PD_Tensor> {
 
   PD_DataType dtype() const {
     C_Status status;
-    auto data_type = PD_TensorGetDataType(raw_data(), &status);
+    auto data_type = PD_TensorGetPDDataType(raw_data(), &status);
     PD_CHECK_STATUS(status);
     return data_type;
   }

paddle/phi/capi/lib/c_tensor.cc

@@ -19,7 +19,7 @@
 #include "paddle/phi/core/dense_tensor.h"
 #include "paddle/phi/core/meta_tensor.h"
 
-PD_DataType PD_TensorGetDataType(const PD_Tensor* tensor, PD_Status* status) {
+PD_DataType PD_TensorGetPDDataType(const PD_Tensor* tensor, PD_Status* status) {
   if (status) {
     if (!tensor) {
       *status = C_FAILED;