Skip to content

Commit e538db9

Browse files
authored
minor: Update TPC-DS microbenchmarks to remove "scan only" and "exec only" runs (#2396)
* save * Minor cleanup of TPC-DS microbenchmarks * revert * prep for review * prep for review * prep for review
1 parent 34daa54 commit e538db9

File tree

2 files changed

+6
-45
lines changed

2 files changed

+6
-45
lines changed

spark/src/test/scala/org/apache/spark/sql/benchmark/CometTPCDSMicroBenchmark.scala

Lines changed: 4 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ object CometTPCDSMicroBenchmark extends CometTPCQueryBenchmarkBase {
8484
benchmarkName: String,
8585
nameSuffix: String = ""): Unit = {
8686
queries.foreach { name =>
87-
val source = Source.fromFile(s"spark/src/test/resources/tpcds-micro-benchmarks/$name.sql")
87+
val source = Source.fromFile(s"src/test/resources/tpcds-micro-benchmarks/$name.sql")
8888
val queryString = source
8989
.getLines()
9090
.filterNot(_.startsWith("--"))
@@ -108,34 +108,13 @@ object CometTPCDSMicroBenchmark extends CometTPCQueryBenchmarkBase {
108108
benchmark.addCase(s"$name$nameSuffix") { _ =>
109109
cometSpark.sql(queryString).noop()
110110
}
111-
benchmark.addCase(s"$name$nameSuffix: Comet (Scan)") { _ =>
112-
withSQLConf(CometConf.COMET_ENABLED.key -> "true") {
113-
cometSpark.sql(queryString).noop()
114-
}
115-
}
116-
benchmark.addCase(s"$name$nameSuffix: Comet (Scan, Exec)") { _ =>
117-
withSQLConf(
118-
CometConf.COMET_ENABLED.key -> "true",
119-
CometConf.COMET_EXEC_SHUFFLE_ENABLED.key -> "true",
120-
CometConf.COMET_SHUFFLE_MODE.key -> "auto",
121-
CometConf.COMET_REGEXP_ALLOW_INCOMPATIBLE.key -> "true",
122-
// enabling COMET_EXPLAIN_NATIVE_ENABLED may add overhead but is useful for debugging
123-
CometConf.COMET_EXPLAIN_NATIVE_ENABLED.key -> "false",
124-
CometConf.COMET_EXEC_ENABLED.key -> "true") {
125-
cometSpark.sql(queryString).noop()
126-
}
127-
}
128-
benchmark.addCase(s"$name$nameSuffix: Comet (Exec)") { _ =>
111+
benchmark.addCase(s"$name$nameSuffix: Comet") { _ =>
129112
withSQLConf(
130113
CometConf.COMET_ENABLED.key -> "true",
131-
CometConf.COMET_NATIVE_SCAN_ENABLED.key -> "false",
132-
CometConf.COMET_CONVERT_FROM_PARQUET_ENABLED.key -> "true",
133-
CometConf.COMET_EXEC_SHUFFLE_ENABLED.key -> "true",
134-
CometConf.COMET_SHUFFLE_MODE.key -> "auto",
114+
CometConf.COMET_EXEC_ENABLED.key -> "true",
135115
CometConf.COMET_REGEXP_ALLOW_INCOMPATIBLE.key -> "true",
136116
// enabling COMET_EXPLAIN_NATIVE_ENABLED may add overhead but is useful for debugging
137-
CometConf.COMET_EXPLAIN_NATIVE_ENABLED.key -> "false",
138-
CometConf.COMET_EXEC_ENABLED.key -> "true") {
117+
CometConf.COMET_EXPLAIN_NATIVE_ENABLED.key -> "false") {
139118
cometSpark.sql(queryString).noop()
140119
}
141120
}

spark/src/test/scala/org/apache/spark/sql/benchmark/CometTPCQueryBenchmarkBase.scala

Lines changed: 2 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -65,28 +65,10 @@ trait CometTPCQueryBenchmarkBase extends SqlBasedBenchmark with CometTPCQueryBas
6565
benchmark.addCase(s"$name$nameSuffix") { _ =>
6666
cometSpark.sql(queryString).noop()
6767
}
68-
benchmark.addCase(s"$name$nameSuffix: Comet (Scan)") { _ =>
69-
withSQLConf(CometConf.COMET_ENABLED.key -> "true") {
70-
cometSpark.sql(queryString).noop()
71-
}
72-
}
73-
benchmark.addCase(s"$name$nameSuffix: Comet (Scan, Exec)") { _ =>
74-
withSQLConf(
75-
CometConf.COMET_ENABLED.key -> "true",
76-
CometConf.COMET_EXEC_ENABLED.key -> "true",
77-
CometConf.COMET_EXEC_SHUFFLE_ENABLED.key -> "true",
78-
CometConf.COMET_SHUFFLE_MODE.key -> "auto") {
79-
cometSpark.sql(queryString).noop()
80-
}
81-
}
82-
benchmark.addCase(s"$name$nameSuffix: Comet (Exec)") { _ =>
68+
benchmark.addCase(s"$name$nameSuffix: Comet") { _ =>
8369
withSQLConf(
8470
CometConf.COMET_ENABLED.key -> "true",
85-
CometConf.COMET_EXEC_ENABLED.key -> "true",
86-
CometConf.COMET_NATIVE_SCAN_ENABLED.key -> "false",
87-
CometConf.COMET_CONVERT_FROM_PARQUET_ENABLED.key -> "true",
88-
CometConf.COMET_EXEC_SHUFFLE_ENABLED.key -> "true",
89-
CometConf.COMET_SHUFFLE_MODE.key -> "auto") {
71+
CometConf.COMET_EXEC_ENABLED.key -> "true") {
9072
cometSpark.sql(queryString).noop()
9173
}
9274
}

0 commit comments

Comments
 (0)