[WIP][FlexAttention] Enable tensor descriptor for FlexAttention backward

[liangan1]

Fixes #163543

cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @ipiszy @chenyang78 @kadeng @muchulee8 @amjames @chauhang @aakhundov @coconutruben

[pytorch-bot[bot]]

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/166927

• :page_facing_up: Preview Python docs built from this PR
• :page_facing_up: Preview C++ docs built from this PR
• :question: Need help or want to give feedback on the CI? Visit the bot commands wiki

Note: Links to docs will display an error until the docs builds have been completed.

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[liangan1]

@pytorchbot rebase

[pytorchmergebot]

@pytorchbot started a rebase job onto refs/remotes/origin/viable/strict. Check the current status here

[pytorchmergebot]

Rebase failed due to Command git -C /home/runner/work/pytorch/pytorch rebase refs/remotes/origin/viable/strict pull/166927/head returned non-zero exit code 1

Rebasing (1/12)
Rebasing (2/12)
Rebasing (3/12)
Rebasing (4/12)
Rebasing (5/12)
Rebasing (6/12)
Rebasing (7/12)
Rebasing (8/12)
Rebasing (9/12)
Rebasing (10/12)
Auto-merging torch/_inductor/template_heuristics/triton.py
CONFLICT (content): Merge conflict in torch/_inductor/template_heuristics/triton.py
error: could not apply 46176382eac... Fix test_num_warps_8_error_xpu fail
hint: Resolve all conflicts manually, mark them as resolved with
hint: "git add/rm <conflicted_files>", then run "git rebase --continue".
hint: You can instead skip this commit: run "git rebase --skip".
hint: To abort and get back to the state before "git rebase", run "git rebase --abort".
hint: Disable this message with "git config set advice.mergeConflict false"
Could not apply 46176382eac... # Fix test_num_warps_8_error_xpu fail

Raised by https://github.com/pytorch/pytorch/actions/runs/19091996868

[liangan1]

@pytorchbot rebase

[pytorchmergebot]

@pytorchbot started a rebase job onto refs/remotes/origin/viable/strict. Check the current status here

[pytorchmergebot]

Rebase failed due to Command git -C /home/runner/work/pytorch/pytorch rebase refs/remotes/origin/viable/strict pull/166927/head returned non-zero exit code 1

Rebasing (1/47)
Rebasing (2/47)
Rebasing (3/47)
Rebasing (4/47)
Rebasing (5/47)
Rebasing (6/47)
Rebasing (7/47)
Rebasing (8/47)
Rebasing (9/47)
Rebasing (10/47)
Auto-merging torch/_inductor/template_heuristics/triton.py
CONFLICT (content): Merge conflict in torch/_inductor/template_heuristics/triton.py
error: could not apply 46176382eac... Fix test_num_warps_8_error_xpu fail
hint: Resolve all conflicts manually, mark them as resolved with
hint: "git add/rm <conflicted_files>", then run "git rebase --continue".
hint: You can instead skip this commit: run "git rebase --skip".
hint: To abort and get back to the state before "git rebase", run "git rebase --abort".
hint: Disable this message with "git config set advice.mergeConflict false"
Could not apply 46176382eac... # Fix test_num_warps_8_error_xpu fail

Raised by https://github.com/pytorch/pytorch/actions/runs/19092123369

[pytorch-bot[bot]]

To add the ciflow label ciflow/xpu please first approve the workflows that are awaiting approval (scroll to the bottom of this page).

This helps ensure we don't trigger CI on this PR until it is actually authorized to do so. Please ping one of the reviewers if you do not have access to approve and run workflows.

[pytorch-bot[bot]]

To add the ciflow label ciflow/xpu please first approve the workflows that are awaiting approval (scroll to the bottom of this page).

This helps ensure we don't trigger CI on this PR until it is actually authorized to do so. Please ping one of the reviewers if you do not have access to approve and run workflows.

[hoshibara]

@pytorchbot rebase

[pytorchmergebot]

@pytorchbot started a rebase job onto refs/remotes/origin/viable/strict. Check the current status here

[pytorchmergebot]

Rebase failed due to Command git -C /home/runner/work/pytorch/pytorch rebase refs/remotes/origin/viable/strict pull/166927/head returned non-zero exit code 1

Rebasing (1/19)
Rebasing (2/19)
Rebasing (3/19)
Rebasing (4/19)
Rebasing (5/19)
Rebasing (6/19)
Rebasing (7/19)
Rebasing (8/19)
Rebasing (9/19)
Rebasing (10/19)
Auto-merging torch/_inductor/template_heuristics/triton.py
CONFLICT (content): Merge conflict in torch/_inductor/template_heuristics/triton.py
error: could not apply 46176382eac... Fix test_num_warps_8_error_xpu fail
hint: Resolve all conflicts manually, mark them as resolved with
hint: "git add/rm <conflicted_files>", then run "git rebase --continue".
hint: You can instead skip this commit: run "git rebase --skip".
hint: To abort and get back to the state before "git rebase", run "git rebase --abort".
hint: Disable this message with "git config set advice.mergeConflict false"
Could not apply 46176382eac... # Fix test_num_warps_8_error_xpu fail

Raised by https://github.com/pytorch/pytorch/actions/runs/19352471300