add conv optimizations

Ryo-not-rio · Ryo-not-rio · commit ed04508b1a44 · 2025-12-01T18:30:17.000Z
diff --git a/ML-Frameworks/pytorch-aarch64/CHANGELOG.md b/ML-Frameworks/pytorch-aarch64/CHANGELOG.md
@@ -8,6 +8,7 @@ where `YY` is the year, and `MM` the month of the increment.
 ## [unreleased]
 
 ### Added
+ - Adds oneDNN [PR](https://github.com/uxlfoundation/oneDNN/pull/4377) optimize brgemm.
 
 ### Changed
  - Updates hashes for:
diff --git a/ML-Frameworks/pytorch-aarch64/get-source.sh b/ML-Frameworks/pytorch-aarch64/get-source.sh
@@ -97,6 +97,8 @@ git-shallow-clone https://github.com/pytorch/pytorch.git $PYTORCH_HASH
 
             # https://github.com/uxlfoundation/oneDNN/pull/4237 - cpu: aarch64: jit_reorder: cache blocking 4/8 inner blocks
             apply-github-patch uxlfoundation/oneDNN 8bdff1a2a6625432701363185a9bd34f7c22f241
+            # https://github.com/uxlfoundation/oneDNN/pull/4377 - cpu: aarch64: conv: optimize brgemm
+            apply-github-patch uxlfoundation/oneDNN 93b8cc29afc6ee9c2856436a5f5b10d5f1f2f2f1
         )
     )
     (

Original file line number	Diff line number	Diff line change
`@@ -97,6 +97,8 @@ git-shallow-clone https://github.com/pytorch/pytorch.git $PYTORCH_HASH`
`97`	`97`
`98`	`98`	`# https://github.com/uxlfoundation/oneDNN/pull/4237 - cpu: aarch64: jit_reorder: cache blocking 4/8 inner blocks`
`99`	`99`	`apply-github-patch uxlfoundation/oneDNN 8bdff1a2a6625432701363185a9bd34f7c22f241`
	`100`	`+ # https://github.com/uxlfoundation/oneDNN/pull/4377 - cpu: aarch64: conv: optimize brgemm`
	`101`	`+ apply-github-patch uxlfoundation/oneDNN 93b8cc29afc6ee9c2856436a5f5b10d5f1f2f2f1`
`100`	`102`	`)`
`101`	`103`	`)`
`102`	`104`	`(`