Skip to content

GitLab

Unverified Commit 037140c0 authored by thatPepe's avatar thatPepe Committed by GitHub
Browse files

Merge pull request #1067 from InfiniTensor/issue/1033_fix 

skip build flash attn target when disabled
parents a8ea2306 505e9b69
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
xmake.lua
View file @ 037140c0
...@@ -461,7 +461,7 @@ target("infinicore_cpp_api") ...@@ -461,7 +461,7 @@ target("infinicore_cpp_api")
add_linkdirs(INFINI_ROOT.."/lib") add_linkdirs(INFINI_ROOT.."/lib")
add_links("infiniop", "infinirt", "infiniccl") add_links("infiniop", "infinirt", "infiniccl")
if get_config("flash-attn") ~= "" then if get_config("flash-attn") ~= "" and get_config("flash-attn") ~= nil then
add_installfiles("(builddir)/$(plat)/$(arch)/$(mode)/flash-attn*.so", {prefixdir = "lib"}) add_installfiles("(builddir)/$(plat)/$(arch)/$(mode)/flash-attn*.so", {prefixdir = "lib"})
if has_config("nv-gpu") then if has_config("nv-gpu") then
add_deps("flash-attn-nvidia") add_deps("flash-attn-nvidia")
......
xmake/nvidia.lua
View file @ 037140c0
...@@ -181,6 +181,7 @@ target("flash-attn-nvidia") ...@@ -181,6 +181,7 @@ target("flash-attn-nvidia")
before_build(function (target) before_build(function (target)
print("Flash Attention not available, skipping flash-attn-nvidia build") print("Flash Attention not available, skipping flash-attn-nvidia build")
end) end)
on_build(function (target) end)
end end
on_install(function (target) end) on_install(function (target) end)
......
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