share_external_memory of python api support xpu (#55170)

90e37d19 · shentanyue · GitHub · 88402cdb · 90e37d19
显示空白变更内容
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/pybind/inference_api.cc paddle/fluid/pybind/inference_api.cc +2 -0

未找到文件。
--- a/paddle/fluid/pybind/inference_api.cc
+++ b/paddle/fluid/pybind/inference_api.cc
@@ -243,6 +243,8 @@ paddle_infer::PlaceType ToPaddleInferPlace(
    return paddle_infer::PlaceType::kCPU;
  } else if (allocation_type == phi::AllocationType::GPU) {
    return paddle_infer::PlaceType::kGPU;
+  } else if (allocation_type == phi::AllocationType::XPU) {
+    return paddle_infer::PlaceType::kXPU;
  } else {
    return paddle_infer::PlaceType::kCPU;
  }