share_external_memory of python api support xpu (PaddlePaddle#55170)

cqulilujia · Jul 24, 2023 · 2e1b3a0 · 2e1b3a0
1 parent 89f873c
commit 2e1b3a0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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;
   }