triton sdpa kernel with better perf

[Gasoonjia]

No description provided.

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 18 New Failures, 89 Cancelled Jobs

As of commit 7e1ad11 with merge base 1fe59c8 ():

NEW FAILURES - The following jobs have failed:

• docker-builds / docker-build (executorch-ubuntu-22.04-gcc11-aarch64, linux.arm64.2xlarge) (gh)
  Final attempt failed. Child_process exited with error code 1
• docker-builds / docker-build (linux.4xlarge, executorch-ubuntu-22.04-arm-sdk) (gh)
  Final attempt failed. Child_process exited with error code 1
• docker-builds / docker-build (linux.4xlarge, executorch-ubuntu-22.04-clang12-android) (gh)
  Final attempt failed. Child_process exited with error code 1
• docker-builds / docker-build (linux.4xlarge, executorch-ubuntu-22.04-clang12) (gh)
  Final attempt failed. Child_process exited with error code 1
• docker-builds / docker-build (linux.4xlarge, executorch-ubuntu-22.04-gcc9) (gh)
  Final attempt failed. Child_process exited with error code 1
• docker-builds / docker-build (linux.4xlarge, executorch-ubuntu-22.04-linter) (gh)
  Final attempt failed. Child_process exited with error code 1
• docker-builds / docker-build (linux.4xlarge, executorch-ubuntu-22.04-mediatek-sdk) (gh)
  Final attempt failed. Child_process exited with error code 1
• docker-builds / docker-build (linux.4xlarge, executorch-ubuntu-22.04-qnn-sdk) (gh)
  Final attempt failed. Child_process exited with error code 1
• docker-builds / docker-build (linux.4xlarge, executorch-ubuntu-22.04-zephyr-sdk) (gh)
  Final attempt failed. Child_process exited with error code 1
• pull / test-llama-runner-linux (bf16, custom, linux.2xlarge, executorch-ubuntu-22.04-clang12) / linux-job (gh)
  Final attempt failed. Child_process exited with error code 1
• pull / test-llama-runner-linux (fp32, xnnpack+custom+qe, linux.2xlarge, executorch-ubuntu-22.04-clang12) / linux-job (gh)
  Final attempt failed. Child_process exited with error code 1
• pull / test-llama-runner-linux (fp32, xnnpack+custom+qe, linux.arm64.2xlarge, executorch-ubuntu-22.04-gc... / linux-job (gh)
  Final attempt failed. Child_process exited with error code 1
• pull / test-llama-runner-linux (fp32, xnnpack+custom+quantize_kv, linux.2xlarge, executorch-ubuntu-22.04... / linux-job (gh)
  Final attempt failed. Child_process exited with error code 1
• pull / test-llama-runner-linux (fp32, xnnpack+custom+quantize_kv, linux.arm64.2xlarge, executorch-ubuntu... / linux-job (gh)
  Final attempt failed. Child_process exited with error code 1
• pull / test-llama-runner-linux (fp32, xnnpack+quantize_kv, linux.2xlarge, executorch-ubuntu-22.04-clang12) / linux-job (gh)
  Final attempt failed. Child_process exited with error code 1
• pull / test-llama-runner-linux (fp32, xnnpack+quantize_kv, linux.arm64.2xlarge, executorch-ubuntu-22.04-... / linux-job (gh)
  Final attempt failed. Child_process exited with error code 1
• pull / test-llama-runner-qnn-linux (fp32, qnn_16a16w, qnn) / linux-job (gh)
  Final attempt failed. Child_process exited with error code 1
• pull / test-llama-runner-qnn-linux (fp32, qnn_8a8w, qnn) / linux-job (gh)
  Final attempt failed. Child_process exited with error code 1

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

• Build documentation / build (buck2) / Build doc (gh)
  ##[error]The operation was canceled.
• Build Presets / linux (linux, linux.2xlarge, executorch-ubuntu-22.04-clang12) / build (gh)
  ##[error]The operation was canceled.
• Build Presets / linux (linux, linux.arm64.2xlarge, executorch-ubuntu-22.04-gcc11-aarch64) / build (gh)
• Build Presets / linux (llm, linux.2xlarge, executorch-ubuntu-22.04-clang12) / build (gh)
  ##[error]The operation was canceled.
• Build Presets / linux (llm, linux.arm64.2xlarge, executorch-ubuntu-22.04-gcc11-aarch64) / build (gh)
  ##[error]The operation was canceled.
• Build Presets / linux (pybind, linux.2xlarge, executorch-ubuntu-22.04-clang12) / build (gh)
  ##[error]The operation was canceled.
• Build Presets / linux (pybind, linux.arm64.2xlarge, executorch-ubuntu-22.04-gcc11-aarch64) / build (gh)
  ##[error]The operation was canceled.
• Build Presets / zephyr (zephyr) / build (gh)
  ##[error]The operation was canceled.
• Lint / android-java-format / linux-job (gh)
  ##[error]The operation was canceled.
• Lint / link-check / lint-file-size / linux-job (gh)
  ##[error]The operation was canceled.
• Lint / link-check / lint-urls / linux-job (gh)
  ##[error]The operation was canceled.
• Lint / link-check / lint-xrefs / linux-job (gh)
  ##[error]The operation was canceled.
• Lint / lintrunner / linux-job (gh)
  ##[error]The operation was canceled.
• Lint / lintrunner-mypy / linux-job (gh)
  ##[error]The operation was canceled.
• periodic / test-models-linux (buck2, mv3, portable, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• periodic / test-models-linux (buck2, mv3, xnnpack-quantization-delegation, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• periodic / test-models-linux (cmake, mv3, portable, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• periodic / test-models-linux (cmake, mv3, xnnpack-quantization-delegation, linux.2xlarge, 90) / linux-job (gh)
• periodic / test-models-linux (cmake, vit, portable, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• periodic / test-models-linux (cmake, vit, xnnpack-quantization-delegation, linux.2xlarge, 90) / linux-job (gh)
• pull / android / build-llm-demo / linux-job (gh)
  ##[error]The operation was canceled.
• pull / nxp-build-test / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-binary-size-linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-binary-size-linux-gcc / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-build-wasm-linux / linux-job (gh)
• pull / test-custom-ops-linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-eval_llama-wikitext-linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-llama_runner_eager-linux / linux-job (gh)
• pull / test-llama-runner-linux-android / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-lora-linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-mediatek-models-linux / linux-job (gh)
• pull / test-models-linux (add_mul, portable, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (add_mul, xnnpack-quantization-delegation, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (add, portable, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (add, xnnpack-quantization-delegation, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (emformer_join, portable, linux.4xlarge.memory) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (emformer_join, xnnpack-quantization-delegation, linux.4xlarge.memory) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (emformer_transcribe, portable, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (emformer_transcribe, xnnpack-quantization-delegation, linux.2xlarge) / linux-job (gh)
• pull / test-models-linux (ic3, portable, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (ic3, xnnpack-quantization-delegation, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (ic4, portable, linux.4xlarge.memory) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (ic4, xnnpack-quantization-delegation, linux.4xlarge.memory) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (linear, portable, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (linear, xnnpack-quantization-delegation, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (llama3_2_vision_encoder, portable, linux.4xlarge.memory) / linux-job (gh)
• pull / test-models-linux (mobilebert, portable, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (mobilebert, xnnpack-quantization-delegation, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (mv2, portable, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (mv2, xnnpack-quantization-delegation, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (phi_4_mini, portable, linux.4xlarge.memory) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (resnet18, portable, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (resnet18, xnnpack-quantization-delegation, linux.2xlarge) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux (resnet50, portable, linux.2xlarge) / linux-job (gh)
• pull / test-models-linux (resnet50, xnnpack-quantization-delegation, linux.2xlarge) / linux-job (gh)
• pull / test-models-linux (w2l, portable, linux.4xlarge.memory) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux-basic (mv3, portable, buck2, linux.2xlarge, executorch-ubuntu-22.04-clang12) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux-basic (mv3, portable, cmake, linux.2xlarge, executorch-ubuntu-22.04-clang12) / linux-job (gh)
• pull / test-models-linux-basic (mv3, portable, cmake, linux.arm64.2xlarge, executorch-ubuntu-22.04-gcc11... / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux-basic (mv3, xnnpack-quantization-delegation, buck2, linux.2xlarge, executorch-u... / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux-basic (mv3, xnnpack-quantization-delegation, cmake, linux.2xlarge, executorch-u... / linux-job (gh)
• pull / test-models-linux-basic (mv3, xnnpack-quantization-delegation, cmake, linux.arm64.2xlarge, execut... / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux-basic (vit, portable, buck2, linux.2xlarge, executorch-ubuntu-22.04-clang12) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux-basic (vit, portable, cmake, linux.2xlarge, executorch-ubuntu-22.04-clang12) / linux-job (gh)
• pull / test-models-linux-basic (vit, portable, cmake, linux.arm64.2xlarge, executorch-ubuntu-22.04-gcc11... / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux-basic (vit, xnnpack-quantization-delegation, buck2, linux.2xlarge, executorch-u... / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux-basic (vit, xnnpack-quantization-delegation, cmake, linux.2xlarge, executorch-u... / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-models-linux-basic (vit, xnnpack-quantization-delegation, cmake, linux.arm64.2xlarge, execut... / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-moshi-linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-multimodal-linux (gemma3-4b) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-openvino-linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-phi-3-mini-runner-linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-qnn-models-linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-quantized-aot-lib-linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-samsung-models-linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-selective-build-linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-setup-linux-gcc / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-static-llama-qnn-linux (stories_110m) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-static-llama-qnn-linux (stories_260k_bc) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-vulkan-models-linux / linux-job (gh)
• pull / test-vulkan-operators-linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / unittest / linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / unittest-arm-backend-with-no-fvp (test_pytest_models) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / unittest-arm-backend-with-no-fvp (test_pytest_ops) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / unittest-buck / linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / unittest-editable / linux / linux-job (gh)
  ##[error]The operation was canceled.
• pull / unittest-nxp-neutron / linux-job (gh)
  ##[error]The operation was canceled.
• pull / unittest-wasm-bindings (--enable-etdump) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / unittest-wasm-bindings / linux-job (gh)
  ##[error]The operation was canceled.

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

[github-actions]

This PR needs a release notes: label

If your change should be included in the release notes (i.e. would users of this library care about this change?), please use a label starting with release notes:. This helps us keep track and include your important work in the next release notes.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "release notes: none"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.