minni31 commented on code in PR #12242:
URL: https://github.com/apache/gluten/pull/12242#discussion_r3534146120
##########
gluten-ut/spark35/src/test/scala/org/apache/spark/sql/GlutenBitmapExpressionsQuerySuite.scala:
##########
@@ -37,4 +37,19 @@ class GlutenBitmapExpressionsQuerySuite
"Expected native HashAggregateExecBaseTransformer in plan"
)
}
+
+ test("bitmap_or_agg routes to native") {
+ val df = spark.sql(
+ "SELECT bitmap_or_agg(bm) FROM (" +
+ "SELECT bitmap_construct_agg(bitmap_bit_position(col)) AS bm " +
+ "FROM values (1L), (2L), (3L) AS t(col)" +
+ ") sub")
+ df.collect()
+ assert(
+ collectWithSubqueries(df.queryExecution.executedPlan) {
+ case h: HashAggregateExecBaseTransformer => h
+ }.nonEmpty,
+ "Expected native HashAggregateExecBaseTransformer in plan"
+ )
+ }
Review Comment:
Agreed — updated the assertion to pattern-match only on
HashAggregateExecBaseTransformer nodes whose aggregateExpressions contain a
BitmapOrAgg instance. This ensures the test specifically validates that
bitmap_or_agg (not just bitmap_construct_agg) is routed to native.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
