Fix nit

NVIDIA · Sep 21, 2023 · f5bfed2 · f5bfed2
1 parent 2b83cdc
commit f5bfed2
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/RapidsHostMemoryStore.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/RapidsHostMemoryStore.scala
@@ -36,7 +36,6 @@ import org.apache.spark.sql.vectorized.ColumnarBatch
 /**
  * A buffer store using host memory.
  * @param maxSize maximum size in bytes for all buffers in this store
- * @param pageableMemoryPoolSize maximum size in bytes for the internal pageable memory pool
  */
 class RapidsHostMemoryStore(
     maxSize: Option[Long])
@@ -92,7 +91,7 @@ class RapidsHostMemoryStore(
       // this spillStore has a maximum size requirement (host only). We need to spill from it
       // in order to make room for `buffer`.
       val targetTotalSize = ms - buffer.memoryUsedBytes
-      if (targetTotalSize <= 0) {
+      if (targetTotalSize < 0) {
         // lets not spill to host when the buffer we are about
         // to spill is larger than our limit
         false