fix xpu retry allocator bug (#54847)

PaddlePaddle · Jun 27, 2023 · 5bbbf5d · 5bbbf5d
1 parent 158b7ae
commit 5bbbf5d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);