biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

cpu: aarch64: Upgrade OneDNN for aarch64 to add support for int8 matmul (#36147)

Browse Source 
Signed-off-by: Nikhil Gupta <nikhil.gupta2@arm.com>
This commit is contained in:
Nikhil Gupta
2026-03-06 03:48:59 +00:00
committed by GitHub
parent c012a8c477
commit 0a49676fb0
1 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
cmake/cpu_extension.cmake
View File

@@ -242,13 +242,24 @@ if (ENABLE_X86_ISA OR (ASIMD_FOUND AND NOT APPLE_SILICON_FOUND) OR POWER9_FOUND
) )
else() else()
message(STATUS "Downloading oneDNN from GitHub") message(STATUS "Downloading oneDNN from GitHub")
FetchContent_Declare( if(ASIMD_FOUND AND NOT APPLE_SILICON_FOUND)
oneDNN message(STATUS "aarch64 detected: using pinned oneDNN commit 9c5be1cc59e368aebf0909e6cf20f981ea61462a")
GIT_REPOSITORY https://github.com/oneapi-src/oneDNN.git FetchContent_Declare(
GIT_TAG v3.10 oneDNN
GIT_PROGRESS TRUE GIT_REPOSITORY https://github.com/oneapi-src/oneDNN.git
GIT_SHALLOW TRUE 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() endif()
set(ONEDNN_LIBRARY_TYPE "STATIC") set(ONEDNN_LIBRARY_TYPE "STATIC")