try again if kunlun memory malloc failed (#30855)

* try again if kunlun memory malloc failed

* minor

paddle/fluid/memory/allocation/naive_best_fit_allocator.cc

@@ -135,6 +135,11 @@ void *Alloc<platform::XPUPlace>(const platform::XPUPlace &place, size_t size) {
                         "Baidu Kunlun Card is properly installed.",
                         ret));
   ret = xpu_malloc(reinterpret_cast<void **>(&p), size);
+  if (ret != XPU_SUCCESS) {
+    std::cout << "xpu memory malloc(" << size << ") failed, try again\n";
+    xpu_wait();
+    ret = xpu_malloc(reinterpret_cast<void **>(&p), size);
+  }
   PADDLE_ENFORCE_EQ(
       ret, XPU_SUCCESS,
       platform::errors::External(