Skip to content

DecomposeAtenAminAmaxOp negative axis fix#4556

Open
voonter wants to merge 1 commit into
llvm:mainfrom
voonter:fix-decompatenaminamaxop-negative-axis
Open

DecomposeAtenAminAmaxOp negative axis fix#4556
voonter wants to merge 1 commit into
llvm:mainfrom
voonter:fix-decompatenaminamaxop-negative-axis

Conversation

@voonter

@voonter voonter commented May 4, 2026

Copy link
Copy Markdown

Fixes #4553

Fixed support for negative axis in DecomposeAtenAminAmaxOp

@zjgarvey zjgarvey self-requested a review May 14, 2026 08:48
@voonter voonter force-pushed the fix-decompatenaminamaxop-negative-axis branch from 37a5bb3 to f4679a0 Compare May 14, 2026 20:14

@zjgarvey zjgarvey left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can you explain what the issue was a bit more? E.g., is it that the sorting needs to happen after making the dims positive? If so, LGTM

@voonter

voonter commented May 16, 2026

Copy link
Copy Markdown
Author

Can you explain what the issue was a bit more? E.g., is it that the sorting needs to happen after making the dims positive? If so, LGTM

Yes. Reducing from the last axis only works when the axes are sorted as positive values. Negative axes break that ordering. And with keepDim=false it results in segfault.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

DecomposeAtenAminAmaxOp does not support mutliple axes with "-1" in a set.

2 participants