From de2b9877f0eb8983da9d82a106d496ae601b4755 Mon Sep 17 00:00:00 2001 From: Chengcheng Jin Date: Tue, 23 Sep 2025 17:48:05 +0100 Subject: [PATCH] [GLUTEN_8851][VL] Fix the TakeOrderedAndProjectExecTransformer does not respect cudf tag --- .../gluten/execution/TakeOrderedAndProjectExecTransformer.scala | 1 + 1 file changed, 1 insertion(+) 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 f664c8a1d696..ba24786c2f25 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,7 @@ case class TakeOrderedAndProjectExecTransformer( } else { val limitStagePlan = WholeStageTransformer(limitBeforeShuffle)(transformStageCounter.incrementAndGet()) + limitStagePlan.copyTagsFrom(child) val shuffleExec = ShuffleExchangeExec(SinglePartition, limitStagePlan) val transformedShuffleExec = ColumnarShuffleExchangeExec(shuffleExec, limitStagePlan, shuffleExec.child.output)