From afed30c61c20f3e85aaf07396fee22dd798fcbf9 Mon Sep 17 00:00:00 2001 From: liuhao2638 <2284903223@qq.com> Date: Fri, 17 Jul 2026 15:04:50 +0800 Subject: [PATCH 1/2] [Build] Compile FlashAttention on arm64 platforms (#79488) --------- Co-authored-by: Claude --- cmake/external/flashattn.cmake | 3 +++ cmake/third_party.cmake | 2 -- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/cmake/external/flashattn.cmake b/cmake/external/flashattn.cmake index 8600c927c3f5b3..5d5059d800c0dd 100644 --- a/cmake/external/flashattn.cmake +++ b/cmake/external/flashattn.cmake @@ -171,6 +171,9 @@ else() set(CACHE_TAR_DIR "${FA_BUILD_DIR}/flashattn_libs_${FLASHATTN_TAG}") set(SKIP_BUILD_FA OFF) + if(WITH_ARM) + set(WITH_FA_BUILD_WITH_CACHE OFF) + endif() if(WITH_FA_BUILD_WITH_CACHE) message(STATUS "Downloading ${TAR_FILE_URL} to ${CACHE_TAR_PATH}") diff --git a/cmake/third_party.cmake b/cmake/third_party.cmake index 290d1fd198e3ac..70ff374b35ee80 100755 --- a/cmake/third_party.cmake +++ b/cmake/third_party.cmake @@ -349,7 +349,6 @@ include(external/glog) # download, build, install glog ########################### include third_party according to flags ############################### if(WITH_GPU - AND NOT WITH_ARM AND NOT WIN32 AND NOT APPLE) if(${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 11.0) @@ -601,7 +600,6 @@ if(WITH_ROCM) endif() if(WITH_GPU - AND NOT WITH_ARM AND NOT WIN32 AND NOT APPLE) if(${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 12.3 From d62b6cb426789d200ed9809c6a891727e7a2fd47 Mon Sep 17 00:00:00 2001 From: liuhao2638 <2284903223@qq.com> Date: Sat, 18 Jul 2026 04:14:42 +0800 Subject: [PATCH 2/2] [Build] Avoid FlashAttention v3 for sm100 builds (#79500) --------- Co-authored-by: risemeup1111 <13125134909@163.com> --- cmake/third_party.cmake | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cmake/third_party.cmake b/cmake/third_party.cmake index 70ff374b35ee80..5663a4dcd56ecb 100755 --- a/cmake/third_party.cmake +++ b/cmake/third_party.cmake @@ -605,7 +605,7 @@ if(WITH_GPU if(${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 12.3 AND ${CMAKE_CUDA_COMPILER_VERSION} LESS_EQUAL 13.0) foreach(arch ${NVCC_ARCH_BIN}) - if(${arch} GREATER_EQUAL 90) + if(${arch} EQUAL 90) set(WITH_FLASHATTN_V3 ON) break() endif()