[XPU] support python memory api in XPU

Author: cqulilujia

paddle/fluid/pybind/pybind.cc

@@ -2930,6 +2930,7 @@ All parameter, weight, gradient are variables in Paddle.
 
 #ifdef PADDLE_WITH_XPU
   m.def("get_xpu_device_count", platform::GetXPUDeviceCount);
+  m.def("get_xpu_current_device_id", &platform::GetXPUCurrentDeviceId);
   m.def("xpu_empty_cache", platform::EmptyCache);
   m.def("get_xpu_device_utilization_rate",
         platform::GetXPUDeviceUtilizationRate);

paddle/phi/backends/xpu/xpu_info.cc

@@ -237,7 +237,7 @@ int GetXPUDeviceUtilizationRate(int dev_id) {
   return dev_util.xpu;
 }
 
-int GetXPUDeviceTotalMemory(int dev_id) {
+int64_t GetXPUDeviceTotalMemory(int dev_id) {
   std::call_once(xpuml_init_flag, xpumlInit);
   if (dev_id == -1) {
     dev_id = GetXPUCurrentDeviceId();
@@ -249,10 +249,10 @@ int GetXPUDeviceTotalMemory(int dev_id) {
   xpumlMemory_t dev_mem_info;
   PADDLE_ENFORCE_XPUML_SUCCESS(
       xpumlDeviceGetMemoryInfo(dev_handle, &dev_mem_info));
-  return dev_mem_info.totalGlobalMemory / 1024 / 1024;  // MB
+  return dev_mem_info.totalGlobalMemory;  // with Byte
 }
 
-int GetXPUDeviceUsedMemory(int dev_id) {
+int64_t GetXPUDeviceUsedMemory(int dev_id) {
   std::call_once(xpuml_init_flag, xpumlInit);
   if (dev_id == -1) {
     dev_id = GetXPUCurrentDeviceId();
@@ -264,7 +264,7 @@ int GetXPUDeviceUsedMemory(int dev_id) {
   xpumlMemory_t dev_mem_info;
   PADDLE_ENFORCE_XPUML_SUCCESS(
       xpumlDeviceGetMemoryInfo(dev_handle, &dev_mem_info));
-  return dev_mem_info.usedGlobalMemory / 1024 / 1024;  // MB
+  return dev_mem_info.usedGlobalMemory;  // with Byte
 }
 
 XPUVersion get_xpu_version(int dev_id) {

paddle/phi/backends/xpu/xpu_info.h

@@ -115,8 +115,8 @@ void set_xpu_debug_level(int level);
 
 int get_xpu_max_ptr_size(int dev_id);
 int GetXPUDeviceUtilizationRate(int dev_id);
-int GetXPUDeviceTotalMemory(int dev_id);
-int GetXPUDeviceUsedMemory(int dev_id);
+int64_t GetXPUDeviceTotalMemory(int dev_id);
+int64_t GetXPUDeviceUsedMemory(int dev_id);
 
 }  // namespace xpu
 }  // namespace backends

paddle/phi/core/platform/device/xpu/xpu_info.cc

@@ -123,11 +123,11 @@ int GetXPUDeviceUtilizationRate(int dev_id) {
   return phi::backends::xpu::GetXPUDeviceUtilizationRate(dev_id);
 }
 
-int GetXPUDeviceTotalMemory(int dev_id) {
+int64_t GetXPUDeviceTotalMemory(int dev_id) {
   return phi::backends::xpu::GetXPUDeviceTotalMemory(dev_id);
 }
 
-int GetXPUDeviceUsedMemory(int dev_id) {
+int64_t GetXPUDeviceUsedMemory(int dev_id) {
   return phi::backends::xpu::GetXPUDeviceUsedMemory(dev_id);
 }
 

paddle/phi/core/platform/device/xpu/xpu_info.h

@@ -95,8 +95,8 @@ bool IsXPUMallocRecorded(int dev_id);
 void EmptyCache(void);
 
 int GetXPUDeviceUtilizationRate(int dev_id);
-int GetXPUDeviceTotalMemory(int dev_id);
-int GetXPUDeviceUsedMemory(int dev_id);
+int64_t GetXPUDeviceTotalMemory(int dev_id);
+int64_t GetXPUDeviceUsedMemory(int dev_id);
 
 }  // namespace platform
 }  // namespace paddle