未验证 提交 e36cac06 编写于 作者: C Chitsing KUI 提交者: GitHub

with flash gt 11.2 (#51307)

上级 5f678698
......@@ -531,6 +531,8 @@ if(WITH_GPU
include(external/cutlass) # download, build, install cusparselt
list(APPEND third_party_deps extern_cutlass)
set(WITH_CUTLASS ON)
endif()
if(${CMAKE_CUDA_COMPILER_VERSION} GREATER_EQUAL 11.2)
include(external/flashattn)
list(APPEND third_party_deps extern_flashattn)
set(WITH_FLASHATTN ON)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册