[ROCm][CI] Fix AITER state leak in shared_fused_moe_routed_transform test (#38137)

Signed-off-by: Andreas Karatzas <akaratza@amd.com>
This commit is contained in:
Andreas Karatzas
2026-03-26 11:26:46 -05:00
committed by GitHub
parent 0aac2048bf
commit bdc1719eb9
3 changed files with 21 additions and 3 deletions

View File

@@ -125,8 +125,15 @@ def test_routing_strategy_integration(monkeypatch, device):
env_name = "VLLM_MOE_ROUTING_SIMULATION_STRATEGY"
monkeypatch.setenv(env_name, strategy)
# Force reload of environment variable
envs.environment_variables[env_name] = lambda s=strategy: s
# Temporarily override the envs lookup so the router factory
# reads the monkeypatched value instead of the module-load-time
# default. Use monkeypatch.setitem so the original lambda is
# restored automatically at teardown.
monkeypatch.setitem(
envs.environment_variables,
env_name,
lambda s=strategy: s,
)
# Test the select_experts method
topk_weights, topk_ids = fused_moe.router.select_experts(