fix xpu retry allocator bug (#54847)

5bbbf5dd · ykkk2333 · GitHub · 158b7ae5 · 5bbbf5dd
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/platform/device/xpu/xpu_info.cc paddle/fluid/platform/device/xpu/xpu_info.cc +1 -1

未找到文件。
--- a/paddle/fluid/platform/device/xpu/xpu_info.cc
+++ b/paddle/fluid/platform/device/xpu/xpu_info.cc
@@ -98,7 +98,7 @@ phi::backends::xpu::XPUVersion get_xpu_version(int dev_id) {
 size_t XPUMinChunkSize() { return 1 << 6; }

 static void RaiseNonOutOfMemoryError(int status) {
-  if (status == XPUERR_NOMEM) {
+  if (-1 * status == XPUERR_NOMEM) {
    status = XPU_SUCCESS;
  }
  PADDLE_ENFORCE_XRE_SUCCESS(status);