lalala-sh
a3ccd5dca1
[rocm-libraries] ROCm/rocm-libraries#5225 (commit 880166b)
...
[CK] fix moe memset size which is bigger than alloc
## Motivation
Fix an out-of-bounds hipMemsetAsync in DeviceMoeGemmBlockScale that
crashes split-K MOE GEMM with "HIP runtime error: invalid argument".
When KBatch > 1, the invoker zeroes the output buffer using arg.M *
arg.N as the byte count. However, arg.M is the padded sorted-token-id
length from MOE routing, which can be much larger than the actual output
allocation (NumTokens * TopK * N). This causes hipMemsetAsync to write
beyond the buffer, and the silently-swallowed HIP error propagates to
the subsequent kernel launch via hipGetLastError().
This patch replaces arg.M with arg.NumTokens * arg.TopK so the memset
matches the actual output size.
## Technical Details
<!-- Explain the changes along with any relevant GitHub links. -->
## Test Plan
<!-- Explain any relevant testing done to verify this PR. -->
## Test Result
<!-- Briefly summarize test outcomes. -->
## Submission Checklist
- [x] Look over the contributing guidelines at
https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests .
2026-03-16 09:30:57 +00:00
..
2026-01-27 12:49:47 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2026-01-26 12:57:09 -08:00
2025-11-18 10:17:18 -08:00
2026-01-27 12:49:47 -08:00
2025-11-18 10:17:18 -08:00
2026-01-26 10:20:30 -08:00
2026-01-26 10:20:30 -08:00
2026-01-27 12:49:47 -08:00
2026-01-27 12:49:47 -08:00
2025-11-20 10:45:57 -08:00
2026-01-26 10:20:30 -08:00
2026-01-27 12:49:47 -08:00
2026-01-26 12:57:09 -08:00
2026-01-27 12:49:47 -08:00
2026-01-26 12:57:09 -08:00
2026-01-27 12:49:47 -08:00
2025-11-18 10:17:18 -08:00
2026-01-27 12:49:47 -08:00
2026-01-27 12:49:47 -08:00
2026-01-23 12:39:03 -08:00
2026-01-26 12:57:09 -08:00
2026-01-23 12:39:03 -08:00
2026-01-27 12:49:47 -08:00
2026-01-27 12:49:47 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2026-01-27 12:49:47 -08:00
2026-01-27 12:49:47 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-12-18 07:59:45 +01:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2026-01-26 12:57:09 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-20 10:45:57 -08:00
2026-01-26 12:57:09 -08:00
2025-11-18 10:17:18 -08:00
2026-01-27 12:49:47 -08:00
2025-12-11 09:06:20 +01:00
2025-12-11 09:06:20 +01:00
2025-12-11 09:06:20 +01:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-12-11 09:06:20 +01:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2026-01-27 12:49:47 -08:00
2026-01-26 12:57:09 -08:00
2025-11-18 10:17:18 -08:00
2026-01-15 07:19:31 -08:00
2025-12-11 09:06:20 +01:00
2026-01-26 12:57:09 -08:00
2025-11-18 10:17:18 -08:00
2026-01-26 12:57:09 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2026-01-27 12:49:47 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2026-01-27 12:49:47 -08:00
2026-01-27 12:49:47 -08:00
2025-11-18 10:17:18 -08:00
2026-01-27 12:49:47 -08:00
2026-03-11 16:40:12 +00:00
2026-02-27 03:06:29 +00:00
2026-02-27 03:06:29 +00:00
2025-12-31 15:41:15 -08:00
2026-01-29 17:40:28 +00:00
2026-03-11 16:40:12 +00:00
2026-01-29 17:40:28 +00:00
2026-03-11 16:40:12 +00:00
2026-01-27 12:49:47 -08:00
2026-03-11 16:40:12 +00:00
2025-12-31 15:41:15 -08:00
2026-02-11 09:08:38 +00:00
2026-03-09 16:35:26 +00:00
2025-12-14 12:49:12 -08:00
2025-11-20 10:45:57 -08:00
2026-03-11 16:40:12 +00:00
2026-03-11 22:39:20 +00:00
2026-03-11 22:39:20 +00:00
2026-01-27 12:49:47 -08:00
2025-11-18 10:17:18 -08:00
2026-01-23 12:19:51 +01:00
2025-12-14 12:49:12 -08:00
2025-11-18 10:17:18 -08:00
2026-03-11 22:39:20 +00:00
2025-12-18 13:12:15 -07:00
2026-02-25 05:17:08 +00:00
2026-02-25 05:17:08 +00:00
2025-11-20 10:45:57 -08:00
2025-12-23 11:33:09 +01:00
2026-01-26 12:57:09 -08:00
2026-01-27 12:49:47 -08:00
2025-12-19 15:58:51 -07:00
2026-02-26 00:28:58 +00:00
2026-01-26 12:57:09 -08:00
2026-02-25 05:17:08 +00:00
2026-01-27 12:49:47 -08:00
2026-01-27 12:49:47 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2026-03-16 09:30:57 +00:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-20 17:27:05 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-18 10:17:18 -08:00
2025-11-26 11:00:05 -07:00
2025-11-26 11:00:05 -07:00
2026-01-27 12:49:47 -08:00
2025-11-26 11:00:05 -07:00
2026-01-08 08:02:02 +01:00
2025-11-26 11:00:05 -07:00