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

fix: remove n_kv_tiles from standalone test (struct doesn't have it anymore)

Browse Source
This commit is contained in:
biondizzle
2026-05-30 10:28:38 +00:00
parent f032800eaa
commit a5b47602b5
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/unit/test_fmha_6warp_multihead.cu
View File

@@ -99,7 +99,6 @@ static int test_mha(int n_h) {
params.s_k = SK;
params.scale = SCALE;
params.head_dim = HD;
params.n_kv_tiles = 0; // auto-calculate
params.q_head_stride = HD; // T=1, stride = 1 * hd
params.q_batch_stride = n_h * HD;
params.k_head_stride = SK * HD; // each head has its own K
@@ -219,7 +218,6 @@ static int test_mqa(int n_q, int n_kv) {
params.s_k = SK;
params.scale = SCALE;
params.head_dim = HD;
params.n_kv_tiles = 0; // auto-calculate
params.q_head_stride = HD;
params.q_batch_stride = n_q * HD;
params.k_head_stride = 0; // MQA: all heads share same K
@@ -314,7 +312,6 @@ static int test_batched(int n_h, int batch_size) {
params.s_k = SK;
params.scale = SCALE;
params.head_dim = HD;
params.n_kv_tiles = 0; // auto-calculate
params.q_head_stride = HD;
params.q_batch_stride = n_h * HD;
params.k_head_stride = SK * HD;