fix the deepep 8 gpu unit test (#14601)

2025-12-09 01:40:09 -08:00
parent 9a426fc5ef
commit f0e948a0f1
2 changed files with 5 additions and 2 deletions
--- a/python/sglang/srt/layers/moe/utils.py
+++ b/python/sglang/srt/layers/moe/utils.py
@@ -284,7 +284,7 @@ def speculative_moe_a2a_backend_context():
    global MOE_A2A_BACKEND
    original_backend = MOE_A2A_BACKEND
    try:
-        MOE_A2A_BACKEND = MoeA2ABackend.NONE
+        MOE_A2A_BACKEND = get_speculative_moe_a2a_backend()
        yield
    finally:
        MOE_A2A_BACKEND = original_backend