Skip to content

Commit 82b40cf

Browse files
committed
Fix the reader which was still using a dynamic flag, instead of the checksum flag in the gpu dependency
1 parent 619df93 commit 82b40cf

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

sql-plugin/src/main/scala/org/apache/spark/sql/rapids/RapidsShuffleInternalManagerBase.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2015,7 +2015,7 @@ class RapidsShuffleInternalManagerBase(conf: SparkConf, val isDriver: Boolean)
20152015
// Exchange node.
20162016
baseHandle.dependency match {
20172017
case gpuDep: GpuShuffleDependency[K, C, C]
2018-
if gpuDep.useMultiThreadedShuffle && !shouldFallThroughForShuffle =>
2018+
if gpuDep.useMultiThreadedShuffle && !gpuDep.checksumFallback =>
20192019
// We want to use batch fetch in the non-push shuffle case. Spark
20202020
// checks for a config to see if batch fetch is enabled (this check), and
20212021
// it also checks when getting (potentially merged) map status from

0 commit comments

Comments
 (0)