[GH-ISSUE #242] [Bug] Incorrect usage of SM100ArchSpec in sm90_bf16_k_grouped_gemm #136

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

Originally created by @Jumbo0715 on GitHub (Dec 23, 2025).
Original GitHub issue: https://github.com/deepseek-ai/DeepGEMM/issues/242

I noticed a potential copy-paste error in this function sm90_bf16_k_grouped_gemm, the code currently uses SM100ArchSpec to calculate the block sizes for TMA descriptors. Since this file is intended for SM90 implementation, this appears to be incorrect.

github.com/deepseek-ai/DeepGEMM@9b680f4284/csrc/jit_kernels/impls/sm90_bf16_gemm.hpp (L259-L274)

Originally created by @Jumbo0715 on GitHub (Dec 23, 2025). Original GitHub issue: https://github.com/deepseek-ai/DeepGEMM/issues/242 I noticed a potential copy-paste error in this function `sm90_bf16_k_grouped_gemm`, the code currently uses SM100ArchSpec to calculate the block sizes for TMA descriptors. Since this file is intended for SM90 implementation, this appears to be incorrect. https://github.com/deepseek-ai/DeepGEMM/blob/9b680f428484625f4f35dc3617f134187c6bcd4a/csrc/jit_kernels/impls/sm90_bf16_gemm.hpp#L259-L274
Author
Owner

@zheanxu commented on GitHub (Jan 6, 2026):

Thanks for the heads up! Fixed in #270

<!-- gh-comment-id:3712796352 --> @zheanxu commented on GitHub (Jan 6, 2026): Thanks for the heads up! Fixed in #270
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#136
No description provided.