Pass dirs to getBlockData for a wrapped shuffle resolver (NVIDIA#2002)

Signed-off-by: Alessandro Bellina <[email protected]>
nartal1 · Mar 23, 2021 · 8746585 · 8746585
1 parent 203abf0
commit 8746585
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/RapidsShuffleInternalManagerBase.scala b/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/RapidsShuffleInternalManagerBase.scala
@@ -56,7 +56,7 @@ class GpuShuffleBlockResolver(
     if (hasActiveShuffle) {
       throw new IllegalStateException(s"The block $blockId is being managed by the catalog")
     }
-    wrapped.getBlockData(blockId)
+    wrapped.getBlockData(blockId, dirs)
   }
 
   override def stop(): Unit = wrapped.stop()