[XPU] fix xpu pin_memory stats (PaddlePaddle#76147)

* [xpu] fix_xpu_pin_memory_stats

Author: DongBaiYue

paddle/phi/common/place.cc

@@ -191,6 +191,11 @@ TEST_API bool is_cpu_place(const Place &p) {
   return p.GetType() == phi::AllocationType::CPU;
 }
 
+bool is_pinned_place(const Place &p) {
+  return p.GetType() == phi::AllocationType::GPUPINNED ||
+         p.GetType() == phi::AllocationType::XPUPINNED;
+}
+
 bool is_cuda_pinned_place(const Place &p) {
   return p.GetType() == phi::AllocationType::GPUPINNED;
 }

paddle/phi/common/place.h

@@ -225,6 +225,7 @@ PADDLE_API bool is_gpu_place(const Place&);
 PADDLE_API bool is_xpu_place(const Place&);
 PADDLE_API bool is_ipu_place(const Place&);
 PADDLE_API bool is_cpu_place(const Place&);
+PADDLE_API bool is_pinned_place(const Place&);
 PADDLE_API bool is_cuda_pinned_place(const Place&);
 PADDLE_API bool is_xpu_pinned_place(const Place&);
 PADDLE_API bool is_custom_place(const Place& p);

paddle/phi/core/memory/allocation/stat_allocator.h

@@ -32,7 +32,7 @@ class StatAllocator : public Allocator {
  protected:
   void FreeImpl(phi::Allocation* allocation) override {
     if (phi::is_cpu_place(allocation->place()) ||
-        phi::is_cuda_pinned_place(allocation->place())) {
+        phi::is_pinned_place(allocation->place())) {
       HOST_MEMORY_STAT_UPDATE(
           Allocated, allocation->place().GetDeviceId(), -allocation->size());
     } else {
@@ -51,7 +51,7 @@ class StatAllocator : public Allocator {
         underlying_allocator_->Allocate(size);
 
     const phi::Place& place = allocation->place();
-    if (phi::is_cpu_place(place) || phi::is_cuda_pinned_place(place)) {
+    if (phi::is_cpu_place(place) || phi::is_pinned_place(place)) {
       HOST_MEMORY_STAT_UPDATE(
           Allocated, place.GetDeviceId(), allocation->size());
     } else {