Skip to content

Fix floating point argmax#993

Merged
inducer merged 1 commit intoinducer:mainfrom
djanekovic:bugfix/float_argmax
Mar 26, 2026
Merged

Fix floating point argmax#993
inducer merged 1 commit intoinducer:mainfrom
djanekovic:bugfix/float_argmax

Conversation

@djanekovic
Copy link
Copy Markdown
Contributor

Hey there!

This PR fixes arg(max|min) for floating point dtypes. It was failing because INFINITY and HUGE_VAL weren't mangled. Added two tests that fail without this change. A better test would maybe also execute the kernel, let me know what you think and I can change it.

Kind regards,
Darko

@inducer inducer force-pushed the bugfix/float_argmax branch from 0edaa8e to 32bd8f9 Compare March 26, 2026 15:57
@inducer inducer enabled auto-merge (rebase) March 26, 2026 15:58
@inducer
Copy link
Copy Markdown
Owner

inducer commented Mar 26, 2026

Thanks!

@inducer inducer force-pushed the bugfix/float_argmax branch from 32bd8f9 to e4bbaaa Compare March 26, 2026 15:59
@inducer inducer merged commit fcd25b4 into inducer:main Mar 26, 2026
17 checks passed
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.

2 participants