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

[AMD] Add transpose_scale arg for o_proj to fix GLM accuracy issue (#27798)

Browse Source
This commit is contained in:
Thomas Wang
2026-06-17 01:01:02 -07:00
committed by GitHub
parent f86e9b48e8
commit 21a95333d4
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/models/deepseek_common/attention_forward_methods/forward_mla.py
+8 -2
View File
@@ -664,7 +664,10 @@ class DeepseekMLAForwardMixin:
attn_bmm_output = fused_flatten_mxfp4_quant(_bmm_buf)
elif self.o_proj.weight.dtype == torch.float8_e4m3fn:
attn_bmm_output = fused_flatten_fp8_group_quant(
_bmm_buf, group_size=128, dtype_quant=torch.float8_e4m3fn
_bmm_buf,
group_size=128,
dtype_quant=torch.float8_e4m3fn,
transpose_scale=_use_aiter_bpreshuffle_gfx95,
)
else:
attn_bmm_output = _bmm_buf.flatten(1, 2)
@@ -674,7 +677,10 @@ class DeepseekMLAForwardMixin:
elif self.o_proj.weight.dtype == torch.float8_e4m3fn:
attn_bmm_output = attn_bmm_output.transpose(0, 1)
attn_bmm_output = fused_flatten_fp8_group_quant(
attn_bmm_output, group_size=128, dtype_quant=torch.float8_e4m3fn
attn_bmm_output,
group_size=128,
dtype_quant=torch.float8_e4m3fn,
transpose_scale=_use_aiter_bpreshuffle_gfx95,
)
else:
attn_bmm_output = attn_bmm_output.transpose(0, 1).flatten(1, 2)