File tree Expand file tree Collapse file tree 1 file changed +13
-6
lines changed Expand file tree Collapse file tree 1 file changed +13
-6
lines changed Original file line number Diff line number Diff line change @@ -75,6 +75,7 @@ if (MACOSX_FOUND AND CMAKE_SYSTEM_PROCESSOR STREQUAL "arm64")
7575else ()
7676 find_isa(${CPUINFO} "avx2" AVX2_FOUND)
7777 find_isa(${CPUINFO} "avx512f" AVX512_FOUND)
78+ find_isa(${CPUINFO} "Power11" POWER11_FOUND)
7879 find_isa(${CPUINFO} "POWER10" POWER10_FOUND)
7980 find_isa(${CPUINFO} "POWER9" POWER9_FOUND)
8081 find_isa(${CPUINFO} "asimd" ASIMD_FOUND) # Check for ARM NEON support
@@ -106,13 +107,19 @@ elseif (AVX2_FOUND)
106107 list (APPEND CXX_COMPILE_FLAGS "-mavx2" )
107108 message (WARNING "vLLM CPU backend using AVX2 ISA" )
108109
109- elseif (POWER9_FOUND OR POWER10_FOUND)
110+ elseif (POWER9_FOUND OR POWER10_FOUND OR POWER11_FOUND )
110111 message (STATUS "PowerPC detected" )
111- # Check for PowerPC VSX support
112- list (APPEND CXX_COMPILE_FLAGS
113- "-mvsx"
114- "-mcpu=native"
115- "-mtune=native" )
112+ if (POWER9_FOUND)
113+ list (APPEND CXX_COMPILE_FLAGS
114+ "-mvsx"
115+ "-mcpu=power9"
116+ "-mtune=power9" )
117+ elseif (POWER10_FOUND OR POWER11_FOUND)
118+ list (APPEND CXX_COMPILE_FLAGS
119+ "-mvsx"
120+ "-mcpu=power10"
121+ "-mtune=power10" )
122+ endif ()
116123
117124elseif (ASIMD_FOUND)
118125 message (STATUS "ARMv8 or later architecture detected" )
You can’t perform that action at this time.
0 commit comments