Skip to content

GitLab

"cacheflow/git@developer.sourcefind.cn:SIYIXNI/vllm.git" did not exist on "eedb46bf03818796536358f3767ee2b6a619b4f5"
Commit e34c2ee3 authored by Andriy Roshchenko's avatar Andriy Roshchenko
Browse files

Add FP8 type selection into client_axample CMakeLists.txt

parent 1ccb8112
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 0 deletions
client_example/CMakeLists.txt
View file @ e34c2ee3
......@@ -56,6 +56,14 @@ if (GPU_TARGETS)
add_definitions(-DCK_USE_WMMA)
set(CK_USE_WMMA "ON")
endif()
if (GPU_TARGETS MATCHES "gfx12" OR GPU_TARGETS MATCHES "gfx950")
add_definitions(-DCK_USE_OCP_FP8)
set(CK_USE_OCP_FP8 "ON")
endif()
if (GPU_TARGETS MATCHES "gfx90a" OR GPU_TARGETS MATCHES "gfx94")
add_definitions(-DCK_USE_FNUZ_FP8)
set(CK_USE_FNUZ_FP8 "ON")
endif()
else()
add_definitions(-DCK_USE_WMMA -DCK_USE_XDL)
set(CK_USE_XDL "ON")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment