[GH-ISSUE #92] AssertionError: best_smem_config is not None #50

Closed
opened 2026-10-26 21:11:14 +00:00 by bsakaguchi · 1 comment
Owner

Originally created by @hewr2010 on GitHub (Apr 30, 2025).
Original GitHub issue: https://github.com/deepseek-ai/DeepGEMM/issues/92

A recent update breaks MLA forward_absorb

With DeepSeek-V3, we have q_nope [S,128,128] and q_nope_val [128,ceil(S/256)*256,128] after fp8 quant. We have k=q_nope_val.shape[2]=128 and this line would lead to empty stage_candidates

Originally created by @hewr2010 on GitHub (Apr 30, 2025). Original GitHub issue: https://github.com/deepseek-ai/DeepGEMM/issues/92 A recent update breaks MLA [forward_absorb](https://github.com/sgl-project/sglang/blob/v0.4.5.post3/python/sglang/srt/models/deepseek_v2.py#L708) * https://github.com/deepseek-ai/DeepGEMM/commit/d374456787c With DeepSeek-V3, we have `q_nope [S,128,128]` and `q_nope_val [128,ceil(S/256)*256,128]` after fp8 quant. We have `k=q_nope_val.shape[2]=128` and [this line](https://github.com/deepseek-ai/DeepGEMM/commit/d374456787c51ad9c4b3c5dbc3668ea3ad9bb9c2#diff-d07651f3d119035344e6b5ca25f1e6825d2a4029b3b5f843e2986269da2db2a9R133) would lead to empty `stage_candidates`
Author
Owner

@LyricZhao commented on GitHub (May 7, 2025):

Fixed in github.com/deepseek-ai/DeepGEMM@daec8fd2fc, thanks for feedback! Close it now.

<!-- gh-comment-id:2856929593 --> @LyricZhao commented on GitHub (May 7, 2025): Fixed in https://github.com/deepseek-ai/DeepGEMM/commit/daec8fd2fc04c27ec141075e0f0c257e178ac72c, thanks for feedback! Close it now.
Sign in to join this conversation.
No labels
pull-request
No milestone
No project
No assignees
1 participant
Notifications
Due date
The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference
deepseek-ai/DeepGEMM#50
No description provided.