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

fix: T=1 for QK direct test (write_q_to_smem only handles row 0)

Browse Source
This commit is contained in:
biondizzle
2026-05-29 18:33:35 +00:00
parent bee7cc5f8f
commit 35d0596893
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_qk_direct.cu
View File

@@ -118,7 +118,7 @@ test_qk_direct_kernel(float* __restrict__ out_s,
int main() {
printf("Direct QK Test (HD=%d, SK=%d)\n", HD, SK);
const int T = 4;
const int T = 1;
bf16_t* h_q = (bf16_t*)calloc(128 * HD, sizeof(bf16_t));
bf16_t* h_k = (bf16_t*)calloc(SK * HD, sizeof(bf16_t));