Skip to content

[SWDEV-538312] Fix num_stages=8 configs for flex attention #2277

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.

Sign up for GitHub

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

Jump to bottom
Merged
merged 1 commit into from
Jun 16, 2025
Merged

[SWDEV-538312] Fix num_stages=8 configs for flex attention #2277

merged 1 commit into from
Jun 16, 2025

Conversation

@jataylo
Copy link

@jataylo jataylo commented Jun 16, 2025

num_stages==8 configs are always skipped causing breakages

Example error:

torch._inductor.exc.LoweringException: NoValidChoicesError: No choices to select, please consider adding ATEN into max_autotune_gemm_backends config (defined in torch/_inductor/config.py) to allow at least one choice. 
  target: flex_attention_backward

@jataylo
Flex_attention logic error resolved
f9d62b3 
num_stages==8 configs are always skipped causing breakages
@jataylo jataylo marked this pull request as ready for review June 16, 2025 14:15
@rocm-repo-management-api
Copy link

rocm-repo-management-api bot commented Jun 16, 2025

Jenkins build for f9d62b31b845c16ecb71760e2d19013885b5b6d0 commit is in progress
Links: Blue Ocean view / Build artifacts

@jithunnair-amd jithunnair-amd merged commit d1c90a0 into release/2.6 Jun 16, 2025
0 of 2 checks passed
@jithunnair-amd jithunnair-amd deleted the SWDEV-538312-fix branch June 16, 2025 15:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

@jithunnair-amd jithunnair-amd

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

3 participants

@jataylo @jithunnair-amd