debug: check full HD for chunk1 test

This commit is contained in:
2026-05-30 07:00:46 +00:00
parent 3dbd3c5e7f
commit dcf89fdd1c

View File

@@ -180,7 +180,7 @@ static int test_single(int T, int s_k, int n_h = 1, int batch = 1) {
o_ref, nullptr, HD, T, s_k, SCALE);
float cs = 0, na = 0, nb = 0;
int check_hd = HD_CHUNK; // Only check first hd_chunk values (for partial debug)
int check_hd = HD; // Check full HD (all chunks)
for (int t = 0; t < T; t++) {
for (int d = 0; d < check_hd; d++) {
float a = bf16_to_f32_host(h_o[h * MAX_T * HD + t * HD + d]);