Skip to content

Commit d88d52e

Browse files
authored
[aarch64] Remove mmla kernel support from apple (#18082)
### Description <!-- Describe your changes. --> The mmla kernels require additional ISA flags and are currently supported only on Linux ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. --> more context is in #15270 cc: @skottmckay , @chenfucn , @snnn
1 parent 706e13e commit d88d52e

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

cmake/onnxruntime_mlas.cmake

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -325,9 +325,7 @@ else()
325325
${MLAS_SRC_DIR}/aarch64/QgemmU8X8KernelNeon.S
326326
${MLAS_SRC_DIR}/aarch64/QgemmS8S8KernelNeon.S
327327
${MLAS_SRC_DIR}/aarch64/QgemmU8X8KernelUdot.S
328-
${MLAS_SRC_DIR}/aarch64/QgemmU8X8KernelUmmla.S
329328
${MLAS_SRC_DIR}/aarch64/QgemmS8S8KernelSdot.S
330-
${MLAS_SRC_DIR}/aarch64/QgemmS8S8KernelSmmla.S
331329
${MLAS_SRC_DIR}/aarch64/SgemmKernelNeon.S
332330
${MLAS_SRC_DIR}/aarch64/SgemvKernelNeon.S
333331
${MLAS_SRC_DIR}/aarch64/SymQgemmS8KernelNeon.S
@@ -336,24 +334,26 @@ else()
336334
${MLAS_SRC_DIR}/qgemm_kernel_neon.cpp
337335
${MLAS_SRC_DIR}/qgemm_kernel_udot.cpp
338336
${MLAS_SRC_DIR}/qgemm_kernel_sdot.cpp
339-
${MLAS_SRC_DIR}/qgemm_kernel_ummla.cpp
340-
${MLAS_SRC_DIR}/qgemm_kernel_smmla.cpp
341337
)
342338
if (NOT APPLE)
343339
set(mlas_platform_srcs
344340
${mlas_platform_srcs}
345341
${MLAS_SRC_DIR}/aarch64/HalfGemmKernelNeon.S
342+
${MLAS_SRC_DIR}/aarch64/QgemmS8S8KernelSmmla.S
343+
${MLAS_SRC_DIR}/aarch64/QgemmU8X8KernelUmmla.S
346344
${MLAS_SRC_DIR}/activate_fp16.cpp
347345
${MLAS_SRC_DIR}/dwconv.cpp
348346
${MLAS_SRC_DIR}/halfgemm_kernel_neon.cpp
349347
${MLAS_SRC_DIR}/pooling_fp16.cpp
348+
${MLAS_SRC_DIR}/qgemm_kernel_smmla.cpp
349+
${MLAS_SRC_DIR}/qgemm_kernel_ummla.cpp
350350
)
351351
set_source_files_properties(${MLAS_SRC_DIR}/aarch64/HalfGemmKernelNeon.S PROPERTIES COMPILE_FLAGS " -march=armv8.2-a+fp16 ")
352+
set_source_files_properties(${MLAS_SRC_DIR}/aarch64/QgemmS8S8KernelSmmla.S PROPERTIES COMPILE_FLAGS " -march=armv8.2-a+i8mm ")
353+
set_source_files_properties(${MLAS_SRC_DIR}/aarch64/QgemmU8X8KernelUmmla.S PROPERTIES COMPILE_FLAGS " -march=armv8.2-a+i8mm ")
352354
set_source_files_properties(${MLAS_SRC_DIR}/activate_fp16.cpp PROPERTIES COMPILE_FLAGS " -march=armv8.2-a+fp16 ")
353355
set_source_files_properties(${MLAS_SRC_DIR}/dwconv.cpp PROPERTIES COMPILE_FLAGS " -march=armv8.2-a+fp16 ")
354356
set_source_files_properties(${MLAS_SRC_DIR}/pooling_fp16.cpp PROPERTIES COMPILE_FLAGS " -march=armv8.2-a+fp16 ")
355-
set_source_files_properties(${MLAS_SRC_DIR}/aarch64/QgemmU8X8KernelUmmla.S PROPERTIES COMPILE_FLAGS " -march=armv8.2-a+i8mm ")
356-
set_source_files_properties(${MLAS_SRC_DIR}/aarch64/QgemmS8S8KernelSmmla.S PROPERTIES COMPILE_FLAGS " -march=armv8.2-a+i8mm ")
357357
endif()
358358

359359
if(ONNXRUNTIME_MLAS_MULTI_ARCH)

0 commit comments

Comments
 (0)