xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-22 03:25:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix] Fix modular kernel tests (#27707)

Browse Source 
Signed-off-by: Bill Nell <bnell@redhat.com>
This commit is contained in:
bnellnm 2025-10-29 04:14:33 -04:00 committed by GitHub
parent 8df98c2161
commit 1891cf605a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.buildkite/test-pipeline.yaml
View File

@ -498,6 +498,8 @@ steps:
- tests/kernels/moe - tests/kernels/moe
- vllm/model_executor/layers/fused_moe/ - vllm/model_executor/layers/fused_moe/
- vllm/distributed/device_communicators/ - vllm/distributed/device_communicators/
- vllm/envs.py
- vllm/config
commands: commands:
- pytest -v -s kernels/moe --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT - pytest -v -s kernels/moe --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
parallelism: 2 parallelism: 2

1
tests/kernels/moe/modular_kernel_tools/common.py
View File

@ -138,6 +138,7 @@ class Config:
} }
backend = self.all2all_backend() backend = self.all2all_backend()
vllm_config.parallel_config.all2all_backend = backend
if backend is not None: if backend is not None:
env_dict.update({"VLLM_ALL2ALL_BACKEND": backend}) env_dict.update({"VLLM_ALL2ALL_BACKEND": backend})