Align meta deducing for fft_r2c with fft_r2c_mkl on XPU

[CuiYifeng]

There is a memory layout mismatching between fft_r2c XPU and Inductor meta deducing.
Original fft_r2c Inductor meta deducing for XPU backend is aligned with CPU (fallback). This PR is to correct the Inductor meta deducing and update the torch-xpu-ops commit to intel/torch-xpu-ops@3a9419c.
The XPU implementation first performs the R2C transform on the last dimension, followed by iterative C2C transforms on the remaining dimensions.

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/156048

• 📄 Preview Python docs built from this PR
• 📄 Preview C++ docs built from this PR
• ❓ Need help or want to give feedback on the CI? Visit the bot commands wiki or our office hours

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

✅ You can merge normally! (1 Unrelated Failure)

As of commit 8875a94 with merge base 96df866 ():

FLAKY - The following job failed but was likely due to flakiness present on trunk:

• pull / linux-jammy-py3.9-clang12 / test (dynamo_wrapped, 3, 3, linux.2xlarge) (gh) (disabled by #156438)
  test_jit_fuser_te.py::TestTEFuserDynamic::test_inlined_optimized_graph

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

[CuiYifeng]

@etaf Please help review, thanks.

[etaf]

Let's put the torch-xpu-ops commit pin update which supports fft in this PR.

[CuiYifeng]

@pytorchbot rebase

[pytorchmergebot]

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

[pytorchmergebot]

Successfully rebased yifeng/fft_r2c_meta_cal onto refs/remotes/origin/viable/strict, please pull locally before adding more changes (for example, via git checkout yifeng/fft_r2c_meta_cal && git pull --rebase)

[CuiYifeng]

Let's put the torch-xpu-ops commit pin update which supports fft in this PR.

@etaf The commit pin has been updated.

[etaf]

@pytorchbot rebase -b main

[pytorchmergebot]

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

[pytorchmergebot]

Successfully rebased yifeng/fft_r2c_meta_cal onto refs/remotes/origin/main, please pull locally before adding more changes (for example, via git checkout yifeng/fft_r2c_meta_cal && git pull --rebase)

[pytorch-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]

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]

To add the ciflow label ciflow/trunk 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]

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.

[guangyey]

@pytorchbot rebase

[pytorchmergebot]

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

[pytorchmergebot]

Tried to rebase and push PR #156048, but it was already up to date. Try rebasing against main by issuing:
@pytorchbot rebase -b main

[CuiYifeng]

The failed case should be irrelevant to this PR.

[guangyey]

"Only XPU code change, the failed case should be irrelevant to this PR."
@pytorchbot merge -i

[pytorchmergebot]

Merge started

Your change will be merged while ignoring the following 1 checks: pull / linux-jammy-py3.9-clang12 / test (dynamo_wrapped, 3, 3, linux.2xlarge)

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging

Check the merge workflow status
here