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

fix: ambiguous MMA_K_BF16 in test

Browse Source
This commit is contained in:
biondizzle
2026-05-29 19:32:15 +00:00
parent 4459ddefdd
commit bd4f09d514
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/unit/test_fmha_6warp_tma.cu
View File

@@ -25,9 +25,9 @@ static float bf16_to_f32_host(bf16_t h) { uint32_t u=(uint32_t)h<<16; float f; m
constexpr int HD = HD_VAL;
constexpr int SK = 128;
constexpr int MMA_K_BF16 = 16;
constexpr int TILE_SZ = 128 * MMA_K_BF16;
constexpr int V_SUB_SZ = 16 * MMA_K_BF16;
constexpr int MY_MMA_K = 16;
constexpr int TILE_SZ = 128 * MY_MMA_K;
constexpr int V_SUB_SZ = 16 * MY_MMA_K;
constexpr int TMEM_N = (HD <= 128) ? 128 : 256;
#include "dsv4/kernels/attention/fmha_6warp_tma.cuh"