Only check incomplete split in first&last iterations

include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_qr_ks_vs_async.hpp

@@ -473,21 +473,24 @@ struct BlockFmhaFwdSplitKVPipelineQRKSVSAsync
             }
             move_tile_window(bias_dram_window, {0, kN0});
 
-            /// TODO: only check in first/last iteration without increasing code size
+            // only check in first/last iterations
             if constexpr(kHasUnevenSplits)
+            {
+                if(1 < num_splits && (i_total_loops == 0 || i_total_loops == num_total_loop - 1))
                 {
                     const auto k_origin = k_page_block_navigator.to_global_window_origin(
                         i_page_block_k, k_dram_block_window.get_window_origin());
-                set_tile_if(
-                    s_acc,
+                    set_tile_if(s_acc,
                                 -numeric<SMPLComputeDataType>::infinity(),
                                 [&,
                                  physical_seqlen_k_start_ = physical_seqlen_k_start,
                                  physical_seqlen_k_end_   = physical_seqlen_k_end](auto tile_idx) {
-                        const auto col = k_origin.at(number<0>{}) + tile_idx.at(number<1>{});
+                                    const auto col =
+                                        k_origin.at(number<0>{}) + tile_idx.at(number<1>{});
                                     if constexpr(kIsPagedKV)
                                     {
-                            return col < physical_seqlen_k_start_ || physical_seqlen_k_end_ <= col;
+                                        return col < physical_seqlen_k_start_ ||
+                                               physical_seqlen_k_end_ <= col;
                                     }
                                     else
                                     {
@@ -495,6 +498,7 @@ struct BlockFmhaFwdSplitKVPipelineQRKSVSAsync
                                     }
                                 });
                 }
+            }
 
             if constexpr(kPadSeqLenK || FmhaMask::IsMasking)
             {