Unverified Commit 6c749399 authored by Richard Zou's avatar Richard Zou Committed by GitHub
Browse files

[BugFix] fix tests/kernels/moe/test_moe_layer.py (#39404)


Signed-off-by: default avatarRichard Zou <zou3519@gmail.com>
parent 91eea723
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment