Remove debug print lines referencing deleted sP_2d
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user