Reorg moe code (#2563)

2024-12-24 01:10:22 +08:00
parent 23e5e50fd5
commit e835a50021
88 changed files with 338 additions and 344 deletions
--- a/test/srt/test_fused_moe.py
+++ b/test/srt/test_fused_moe.py
@@ -4,7 +4,7 @@ import torch
 from vllm.model_executor.layers.fused_moe import fused_moe as fused_moe_vllm

 from sglang.srt.layers.activation import SiluAndMul
-from sglang.srt.layers.fused_moe_triton.fused_moe import fused_moe
+from sglang.srt.layers.moe.fused_moe_triton.fused_moe import fused_moe


 class TestFusedMOE(unittest.TestCase):