From ffafd47d0774293b799ec2a62aa181f45976cd19 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 23 May 2026 09:34:09 +0000 Subject: [PATCH] Remove debug print lines referencing deleted sP_2d --- dsv4/kernels/attention/fmha.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/dsv4/kernels/attention/fmha.py b/dsv4/kernels/attention/fmha.py index d21ed3c1..b4657739 100644 --- a/dsv4/kernels/attention/fmha.py +++ b/dsv4/kernels/attention/fmha.py @@ -268,8 +268,6 @@ class FmhaKernel: ) tiled_smem_copy = cute.make_tiled_copy_C(smem_copy_atom, qk_mma) thr_smem_copy = tiled_smem_copy.get_slice(sfw_idx) - print(f"[SMEM-P DEBUG] sP shape: {cute.shape(sP)}") - print(f"[SMEM-P DEBUG] sP_2d shape: {cute.shape(sP_2d)}") # flatten to 2D for copy tSMEM_CPYsP = thr_smem_copy.partition_D(sP_2d) # destination (SMEM) row_max = -Float32.inf