biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[MoE] Move DEEP_GEMM into experts/ subdirectory (#39005)

Browse Source 
Signed-off-by: Jackmin801 <ongjackm@gmail.com>
Signed-off-by: Robert Shaw <robshaw@redhat.com>
Co-authored-by: Robert Shaw <robshaw@redhat.com>
Co-authored-by: Robert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com>
This commit is contained in:
Jackmin801
2026-04-08 12:23:08 -07:00
committed by GitHub
parent 8477fe427d
commit a776a48b1c
14 changed files with 24 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/kernels/moe/test_deepep_deepgemm_moe.py
View File

@@ -47,10 +47,12 @@ if has_deep_ep():
from .parallel_utils import DeepEPHTArgs, DeepEPLLArgs, make_deepep_a2a
if has_deep_gemm():
from vllm.model_executor.layers.fused_moe.batched_deep_gemm_moe import (
from vllm.model_executor.layers.fused_moe.experts.batched_deep_gemm_moe import (
BatchedDeepGemmExperts,
)
from vllm.model_executor.layers.fused_moe.deep_gemm_moe import DeepGemmExperts
from vllm.model_executor.layers.fused_moe.experts.deep_gemm_moe import (
DeepGemmExperts,
)
requires_deep_ep = pytest.mark.skipif(
not has_deep_ep(),