Skip to content

GitLab

Unverified Commit 29195aaa authored by fzyzcjy's avatar fzyzcjy Committed by GitHub
Browse files

Super tiny fix expert distribution dump error (#12271)

parent 0ee831de
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
python/sglang/srt/eplb/expert_distribution.py
View file @ 29195aaa
...@@ -20,6 +20,7 @@ import time ...@@ -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): ...@@ -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():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment