Commit 993cf390 authored by yuguo-Jack's avatar yuguo-Jack
Browse files

fa

parent e4960fcd
......@@ -33,7 +33,7 @@ if(WITH_ROCM)
set(FLASHATTN_C_FLAGS ${CMAKE_C_FLAGS})
set(FLASHATTN_C_FLAGS_DEBUG ${CMAKE_C_FLAGS_DEBUG})
set(FLASHATTN_C_FLAGS_RELEASE ${CMAKE_C_FLAGS_RELEASE})
set(FLASHATTN_CXX_FLAGS "${CMAKE_CXX_FLAGS} -w -fPIC -O3 -std=c++17 -D__HIP_PLATFORM_HCC__=1 --offload-arch=gfx906 --offload-arch=gfx926 --offload-arch=gfx928 -D__gfx90a__")
set(FLASHATTN_CXX_FLAGS "${CMAKE_CXX_FLAGS} -w -fPIC -O3 -std=c++17 -D__HIP_PLATFORM_HCC__=1 --offload-arch=gfx906 --offload-arch=gfx926 --offload-arch=gfx928 -D__gfx908__")
set(FLASHATTN_CXX_FLAGS_RELEASE ${CMAKE_CXX_FLAGS_RELEASE})
set(FLASHATTN_CXX_FLAGS_DEBUG ${CMAKE_CXX_FLAGS_DEBUG})
......
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