[XNNPACK][Partitioner] SDPA Config

[mcr229]

Stack from ghstack (oldest at bottom):

• [XNNPACK][Partitioner] Migrate completely to new config based partitioner #4765
• -> [XNNPACK][Partitioner] SDPA Config #4764
• [ExecuTorch][XNNPACK] Don't partition 3d and transposed convs #4763
• [XNNPACK][Partitioner] enable src based partitioner #4762

We add the SDPA Config here for partitioner.

Currently there is an issue with SDPA when used from the FairSeq Multihead attention models, so I currently have it disabled for the base partitioner until we resolve that. Otherwise, for our tests, we can use the SDPA correctly from there. We have to track D60553559. Will follow up on this later.

Differential Revision: D60323285

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 23 Cancelled Jobs

As of commit 662f285 with merge base 9e478e8 ():

CANCELLED JOBS - The following jobs were cancelled. Please retry:

• Lint / lintrunner / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-binary-size-linux-gcc / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-custom-ops-linux (buck2) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-export-llava-linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-llama-runner-linux (fp32, buck2, portable) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-llama-runner-linux (fp32, buck2, xnnpack+custom) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-llama-runner-linux (fp32, buck2, xnnpack+custom+qe) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-llama-runner-linux (fp32, cmake, portable) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-llama-runner-linux (fp32, cmake, xnnpack+custom) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-llama-runner-linux (fp32, cmake, xnnpack+custom+qe) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-llama-runner-linux-android (cmake) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-llama-runner-qnn-linux (fp32, cmake, qnn) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (buck2, mv3, portable, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (buck2, mv3, xnnpack-quantization-delegation, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (cmake, mv3, xnnpack-quantization-delegation, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (cmake, vit, portable, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (cmake, vit, xnnpack-delegation, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-quantized-aot-lib-linux (cmake) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-selective-build-linux (cmake) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-setup-linux-gcc (cmake) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / unittest / linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / unittest / macos / macos-job (gh)
  ##[error]The operation was canceled.
• pull / unittest-arm (buck2) / linux-job (gh)
  ##[error]The operation was canceled.

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

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D60323285

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D60323285

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D60323285

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D60323285