From efb854927966ca8ce605daa230a612aa3cc38ebf Mon Sep 17 00:00:00 2001
From: aska-0096 <haocwang@amd.com>
Date: Fri, 8 Aug 2025 17:53:19 +0000
Subject: [PATCH] fix bug

---
 .../fmha/pipeline/block_fmha_bwd_pipeline_default_policy.hpp  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/include/ck_tile/ops/fmha/pipeline/block_fmha_bwd_pipeline_default_policy.hpp b/include/ck_tile/ops/fmha/pipeline/block_fmha_bwd_pipeline_default_policy.hpp
index 5519f4644e..8a78f304e4 100644
--- a/include/ck_tile/ops/fmha/pipeline/block_fmha_bwd_pipeline_default_policy.hpp
+++ b/include/ck_tile/ops/fmha/pipeline/block_fmha_bwd_pipeline_default_policy.hpp
@@ -1708,7 +1708,7 @@ struct BlockFmhaBwdPipelineDefaultPolicy
                         merge_sequences(sequence<1, 1>{}, c_warp_y_lengths));
 
                     pt_out.set_y_sliced_thread_data(
-                        merge_sequences(sequence<mIter, kIter>{}, a_warp_y_index_zeros),
+                        merge_sequences(sequence<kIter, mIter>{}, a_warp_y_index_zeros),
                         merge_sequences(sequence<1, 1>{}, a_warp_y_lengths),
                         pt_warp_tensor.get_thread_buffer());
                 });
@@ -1758,7 +1758,7 @@ struct BlockFmhaBwdPipelineDefaultPolicy
                         merge_sequences(sequence<1, 1>{}, c_warp_y_lengths));
 
                     dst_out.set_y_sliced_thread_data(
-                        merge_sequences(sequence<mIter, kIter>{}, a_warp_y_index_zeros),
+                        merge_sequences(sequence<kIter, mIter>{}, a_warp_y_index_zeros),
                         merge_sequences(sequence<1, 1>{}, a_warp_y_lengths),
                         dst_warp_tensor.get_thread_buffer());
                 });