From 2b3435f97cb5c1460c6226eb21bc9d1aa0e9fd12 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sun, 24 May 2026 04:06:26 +0000 Subject: [PATCH] D1: Remove debug prints, clean up --- dsv4/kernels/attention/fmha.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/dsv4/kernels/attention/fmha.py b/dsv4/kernels/attention/fmha.py index f3d39635..fc1442a5 100644 --- a/dsv4/kernels/attention/fmha.py +++ b/dsv4/kernels/attention/fmha.py @@ -299,12 +299,6 @@ class FmhaKernel: # sP_addr = 64*m + (k%16) + 16*((k//16)%4) + 8192*(k//64) # This is representable as a CuTe layout: (128, (16, 4, 2)) -> (64, (1, 16, 8192)) _sP_nostage = sP[(None, None, None, 0)] # remove stage dim - # Debug: print key shapes at trace time - print(f"SMEM-P: sP shape={cute.shape(sP)}, sP_nostage shape={cute.shape(_sP_nostage)}") - print(f"QK: tCrQ mode2 size={cute.size(tCrQ, mode=[2])}, tCrK mode2 size={cute.size(tCrK, mode=[2])}") - print(f"QK: tOrP0 mode2 size={cute.size(tOrP0, mode=[2])}") - print(f"PV: tCrP mode2 size={cute.size(tCrP, mode=[2])}") - print(f"PV: pv_n_tile={self.pv_n_tile}, n_pv_tiles={self.n_pv_tiles}") row_max = -Float32.inf row_sum = Float32(0.0)