Move files related to EPLB (#7580)

This commit is contained in:
fzyzcjy
2025-06-29 15:39:38 -07:00
committed by GitHub
parent e3f9b54819
commit 0c9c6c75a8
22 changed files with 42 additions and 54 deletions
+3 -3
View File
@@ -18,12 +18,12 @@ from typing import Callable, Optional
import torch
import torch.nn.functional as F
from sglang.srt.managers import expert_location_dispatch
from sglang.srt.managers.expert_distribution import (
from sglang.srt.eplb import expert_location_dispatch
from sglang.srt.eplb.expert_distribution import (
ExpertDistributionRecorder,
get_global_expert_distribution_recorder,
)
from sglang.srt.managers.expert_location_dispatch import (
from sglang.srt.eplb.expert_location_dispatch import (
ExpertLocationDispatchInfo,
topk_ids_logical_to_physical,
)