minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Super tiny fix expert distribution dump error (#12271)

Browse Source
This commit is contained in:
fzyzcjy
2025-10-28 15:20:55 -07:00
committed by GitHub
parent 0ee831dee0
commit 29195aaa6e
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/eplb/expert_distribution.py
+2 -1
View File
@@ -20,6 +20,7 @@ import time
from abc import ABC from abc import ABC
from collections import deque from collections import deque
from contextlib import contextmanager from contextlib import contextmanager
from pathlib import Path
from typing import TYPE_CHECKING, Any, Dict, List, Literal, Optional, Tuple, Type from typing import TYPE_CHECKING, Any, Dict, List, Literal, Optional, Tuple, Type
import einops import einops
@@ -868,7 +869,7 @@ class _StatAccumulator(_UtilizationRateAccumulatorMixin):
def _dump_to_file(name, data): def _dump_to_file(name, data):
save_dir = envs.SGLANG_EXPERT_DISTRIBUTION_RECORDER_DIR.get() save_dir = Path(envs.SGLANG_EXPERT_DISTRIBUTION_RECORDER_DIR.get())
path_output = save_dir / name path_output = save_dir / name
logger.info(f"Write expert distribution to {path_output}") logger.info(f"Write expert distribution to {path_output}")
if not save_dir.exists(): if not save_dir.exists():