diff --git a/spark/src/test/scala/org/apache/spark/sql/benchmark/CometTPCDSMicroBenchmark.scala b/spark/src/test/scala/org/apache/spark/sql/benchmark/CometTPCDSMicroBenchmark.scala index 016c2a371..40a84a125 100644 --- a/spark/src/test/scala/org/apache/spark/sql/benchmark/CometTPCDSMicroBenchmark.scala +++ b/spark/src/test/scala/org/apache/spark/sql/benchmark/CometTPCDSMicroBenchmark.scala @@ -124,6 +124,20 @@ object CometTPCDSMicroBenchmark extends CometTPCQueryBenchmarkBase { cometSpark.sql(queryString).noop() } } + benchmark.addCase(s"$name$nameSuffix: Comet (Exec)") { _ => + withSQLConf( + CometConf.COMET_ENABLED.key -> "true", + CometConf.COMET_NATIVE_SCAN_ENABLED.key -> "false", + CometConf.COMET_CONVERT_FROM_PARQUET_ENABLED.key -> "true", + CometConf.COMET_EXEC_SHUFFLE_ENABLED.key -> "true", + CometConf.COMET_SHUFFLE_MODE.key -> "auto", + CometConf.COMET_REGEXP_ALLOW_INCOMPATIBLE.key -> "true", + // enabling COMET_EXPLAIN_NATIVE_ENABLED may add overhead but is useful for debugging + CometConf.COMET_EXPLAIN_NATIVE_ENABLED.key -> "false", + CometConf.COMET_EXEC_ENABLED.key -> "true") { + cometSpark.sql(queryString).noop() + } + } benchmark.run() } } diff --git a/spark/src/test/scala/org/apache/spark/sql/benchmark/CometTPCQueryBenchmarkBase.scala b/spark/src/test/scala/org/apache/spark/sql/benchmark/CometTPCQueryBenchmarkBase.scala index 2361346b8..7e9bdbc9e 100644 --- a/spark/src/test/scala/org/apache/spark/sql/benchmark/CometTPCQueryBenchmarkBase.scala +++ b/spark/src/test/scala/org/apache/spark/sql/benchmark/CometTPCQueryBenchmarkBase.scala @@ -79,6 +79,17 @@ trait CometTPCQueryBenchmarkBase extends SqlBasedBenchmark with CometTPCQueryBas cometSpark.sql(queryString).noop() } } + benchmark.addCase(s"$name$nameSuffix: Comet (Exec)") { _ => + withSQLConf( + CometConf.COMET_ENABLED.key -> "true", + CometConf.COMET_EXEC_ENABLED.key -> "true", + CometConf.COMET_NATIVE_SCAN_ENABLED.key -> "false", + CometConf.COMET_CONVERT_FROM_PARQUET_ENABLED.key -> "true", + CometConf.COMET_EXEC_SHUFFLE_ENABLED.key -> "true", + CometConf.COMET_SHUFFLE_MODE.key -> "auto") { + cometSpark.sql(queryString).noop() + } + } benchmark.run() } }