diff --git a/server/src/main/java/org/apache/uniffle/server/buffer/ShuffleBufferManager.java b/server/src/main/java/org/apache/uniffle/server/buffer/ShuffleBufferManager.java index 660cd5c5e7..f934f0f6e6 100644 --- a/server/src/main/java/org/apache/uniffle/server/buffer/ShuffleBufferManager.java +++ b/server/src/main/java/org/apache/uniffle/server/buffer/ShuffleBufferManager.java @@ -165,9 +165,7 @@ public ShuffleDataResult getShuffleData( void flushSingleBufferIfNecessary(ShuffleBuffer buffer, String appId, int shuffleId, int startPartition, int endPartition) { - // When we use multistorage and trigger single buffer flush, the buffer size should be bigger - // than rss.server.flush.cold.storage.threshold.size, otherwise cold storage will be useless. - if (this.bufferFlushEnabled && buffer.getSize() > this.bufferFlushThreshold) { + if (this.bufferFlushEnabled && buffer.getSize() >= this.bufferFlushThreshold) { flushBuffer(buffer, appId, shuffleId, startPartition, endPartition); } }