-
Notifications
You must be signed in to change notification settings - Fork 232
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Update HashAggregateSuite to work with AQE #469
Conversation
Signed-off-by: Niranjan Artal <nartal@nvidia.com>
build |
Signed-off-by: Niranjan Artal <nartal@nvidia.com>
|
||
gpuPlan match { | ||
case WholeStageCodegenExec(GpuColumnarToRowExec(plan, _)) => | ||
assert(plan.children.head.isInstanceOf[GpuHashAggregateExec]) | ||
assert(gpuPlan.find(_.isInstanceOf[SortAggregateExec]).isEmpty) | ||
assert(gpuPlan.children.forall(exec => exec.isInstanceOf[GpuExec])) | ||
|
||
case a: AdaptiveSparkPlanExec => |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think it would be good to revisit creating helper utilities for inspecting plans, that can work both with adaptive and non-adaptive plans so we don't end up with lots of duplicate code in tests. I don't think it has to be part of this PR though.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @andygrove . I have created a follow-on issue for creating helper utilities #476
build |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
* Update HashAggregateSuite to work with AQE Signed-off-by: Niranjan Artal <nartal@nvidia.com> * addressed review comments Signed-off-by: Niranjan Artal <nartal@nvidia.com>
* Update HashAggregateSuite to work with AQE Signed-off-by: Niranjan Artal <nartal@nvidia.com> * addressed review comments Signed-off-by: Niranjan Artal <nartal@nvidia.com>
Signed-off-by: spark-rapids automation <70000568+nvauto@users.noreply.github.com>
Signed-off-by: Niranjan Artal nartal@nvidia.com
Updated tests and verified locally. This closes #453.