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

fix: bool type mismatch

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

3
tests/unit/test_fmha_tma.cu
View File

@@ -159,8 +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;
CUresult q_res = create_tma_desc_2d_bf16(&tma_q, d_q, 128, (uint64_t)hd, q_tile_rows, (uint32_t)hd);
if (q_res != true) {
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;
}