diff --git a/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/GpuFileSourceScanExec.scala b/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/GpuFileSourceScanExec.scala index 80acf2b0f9e..b4fca369f71 100644 --- a/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/GpuFileSourceScanExec.scala +++ b/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/GpuFileSourceScanExec.scala @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020-2023, NVIDIA CORPORATION. + * Copyright (c) 2020-2024, NVIDIA CORPORATION. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -633,15 +633,16 @@ case class GpuFileSourceScanExec( } override def doCanonicalize(): GpuFileSourceScanExec = { + logInfo("doCanonicalize with a reuse fix") GpuFileSourceScanExec( relation, - output.map(QueryPlan.normalizeExpressions(_, output)), + originalOutput.map(QueryPlan.normalizeExpressions(_, originalOutput)), requiredSchema, QueryPlan.normalizePredicates( - filterUnusedDynamicPruningExpressions(partitionFilters), output), + filterUnusedDynamicPruningExpressions(partitionFilters), originalOutput), optionalBucketSet, optionalNumCoalescedBuckets, - QueryPlan.normalizePredicates(dataFilters, output), + QueryPlan.normalizePredicates(dataFilters, originalOutput), None, queryUsesInputFile, alluxioPathsMap = alluxioPathsMap)(rapidsConf)