This is an automated email from the ASF dual-hosted git repository.
chengchengjin pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/incubator-gluten.git
The following commit(s) were added to refs/heads/main by this push:
new 27d6c220a5 [GLUTEN-8851][VL] Fix the
TakeOrderedAndProjectExecTransformer does not respect cudf tag (#10821)
27d6c220a5 is described below
commit 27d6c220a52e085a76e539daa1ab5e16a1074019
Author: Jin Chengcheng <[email protected]>
AuthorDate: Mon Sep 29 18:58:01 2025 +0100
[GLUTEN-8851][VL] Fix the TakeOrderedAndProjectExecTransformer does not
respect cudf tag (#10821)
---
.../gluten/execution/TakeOrderedAndProjectExecTransformer.scala | 4 ++++
1 file changed, 4 insertions(+)
diff --git
a/gluten-substrait/src/main/scala/org/apache/gluten/execution/TakeOrderedAndProjectExecTransformer.scala
b/gluten-substrait/src/main/scala/org/apache/gluten/execution/TakeOrderedAndProjectExecTransformer.scala
index f664c8a1d6..2887812c8a 100644
---
a/gluten-substrait/src/main/scala/org/apache/gluten/execution/TakeOrderedAndProjectExecTransformer.scala
+++
b/gluten-substrait/src/main/scala/org/apache/gluten/execution/TakeOrderedAndProjectExecTransformer.scala
@@ -143,6 +143,10 @@ case class TakeOrderedAndProjectExecTransformer(
} else {
val limitStagePlan =
WholeStageTransformer(limitBeforeShuffle)(transformStageCounter.incrementAndGet())
+ val cudfTag = child.getTagValue(CudfTag.CudfTag)
+ if (cudfTag.isDefined) {
+ limitStagePlan.setTagValue(CudfTag.CudfTag, cudfTag.get)
+ }
val shuffleExec = ShuffleExchangeExec(SinglePartition, limitStagePlan)
val transformedShuffleExec =
ColumnarShuffleExchangeExec(shuffleExec, limitStagePlan,
shuffleExec.child.output)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
