saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-31 23:27:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix for the build break in AMX feature on Mac OS. (#16390)

Browse source 
### Description
Fixing the build break issue in Apple pipeline due to AMX flag removal.
This commit is contained in:
Dipanjan Sengupta 2023-06-16 21:00:41 -07:00 committed by GitHub
parent 8fdfd20191
commit 35fa6af428
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
cmake/onnxruntime_mlas.cmake
View file

@ -522,15 +522,17 @@ else()
${mlas_platform_srcs_avx512core}
)
set(mlas_platform_srcs
${mlas_platform_srcs}
${MLAS_SRC_DIR}/qgemm_kernel_amx.cpp
${MLAS_SRC_DIR}/amx_common.h
${MLAS_SRC_DIR}/x86_64/QgemmU8S8KernelAmx.S
${MLAS_SRC_DIR}/x86_64/QgemmU8S8KernelAmxCommon.S
)
set_source_files_properties(${MLAS_SRC_DIR}/qgemm_kernel_amx.cpp PROPERTIES COMPILE_FLAGS "-masm=intel -mavx2 -mavx512bw -mavx512dq -mavx512vl -mavx512f")
set_source_files_properties(${MLAS_SRC_DIR}/x86_64/QgemmU8S8KernelAmx.S PROPERTIES COMPILE_FLAGS "-mavx2 -mavx512bw -mavx512dq -mavx512vl -mavx512f")
if(NOT APPLE)
set(mlas_platform_srcs
${mlas_platform_srcs}
${MLAS_SRC_DIR}/qgemm_kernel_amx.cpp
${MLAS_SRC_DIR}/amx_common.h
${MLAS_SRC_DIR}/x86_64/QgemmU8S8KernelAmx.S
${MLAS_SRC_DIR}/x86_64/QgemmU8S8KernelAmxCommon.S
)
set_source_files_properties(${MLAS_SRC_DIR}/qgemm_kernel_amx.cpp PROPERTIES COMPILE_FLAGS "-masm=intel -mavx2 -mavx512bw -mavx512dq -mavx512vl -mavx512f")
set_source_files_properties(${MLAS_SRC_DIR}/x86_64/QgemmU8S8KernelAmx.S PROPERTIES COMPILE_FLAGS "-mavx2 -mavx512bw -mavx512dq -mavx512vl -mavx512f")
endif()
if(ONNXRUNTIME_MLAS_MULTI_ARCH)
onnxruntime_add_static_library(onnxruntime_mlas_x86_64 ${mlas_platform_srcs})

4
onnxruntime/core/mlas/lib/platform.cpp
View file

@ -408,6 +408,8 @@ Return Value:
}
}
#ifndef __APPLE__
//
// Check if the processor supports AMX-TILE and AMX-INT8
// features.
@ -419,6 +421,8 @@ Return Value:
}
}
#endif // __APPLE__
#endif // ORT_MINIMAL_BUILD
}