saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-04 23:59:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[AMX] add assembler check (#15055)

Browse source 
### Description
<!-- Describe your changes. -->

AMX isn't supportted until assembler 2.40 even though the GCC frontend
supports it.


### 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. -->
This commit is contained in:
JiCheng 2023-03-22 07:57:22 +08:00 committed by GitHub
parent 3e2d453b64
commit 126e7bf15f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cmake/onnxruntime_mlas.cmake
View file

@ -7,7 +7,17 @@ set(MLAS_SRC_DIR ${ONNXRUNTIME_ROOT}/core/mlas/lib)
set(MLAS_AMX_SUPPORTED FALSE)
if(CMAKE_CXX_COMPILER_ID STREQUAL "GNU" AND CMAKE_C_COMPILER_VERSION VERSION_GREATER_EQUAL 11)
set(MLAS_AMX_SUPPORTED TRUE)
# match assembler version, AMX instructions are supported from 2.40
if (CMAKE_ASM-ATT_COMPILER_ID STREQUAL "GNU")
execute_process(
COMMAND ${CMAKE_ASM-ATT_COMPILER} --version
OUTPUT_VARIABLE _gas_version
)
# 2.40 or later
if (_gas_version MATCHES "GNU.[Aa]ssembler.*(2\\.[4-9][0-9])")
set(MLAS_AMX_SUPPORTED TRUE)
endif()
endif()
endif()
if(CMAKE_CXX_COMPILER_ID MATCHES "MSVC")