Skip to content

GitLab

Unverified Commit 03bf1f8a authored by Antoni Baum's avatar Antoni Baum Committed by GitHub
Browse files

Don't use kwargs in autograd functions (#3)

parent 7f3b182c
Pipeline #2017 failed with stages
in 0 seconds
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
vllm_flash_attn/flash_attn_interface.py
View file @ 03bf1f8a
......@@ -709,7 +709,7 @@ def flash_attn_qkvpacked_func(
alibi_slopes,
deterministic,
return_attn_probs,
out=out,
out,
)
......@@ -786,7 +786,7 @@ def flash_attn_kvpacked_func(
alibi_slopes,
deterministic,
return_attn_probs,
out=out,
out,
)
......@@ -863,7 +863,7 @@ def flash_attn_func(
alibi_slopes,
deterministic,
return_attn_probs,
out=out,
out,
)
......@@ -928,7 +928,7 @@ def flash_attn_varlen_qkvpacked_func(
alibi_slopes,
deterministic,
return_attn_probs,
out=out,
out,
)
......@@ -1019,7 +1019,7 @@ def flash_attn_varlen_kvpacked_func(
alibi_slopes,
deterministic,
return_attn_probs,
out=out,
out,
)
......@@ -1112,7 +1112,7 @@ def flash_attn_varlen_func(
deterministic,
return_attn_probs,
block_table,
out=out,
out,
)
......
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