Skip to content

Lower isneginf().#8912

Merged
ysiraichi merged 3 commits intomasterfrom
ysiraichi/lower-isneginf
Apr 1, 2025
Merged

Lower isneginf().#8912
ysiraichi merged 3 commits intomasterfrom
ysiraichi/lower-isneginf

Conversation

@ysiraichi
Copy link
Copy Markdown
Collaborator

Fix: #8746

This PR adds a lowering for isneginf() operation. This change should address the one introduced in pytorch/pytorch#139763, where this call was introduced to scaled_dot_product_attention. This should avoid a fallback operation, which also might improve performance.

@ysiraichi ysiraichi merged commit cba9ff9 into master Apr 1, 2025
23 checks passed
jeffhataws pushed a commit that referenced this pull request Apr 2, 2025
ysiraichi added a commit that referenced this pull request Apr 2, 2025
ysiraichi added a commit that referenced this pull request Apr 2, 2025
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.

a HUGE performance degradation in SDPA (scale_dot_product_attention) in xla 2.6.0 on TPU

2 participants