Skip to content

Commit 2818d0a

Browse files
committed
fix merge conflicts
Signed-off-by: augusto.yjh <augusto.yjh@antgroup.com>
1 parent 268d439 commit 2818d0a

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

vllm/attention/ops/common.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,6 @@ def _cp_lse_common(
203203

204204
cp_attn_lse = cp_attn_lse.contiguous()
205205
lses = cp_group.all_gather(cp_attn_lse, dim=0).view_as(lses)
206-
<<<<<<< HEAD
207206
out, lse = correct_attn_out(cp_attn_out, lses, cp_group.rank_in_group, ctx, is_lse_base_on_e=is_lse_base_on_e)
208207
return out, lse
209208

@@ -221,13 +220,6 @@ def cp_lse_ag_out_rs(
221220
cp_attn_lse: [ B, H ]
222221
"""
223222
out, lse = _cp_lse_common(cp_attn_out, cp_attn_lse, cp_group, ctx=ctx, is_lse_base_on_e=is_lse_base_on_e)
224-
out, lse = correct_attn_out(
225-
cp_attn_out,
226-
lses,
227-
cp_group.rank_in_group,
228-
ctx,
229-
is_lse_base_on_e=is_lse_base_on_e,
230-
)
231223
out = cp_group.reduce_scatter(out, dim=1)
232224

233225
if return_lse:

0 commit comments

Comments
 (0)