File tree Expand file tree Collapse file tree
spark/src/test/scala/org/apache/spark/sql/benchmark Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -124,7 +124,8 @@ trait CometBenchmarkBase
124124 val cometExecConfigs = Map (
125125 CometConf .COMET_ENABLED .key -> " true" ,
126126 CometConf .COMET_EXEC_ENABLED .key -> " true" ,
127- " spark.sql.optimizer.constantFolding.enabled" -> " false" ) ++ extraCometConfigs
127+ " spark.sql.optimizer.excludedRules" ->
128+ " org.apache.spark.sql.catalyst.optimizer.ConstantFolding" ) ++ extraCometConfigs
128129
129130 // Check that the plan is fully Comet native before running the benchmark
130131 withSQLConf(cometExecConfigs.toSeq: _* ) {
Original file line number Diff line number Diff line change @@ -100,7 +100,8 @@ object CometRegExpBenchmark extends CometBenchmarkBase {
100100 val baseExec = Map (
101101 CometConf .COMET_ENABLED .key -> " true" ,
102102 CometConf .COMET_EXEC_ENABLED .key -> " true" ,
103- " spark.sql.optimizer.constantFolding.enabled" -> " false" )
103+ " spark.sql.optimizer.excludedRules" ->
104+ " org.apache.spark.sql.catalyst.optimizer.ConstantFolding" )
104105
105106 benchmark.addCase(" Comet (Exec, native Rust regex)" ) { _ =>
106107 val configs = baseExec ++ Map (CometConf .getExprAllowIncompatConfigKey(" RLike" ) -> " true" )
Original file line number Diff line number Diff line change @@ -152,7 +152,8 @@ object CometRegExpExtractBenchmark extends CometBenchmarkBase {
152152 val baseExec = Map (
153153 CometConf .COMET_ENABLED .key -> " true" ,
154154 CometConf .COMET_EXEC_ENABLED .key -> " true" ,
155- " spark.sql.optimizer.constantFolding.enabled" -> " false" )
155+ " spark.sql.optimizer.excludedRules" ->
156+ " org.apache.spark.sql.catalyst.optimizer.ConstantFolding" )
156157
157158 benchmark.addCase(" Comet (Exec, native Rust regex)" ) { _ =>
158159 val configs =
You can’t perform that action at this time.
0 commit comments