From 797770da8da369a55266379fedbca5454680979b Mon Sep 17 00:00:00 2001 From: Xiduo You Date: Fri, 10 Nov 2023 14:58:16 +0800 Subject: [PATCH] [CORE] Fix ExpandExecTransformer miss input (#3665) --- .../io/glutenproject/execution/ExpandExecTransformer.scala | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/gluten-core/src/main/scala/io/glutenproject/execution/ExpandExecTransformer.scala b/gluten-core/src/main/scala/io/glutenproject/execution/ExpandExecTransformer.scala index 245e40d6c42c..0a76d53a64e8 100644 --- a/gluten-core/src/main/scala/io/glutenproject/execution/ExpandExecTransformer.scala +++ b/gluten-core/src/main/scala/io/glutenproject/execution/ExpandExecTransformer.scala @@ -50,7 +50,10 @@ case class ExpandExecTransformer( @transient override lazy val metrics = BackendsApiManager.getMetricsApiInstance.genExpandTransformerMetrics(sparkContext) - val originalInputAttributes: Seq[Attribute] = child.output + @transient + override lazy val references: AttributeSet = { + AttributeSet.fromAttributeSets(projections.flatten.map(_.references)) + } override def metricsUpdater(): MetricsUpdater = BackendsApiManager.getMetricsApiInstance.genExpandTransformerMetricsUpdater(metrics)