Fix duplicate else: line in SMEM-P block

2026-05-23 09:35:47 +00:00
parent 018a961c01
commit 0b09e7e4a2
1 changed files with 0 additions and 1 deletions
--- a/dsv4/kernels/attention/fmha.py
+++ b/dsv4/kernels/attention/fmha.py
@@ -341,7 +341,6 @@ class FmhaKernel:
                    # TMEM-P: store P to TMEM via register bridge
                    cute.copy(tiled_tmem_store, rP_words, tTMEM_STOREtP)
                    cute.arch.fence_view_async_tmem_store()
-                else:
                else:
                    # SMEM-P: Use QK C-fragment layout for source (not TMEM layout)
                    # rP_bf16 uses tTMEM_LOADrS.layout (TMEM layout) causing rank mismatch