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

Merge pull request #1542 from KMSorSMS/main

Browse Source 
[build]: fix amx cmake build support
This commit is contained in:
Jianwei Dong
2025-11-03 20:01:32 +08:00
committed by GitHub
parent 0d7482fcc4 b8f099c8b3
commit 9f2cb4787c
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 -mamx-bf16 -mamx-int8)
message(STATUS "AMX enabled")
list(APPEND ARCH_FLAGS -mamx-tile)
endif()