We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0261206 commit 7131228Copy full SHA for 7131228
1 file changed
tests/src/test/scala/com/nvidia/spark/rapids/GpuPartitioningSuite.scala
@@ -187,7 +187,7 @@ class GpuPartitioningSuite extends AnyFunSuite with BeforeAndAfterEach {
187
val partitionIndices = Array(0, 2, 2)
188
val gp = new GpuPartitioning {
189
override val numPartitions: Int = partitionIndices.length
190
- // Force the >2GB guard to trip on a small batch (no real 2GB allocation).
+ // Force the oversized-batch guard to trip with a 10-byte limit.
191
override protected[rapids] def maxGpuSerializedSliceBytes: Long = 10L
192
}
193
withResource(buildBatch()) { batch =>
@@ -262,4 +262,4 @@ class GpuPartitioningSuite extends AnyFunSuite with BeforeAndAfterEach {
262
263
264
265
-}
+}
0 commit comments