From 1891cf605ac015016cca38acc9e1950f70586f2e Mon Sep 17 00:00:00 2001 From: bnellnm <49004751+bnellnm@users.noreply.github.com> Date: Wed, 29 Oct 2025 04:14:33 -0400 Subject: [PATCH] [Bugfix] Fix modular kernel tests (#27707) Signed-off-by: Bill Nell --- .buildkite/test-pipeline.yaml | 2 ++ tests/kernels/moe/modular_kernel_tools/common.py | 1 + 2 files changed, 3 insertions(+) diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml index 16f87e98afb7f..e166f320f9c3f 100644 --- a/.buildkite/test-pipeline.yaml +++ b/.buildkite/test-pipeline.yaml @@ -498,6 +498,8 @@ steps: - tests/kernels/moe - vllm/model_executor/layers/fused_moe/ - vllm/distributed/device_communicators/ + - vllm/envs.py + - vllm/config commands: - pytest -v -s kernels/moe --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT parallelism: 2 diff --git a/tests/kernels/moe/modular_kernel_tools/common.py b/tests/kernels/moe/modular_kernel_tools/common.py index c517e5c026b46..1d925dc1bea8f 100644 --- a/tests/kernels/moe/modular_kernel_tools/common.py +++ b/tests/kernels/moe/modular_kernel_tools/common.py @@ -138,6 +138,7 @@ class Config: } backend = self.all2all_backend() + vllm_config.parallel_config.all2all_backend = backend if backend is not None: env_dict.update({"VLLM_ALL2ALL_BACKEND": backend})