Skip to content

Commit 7131228

Browse files
committed
Fix oversized batch test comment
Signed-off-by: Allen Xu <allxu@nvidia.com>
1 parent 0261206 commit 7131228

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/src/test/scala/com/nvidia/spark/rapids/GpuPartitioningSuite.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,7 @@ class GpuPartitioningSuite extends AnyFunSuite with BeforeAndAfterEach {
187187
val partitionIndices = Array(0, 2, 2)
188188
val gp = new GpuPartitioning {
189189
override val numPartitions: Int = partitionIndices.length
190-
// Force the >2GB guard to trip on a small batch (no real 2GB allocation).
190+
// Force the oversized-batch guard to trip with a 10-byte limit.
191191
override protected[rapids] def maxGpuSerializedSliceBytes: Long = 10L
192192
}
193193
withResource(buildBatch()) { batch =>
@@ -262,4 +262,4 @@ class GpuPartitioningSuite extends AnyFunSuite with BeforeAndAfterEach {
262262
}
263263
}
264264
}
265-
}
265+
}

0 commit comments

Comments
 (0)