From 0a49676fb0e54c9229a39f6304bc88b7d24e0355 Mon Sep 17 00:00:00 2001 From: Nikhil Gupta Date: Fri, 6 Mar 2026 03:48:59 +0000 Subject: [PATCH] cpu: aarch64: Upgrade OneDNN for aarch64 to add support for int8 matmul (#36147) Signed-off-by: Nikhil Gupta --- cmake/cpu_extension.cmake | 25 ++++++++++++++++++------- 1 file changed, 18 insertions(+), 7 deletions(-) diff --git a/cmake/cpu_extension.cmake b/cmake/cpu_extension.cmake index dde8cc207..f085fe24e 100644 --- a/cmake/cpu_extension.cmake +++ b/cmake/cpu_extension.cmake @@ -242,13 +242,24 @@ if (ENABLE_X86_ISA OR (ASIMD_FOUND AND NOT APPLE_SILICON_FOUND) OR POWER9_FOUND ) else() message(STATUS "Downloading oneDNN from GitHub") - FetchContent_Declare( - oneDNN - GIT_REPOSITORY https://github.com/oneapi-src/oneDNN.git - GIT_TAG v3.10 - GIT_PROGRESS TRUE - GIT_SHALLOW TRUE - ) + if(ASIMD_FOUND AND NOT APPLE_SILICON_FOUND) + message(STATUS "aarch64 detected: using pinned oneDNN commit 9c5be1cc59e368aebf0909e6cf20f981ea61462a") + FetchContent_Declare( + oneDNN + GIT_REPOSITORY https://github.com/oneapi-src/oneDNN.git + GIT_TAG 9c5be1cc59e368aebf0909e6cf20f981ea61462a + GIT_PROGRESS TRUE + GIT_SHALLOW FALSE + ) + else() + FetchContent_Declare( + oneDNN + GIT_REPOSITORY https://github.com/oneapi-src/oneDNN.git + GIT_TAG v3.10 + GIT_PROGRESS TRUE + GIT_SHALLOW TRUE + ) + endif() endif() set(ONEDNN_LIBRARY_TYPE "STATIC")