youngkingdom/vllm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Misc] Refactor AllReduceFusionPass. Remove parameter (#20918)

Browse Source 
Signed-off-by: ilmarkov <imarkov@redhat.com>
Co-authored-by: ilmarkov <imarkov@redhat.com>
This commit is contained in:
Ilya Markov
2025-07-15 08:57:40 +02:00
committed by GitHub
parent d4d309409f
commit 37a7d5d74a
3 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/compile/test_fusion_all_reduce.py
View File

@ -132,9 +132,7 @@ def all_reduce_fusion_pass_on_test_model(local_rank: int, world_size: int,
dtype=dtype,
seed=42)
all_reduce_fusion_pass = AllReduceFusionPass(
vllm_config, vllm_config.compilation_config.pass_config.
fi_allreduce_fusion_max_token_num)
all_reduce_fusion_pass = AllReduceFusionPass(vllm_config)
backend = TestBackend(all_reduce_fusion_pass)
model = test_model_cls(hidden_size)