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

test: disable tmem epilogue test (debugging reference hang)

Browse Source
This commit is contained in:
biondizzle
2026-05-28 07:38:47 +00:00
parent 224d7e24c6
commit 44fb04fa1f
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_sm100_standalone.cu
View File

@@ -155,7 +155,7 @@ int main() {
cudaMemcpy(dv,hvb,B*HD*sk*2,cudaMemcpyHostToDevice);
all_pass &= test_kernel("reference", HD, sk, scale, dq,dk,dv,do_,d_lse,ho_ref,B,H);
all_pass &= test_kernel("tmem_epilogue", HD, sk, scale, dq,dk,dv,do_,d_lse,ho_ref,B,H);
// all_pass &= test_kernel("tmem_epilogue", HD, sk, scale, dq,dk,dv,do_,d_lse,ho_ref,B,H);
cudaFree(dq);cudaFree(dk);cudaFree(dv);cudaFree(do_);cudaFree(d_lse);
free(hq);free(hk);free(hv);free(ho_ref);free(hqb);free(hkb);free(hvb);