kvcache-ai/ktransformers
1
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2026-04-20 06:18:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

[build]: fix amx cmake build support

Browse Source
This commit is contained in:
KMSorSMS
2025-11-03 03:58:36 +00:00
parent 164b13adac
commit 49a49ade66
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
kt-kernel/CMakeLists.txt
View File

@@ -347,6 +347,7 @@ if(HOST_IS_X86)
add_compile_definitions(USE_AMX_AVX_KERNEL=1)
if(KTRANSFORMERS_CPU_USE_AMX)
add_compile_definitions(HAVE_AMX=1)
list(APPEND ARCH_FLAGS -mamx-tile)
message(STATUS "AMX enabled")
endif()
# add_executable(amx-test ${CMAKE_CURRENT_SOURCE_DIR}/operators/amx/amx-test.cpp)