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

Clean up HD=64 debug prints, keep register-math PV check

Browse Source
This commit is contained in:
biondizzle
2026-05-28 15:15:22 +00:00
parent 5c9d471162
commit 6b9b06647a
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/unit/test_fmha_hd64_smem_p.cu
View File

@@ -126,7 +126,6 @@ test_fmha_hd64_smem_p(const bf16_t* q, const bf16_t* k, const bf16_t* v,
}
row_sum = wsum(row_sum);
if (lane == 0) for (int j=0;j<SK;j++) s_vals[j] /= row_sum;
if (lane == 0) { printf("P[0..7]: "); for(int j=0;j<8;j++) printf("%.6f ", s_vals[j]); printf("\n"); }
if (lane == 0) for (int j=0;j<SK;j++) s_p_vals[j] = s_vals[j];
}
__syncthreads();