fix

apache · Oct 7, 2024 · 34596c5 · 34596c5
1 parent 49563e7
commit 34596c5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/spark/src/main/java/org/apache/spark/CometTaskMemoryManager.java b/spark/src/main/java/org/apache/spark/CometTaskMemoryManager.java
@@ -63,9 +63,12 @@ public long acquireMemory(long size) {
         if (size <= available) {
           available -= size;
           return size;
+        } else {
+          long allocated = available;
+          available = 0;
+          return allocated;
         }
       }
-      return 0;
     }
   }