fix some bugs

yizhang2077 · yizhang2077 · commit 3b40e74d9927 · 2025-05-23T06:40:45.000Z
diff --git a/python/sglang/srt/managers/expert_distribution.py b/python/sglang/srt/managers/expert_distribution.py
@@ -690,7 +690,9 @@ def _convert_global_physical_count_to_logical_count(
     )
     logical_count.scatter_add_(
         dim=2,
-        index=physical_to_logical_map.unsqueeze(0).expand(dim_extra, -1, -1).to(torch.int64),
+        index=physical_to_logical_map.unsqueeze(0)
+        .expand(dim_extra, -1, -1)
+        .to(torch.int64),
         src=global_physical_count,
     )
     return logical_count

Original file line number	Diff line number	Diff line change
`@@ -690,7 +690,9 @@ def _convert_global_physical_count_to_logical_count(`
`690`	`690`	`)`
`691`	`691`	`logical_count.scatter_add_(`
`692`	`692`	`dim=2,`
`693`		`- index=physical_to_logical_map.unsqueeze(0).expand(dim_extra, -1, -1).to(torch.int64),`
	`693`	`+ index=physical_to_logical_map.unsqueeze(0)`
	`694`	`+ .expand(dim_extra, -1, -1)`
	`695`	`+ .to(torch.int64),`
`694`	`696`	`src=global_physical_count,`
`695`	`697`	`)`
`696`	`698`	`return logical_count`