Skip to content

Commit b9a1791

Browse files
Akashcodes732Akash Kaothalkar
andauthored
[Hardware][POWER] Add IBM POWER11 Support to CPU Extension Detection (#19082)
Signed-off-by: Akash Kaothalkar <[email protected]> Co-authored-by: Akash Kaothalkar <[email protected]>
1 parent 989dcee commit b9a1791

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

cmake/cpu_extension.cmake

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,7 @@ if (MACOSX_FOUND AND CMAKE_SYSTEM_PROCESSOR STREQUAL "arm64")
7575
else()
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

117124
elseif (ASIMD_FOUND)
118125
message(STATUS "ARMv8 or later architecture detected")

0 commit comments

Comments
 (0)