diff --git a/backends-velox/src/main/scala/org/apache/gluten/execution/ColumnarPartialProjectExec.scala b/backends-velox/src/main/scala/org/apache/gluten/execution/ColumnarPartialProjectExec.scala index 435fd239b364..500c70fefaf7 100644 --- a/backends-velox/src/main/scala/org/apache/gluten/execution/ColumnarPartialProjectExec.scala +++ b/backends-velox/src/main/scala/org/apache/gluten/execution/ColumnarPartialProjectExec.scala @@ -147,9 +147,6 @@ case class ColumnarPartialProjectExec(original: ProjectExec, child: SparkPlan)( } override protected def doValidateInternal(): ValidationResult = { - if (!GlutenConfig.get.enableColumnarPartialProject) { - return ValidationResult.failed("Config disable this feature") - } if (UDFAttrNotExists) { return ValidationResult.failed("Attribute in the UDF does not exists in its child") } diff --git a/backends-velox/src/main/scala/org/apache/gluten/extension/PartialProjectRule.scala b/backends-velox/src/main/scala/org/apache/gluten/extension/PartialProjectRule.scala index 73d1651e2fdb..f345a9a40c1d 100644 --- a/backends-velox/src/main/scala/org/apache/gluten/extension/PartialProjectRule.scala +++ b/backends-velox/src/main/scala/org/apache/gluten/extension/PartialProjectRule.scala @@ -16,6 +16,7 @@ */ package org.apache.gluten.extension +import org.apache.gluten.config.GlutenConfig import org.apache.gluten.execution.ColumnarPartialProjectExec import org.apache.spark.sql.SparkSession @@ -24,6 +25,9 @@ import org.apache.spark.sql.execution.{ProjectExec, SparkPlan} case class PartialProjectRule(spark: SparkSession) extends Rule[SparkPlan] { override def apply(plan: SparkPlan): SparkPlan = { + if (!GlutenConfig.get.enableColumnarPartialProject) { + return plan + } plan.transformUp { case plan: ProjectExec => val transformer = ColumnarPartialProjectExec.create(plan)