From 6c749399b7dc939a592e30d5964d7bdf255427aa Mon Sep 17 00:00:00 2001 From: Richard Zou Date: Thu, 9 Apr 2026 14:48:59 +0200 Subject: [PATCH] [BugFix] fix tests/kernels/moe/test_moe_layer.py (#39404) Signed-off-by: Richard Zou --- vllm/model_executor/layers/fused_moe/runner/moe_runner_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/model_executor/layers/fused_moe/runner/moe_runner_base.py b/vllm/model_executor/layers/fused_moe/runner/moe_runner_base.py index 481e787e2..ce3faba5f 100644 --- a/vllm/model_executor/layers/fused_moe/runner/moe_runner_base.py +++ b/vllm/model_executor/layers/fused_moe/runner/moe_runner_base.py @@ -40,7 +40,7 @@ from vllm.utils.torch_utils import ( def get_layer_from_name(layer_name: str) -> torch.nn.Module: forward_context: ForwardContext = get_forward_context() - if layer_name == "from_forward_context": + if not HAS_OPAQUE_TYPE and layer_name == "from_forward_context": all_moe_layers = forward_context.all_moe_layers assert all_moe_layers is not None moe_layer_index = forward_context.moe_layer_index