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

[build]: in case of missing, adding two more flags: -mamx-bf16 -mamx-int8s

Browse Source
This commit is contained in:
KMSorSMS
2025-11-03 04:07:53 +00:00
parent 1e85faac77
commit b8f099c8b3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
kt-kernel/CMakeLists.txt
View File

@@ -347,7 +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)
list(APPEND ARCH_FLAGS -mamx-tile -mamx-bf16 -mamx-int8)
message(STATUS "AMX enabled")
endif()
# add_executable(amx-test ${CMAKE_CURRENT_SOURCE_DIR}/operators/amx/amx-test.cpp)