Add Fadi's PR to accelerate SDPAs.

Author: puneetmatharu

ML-Frameworks/pytorch-aarch64/CHANGELOG.md

@@ -14,6 +14,7 @@ where `YY` is the year, and `MM` the month of the increment.
 ## [unreleased]
 
 ### Added
+ - Adds PyTorch [PR #177009](https://github.com/pytorch/pytorch/pull/177009), to accelerate SDPA on Arm CPUs.
 
 ### Changed
  - Updates hashes for:
@@ -22,6 +23,7 @@ where `YY` is the year, and `MM` the month of the increment.
    - `ONEDNN_HASH` to `a45a5ba044d1830b88d7fb529efaa52133328d45`, from main, Mar 23rd.
    - `TORCH_AO_HASH` to `fbce9178697ca22d604314b316feb765360e0dec`, from main, Mar 23rd.
    - `KLEIDIAI_HASH` to `72a6a50c1dba714ff27a908e8dd54be3628794b0`, v1.23.0 from main, Mar 20th.
+ - Updates `OPENBLAS_VERSION` from v0.3.30 to faa06bd75900ae8251579d5178152e644273703b from main, Mar 21st.
  - Updates `torchvision` from 0.25.0.dev20260130 to 0.26.0.dev20260322.
 
 ### Removed

ML-Frameworks/pytorch-aarch64/get-source.sh

@@ -38,6 +38,10 @@ git-shallow-clone https://github.com/pytorch/pytorch.git $PYTORCH_HASH
     apply-github-patch pytorch/pytorch b1782bbe0eda5957870e2f6e95b8f167e04843cb
     apply-github-patch pytorch/pytorch 337925aed2babb3ef7808f78536bbbc9df346a4f
 
+    # https://github.com/pytorch/pytorch/pull/177009 - Accelerate SDPA on Arm CPUs: Unroll exp_sum and max_mul kernels
+    apply-github-patch pytorch/pytorch f6a6cc94e92636902acbbe443d00cec4a8efa12d
+    apply-github-patch pytorch/pytorch 253a1ea6db210c01178dd365922eb49fe4892fe7
+
     # Remove deps that we don't need for manylinux AArch64 CPU builds before fetching.
     # Only used when jni.h is present (see .ci/pytorch/build.sh:116), which is not the case for manylinux
     git rm android/libs/fbjni

ML-Frameworks/pytorch-aarch64/versions.sh

@@ -17,7 +17,7 @@ KLEIDIAI_HASH=72a6a50c1dba714ff27a908e8dd54be3628794b0  # v1.23.0 from main, Mar
 
 # build-wheel.sh deps
 ACL_VERSION="v52.8.0"   # Jan 23rd
-OPENBLAS_VERSION="v0.3.30"  # Jan 15th
+OPENBLAS_VERSION=faa06bd75900ae8251579d5178152e644273703b  # Mar 21st
 
 # Dockerfile deps
 TORCHVISION_NIGHTLY="0.26.0.dev20260322"