Skip to content

GitLab

"lib/llm/src/vscode:/vscode.git/clone" did not exist on "a1766e4a51e3568a25781c6f37672e2eafa82535"
Unverified Commit 2d9ee28c authored by Lucas Wilkinson's avatar Lucas Wilkinson Committed by GitHub
Browse files

[CI/Test Fix] Fix CP tests on Blackwell (#29338) 


Signed-off-by: default avatarLucas Wilkinson <lwilkins@redhat.com>
parent 81db702e
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
vllm/attention/ops/common.py
View file @ 2d9ee28c
...@@ -194,7 +194,6 @@ def _cp_lse_common( ...@@ -194,7 +194,6 @@ def _cp_lse_common(
cp_attn_lse = cp_attn_lse.contiguous() cp_attn_lse = cp_attn_lse.contiguous()
lses = cp_group.all_gather(cp_attn_lse, dim=0).view_as(lses) lses = cp_group.all_gather(cp_attn_lse, dim=0).view_as(lses)
out, lse = correct_attn_out(cp_attn_out, lses, cp_group.rank_in_group, ctx) out, lse = correct_attn_out(cp_attn_out, lses, cp_group.rank_in_group, ctx)
assert out.is_contiguous()
return out, lse return out, lse
......
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