From 7412ceb4eb159c92841491075bde63735bac7c8d Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Sat, 31 Jan 2026 01:01:55 -0800 Subject: [PATCH] [Auto Sync] Update linear.py to assert shapes (20260130) (#17966) Co-authored-by: github-actions[bot] Co-authored-by: Byron Hsu --- python/sglang/srt/layers/linear.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/python/sglang/srt/layers/linear.py b/python/sglang/srt/layers/linear.py index 90ff6dc71..90e133ce7 100644 --- a/python/sglang/srt/layers/linear.py +++ b/python/sglang/srt/layers/linear.py @@ -1222,7 +1222,9 @@ class QKVParallelLinear(ColumnParallelLinear): "for all partitions." ) - assert param_data.shape == loaded_weight.shape + assert ( + param_data.shape == loaded_weight.shape + ), f"{param_data.shape=} {loaded_weight.shape=}" param_data.copy_(loaded_weight) @@ -1371,7 +1373,9 @@ class RowParallelLinear(LinearBase): if len(loaded_weight.shape) == 0: loaded_weight = loaded_weight.reshape(1) - assert param_data.shape == loaded_weight.shape + assert ( + param_data.shape == loaded_weight.shape + ), f"{param_data.shape=} {loaded_weight.shape=}" param_data.copy_(loaded_weight) def weight_loader_v2(self, param: BasevLLMParameter, loaded_weight: torch.Tensor):