Skip to content

Commit e7ca073

Browse files
committed
fix: actually include flashmla for blackwell
1 parent 2cacb97 commit e7ca073

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -839,7 +839,7 @@ if (APHRODITE_GPU_LANG STREQUAL "CUDA")
839839
endif()
840840

841841
# FlashMLA build - requires CUDA 12.3 or later and Hopper architecture
842-
cuda_archs_loose_intersection(FLASH_MLA_ARCHS "9.0a 10.0 10.1 12.0" "${CUDA_ARCHS}")
842+
cuda_archs_loose_intersection(FLASH_MLA_ARCHS "9.0;9.0a;10.0;10.1;12.0" "${CUDA_ARCHS}")
843843
if(${CMAKE_CUDA_COMPILER_VERSION} VERSION_GREATER 12.3 AND FLASH_MLA_ARCHS)
844844
set(FlashMLA_SOURCES
845845
kernels/flash_mla/flash_api.cpp

0 commit comments

Comments
 (0)