xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-04-22 05:57:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Suppress FA3 "wgmma.mma_async instructions are serialized" warning

Browse Source 
Signed-off-by: mgoin <mgoin64@gmail.com>
This commit is contained in:
mgoin 2025-09-24 14:35:39 -04:00
parent d83f3f7cb3
commit 1133c53c0e
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
cmake/external_projects/vllm_flash_attn.cmake
View File

@ -62,6 +62,12 @@ install(CODE "set(CMAKE_INSTALL_PREFIX \"\${CMAKE_INSTALL_PREFIX}/vllm/\")" ALL_
FetchContent_MakeAvailable(vllm-flash-attn)
message(STATUS "vllm-flash-attn is available at ${vllm-flash-attn_SOURCE_DIR}")
# Suppress ptxas warnings for flash attention compilation
if(VLLM_GPU_LANG STREQUAL "CUDA")
# Add ptxas flags to suppress C7520 warnings about wgmma.mma_async serialization
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Xptxas --diag_suppress=7520")
endif()
# Restore the install prefix
install(CODE "set(CMAKE_INSTALL_PREFIX \"\${OLD_CMAKE_INSTALL_PREFIX}\")" ALL_COMPONENTS)
install(CODE "set(CMAKE_INSTALL_LOCAL_ONLY TRUE)" ALL_COMPONENTS)