[CustomDevice] enable memory stat for custom device allocator (#61030)

PaddlePaddle · Jan 23, 2024 · 500e21b · 500e21b
1 parent 865b905
commit 500e21b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/memory/allocation/allocator_facade.cc b/paddle/fluid/memory/allocation/allocator_facade.cc
@@ -1349,7 +1349,7 @@ class AllocatorFacadePrivate {
       const platform::Place& place = pair.first;
       if (platform::is_cpu_place(place) ||
           platform::is_cuda_pinned_place(place) ||
-          platform::is_gpu_place(place)) {
+          platform::is_gpu_place(place) || platform::is_custom_place(place)) {
         pair.second = std::make_shared<StatAllocator>(pair.second);
       }
     }