AQE applies to all queries that are:

• Non-streaming

• Contain at least one exchange (usually when there’s a join, aggregate, or window), one sub-query, or both.

Not all AQE-applied queries are necessarily re-optimized. The re-optimization might or might not come up with a different query plan than the one statically compiled. To determine whether a query’s plan has been changed by AQE

But you can use spark.sql.shuffle.partitions in auto or using manual settings