biondizzle/nvfp4-megamoe-kernel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: cast typo

Browse Source
This commit is contained in:
biondizzle
2026-05-29 04:39:21 +00:00
parent 8908b697dd
commit 8c17f65f5b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/unit/test_fmha_tma.cu
View File

@@ -159,7 +159,7 @@ struct TmaDescSet {
// The data in GMEM starts at d_q, shape (T, HD), stride (HD, 1).
// We treat it as (128, HD) — rows beyond T are garbage, kernel ignores them.
uint32_t q_tile_rows = 128;
if (!create_tma_desc_2d_bf16(&tma_q, d_q, 128, (uint64_t)hd, q_tile_rows, (uint32_t*)hd)) {
if (!create_tma_desc_2d_bf16(&tma_q, d_q, 128, (uint64_t)hd, q_tile_rows, (uint32_t)hd)) {
printf(" Failed to create Q TMA desc: rows=128, cols=%d, tile_rows=128, tile_cols=%d\n", hd, hd);
return false;
}