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

Fix constexpr issues in prefill debug test

Browse Source
This commit is contained in:
biondizzle
2026-06-03 03:46:29 +00:00
parent d3dc8cf901
commit 09384a637a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/unit/test_prefill_t2_debug.cu
View File

@@ -332,7 +332,7 @@ __global__ void prefill_t2_debug_kernel(
// Read PV result for row qr
asm volatile("fence.sc.gpu;" ::: "memory");
__syncthreads();
prefill_read_pv_all_subs<HD, N_SUB>(tb, qr, sOacc, p_rescale);
prefill_read_pv_all_subs<512, 32>(tb, qr, sOacc, p_rescale);
__syncthreads();
// Print first few accumulated values
@@ -378,7 +378,7 @@ int main() {
constexpr int HD = 512;
constexpr int NOPE = 448;
constexpr int ROPE = 64;
constexpr float scale = 1.0f / sqrtf((float)HD);
const float scale = 1.0f / sqrtf((float)HD);
printf("=== Prefill T=2 Debug Test ===\n");
printf("T=%d N=%d HD=%d NOPE=%d ROPE=%d scale=%.6f\n", T, N, HD, NOPE, ROPE, scale);