Skip to content

Commit 28ea61f

Browse files
committed
[aarch64] Implement QGEMM S8S8 kernel with SMMLA instructions
1 parent ad9fe3e commit 28ea61f

5 files changed

Lines changed: 1919 additions & 0 deletions

File tree

cmake/onnxruntime_mlas.cmake

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ function(setup_mlas_source_for_windows)
6262
${MLAS_SRC_DIR}/qgemm_kernel_udot.cpp
6363
${MLAS_SRC_DIR}/qgemm_kernel_sdot.cpp
6464
${MLAS_SRC_DIR}/qgemm_kernel_ummla.cpp
65+
${MLAS_SRC_DIR}/qgemm_kernel_smmla.cpp
6566
)
6667

6768
set(mlas_platform_preprocess_srcs
@@ -311,6 +312,7 @@ else()
311312
${MLAS_SRC_DIR}/aarch64/QgemmU8X8KernelUdot.S
312313
${MLAS_SRC_DIR}/aarch64/QgemmU8X8KernelUmmla.S
313314
${MLAS_SRC_DIR}/aarch64/QgemmS8S8KernelSdot.S
315+
${MLAS_SRC_DIR}/aarch64/QgemmS8S8KernelSmmla.S
314316
${MLAS_SRC_DIR}/aarch64/SgemmKernelNeon.S
315317
${MLAS_SRC_DIR}/aarch64/SgemvKernelNeon.S
316318
${MLAS_SRC_DIR}/aarch64/SymQgemmS8KernelNeon.S
@@ -320,6 +322,7 @@ else()
320322
${MLAS_SRC_DIR}/qgemm_kernel_udot.cpp
321323
${MLAS_SRC_DIR}/qgemm_kernel_sdot.cpp
322324
${MLAS_SRC_DIR}/qgemm_kernel_ummla.cpp
325+
${MLAS_SRC_DIR}/qgemm_kernel_smmla.cpp
323326
)
324327
if (NOT APPLE)
325328
set(mlas_platform_srcs

0 commit comments

Comments
 (0)