[Feature] Add AllGather-KV sequence-parallel attention str… (#4968)

Signed-off-by: zhuhh97 <zhuhonghan@huawei.com>
Signed-off-by: dph97 <nenbaying@163.com>
Signed-off-by: zhuhh97 <zhh@unknown>
Co-authored-by: Claude Opus 4.7 <noreply@anthropic.com>
Co-authored-by: dengyunyang <584797741@qq.com>
This commit is contained in:
authored and GitHub committed 2026-07-24 14:45:26 +00:00
1 parent 8927ec057c
commit 84bbd57e4d
33 files changed
+1181 -135

No files matched your search

@@ -45,6 +45,8 @@ def patched_mgr_env(sp_size=1):
target = _TRANSFORMER_MODULE
patches = [
patch(f"{target}.get_sequence_parallel_world_size", return_value=sp_size),
patch(f"{target}.get_allgather_parallel_world_size", return_value=sp_size),
patch(f"{target}.get_ulysses_parallel_world_size", return_value=1, create=True),
patch(f"{target}.get_sequence_parallel_rank", return_value=0),
patch(f"{target}.Attention", MockAttention),
]