chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 11 Packages Projects Releases Wiki Activity

Refactor --debug-tensor-dump-layers to list (#12691)

Browse Source
This commit is contained in:
Yuhong Guo
2025-11-05 19:30:01 +08:00
committed by GitHub
parent dc4f541823
commit 4d84f886e7
3 changed files with 26 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/sglang/srt/server_args.py
View File

@@ -519,8 +519,8 @@ class ServerArgs:
# Debug tensor dumps
debug_tensor_dump_output_folder: Optional[str] = None
# -1 mean dump all layers.
debug_tensor_dump_layers: int = -1
# None means dump all layers.
debug_tensor_dump_layers: Optional[List[int]] = None
# TODO(guoyuhong): clean the old dumper code.
debug_tensor_dump_input_file: Optional[str] = None
debug_tensor_dump_inject: bool = False
@@ -3424,8 +3424,8 @@ class ServerArgs:
parser.add_argument(
"--debug-tensor-dump-layers",
type=int,
default=-1,
help="The layer number for dumping tensors.",
nargs="+",
help="The layer ids to dump. Dump all layers if not specified.",
)
parser.add_argument(
"--debug-tensor-dump-input-file",