| 1 | # SPDX-License-Identifier: Apache-2.0 |
| 2 | # SPDX-FileCopyrightText: Copyright contributors to the vLLM project |
| 3 | import pytest |
| 4 | |
| 5 | import vllm.envs as envs |
| 6 | |
| 7 | |
| 8 | @pytest.fixture(autouse=True) |
| 9 | def enable_batch_invariant_mode(monkeypatch: pytest.MonkeyPatch): |
| 10 | """Automatically enable batch invariant kernel overrides for all tests.""" |
| 11 | monkeypatch.setattr(envs, "VLLM_BATCH_INVARIANT", True) |
| 12 | monkeypatch.setenv("VLLM_BATCH_INVARIANT", "1") |