From 90e37d194ee69a5ebdf306464630fe90ff9b3031 Mon Sep 17 00:00:00 2001
From: shentanyue <34421038+shentanyue@users.noreply.github.com>
Date: Thu, 6 Jul 2023 14:52:20 +0800
Subject: [PATCH] share_external_memory of python api support xpu (#55170)

---
 paddle/fluid/pybind/inference_api.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/paddle/fluid/pybind/inference_api.cc b/paddle/fluid/pybind/inference_api.cc
index bdf54bd76b6..7b8fc63d298 100644
--- 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;
   }
-- 
GitLab