update for new ck interface

ROCm · Nov 26, 2024 · af935b9 · af935b9
1 parent 0538b43
commit af935b9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/csrc/flash_attn_ck/mha_varlen_fwd.cpp b/csrc/flash_attn_ck/mha_varlen_fwd.cpp
@@ -217,6 +217,7 @@ fmha_fwd_splitkv_args get_ck_fmha_varlen_fwd_splitkv_args(bool has_lse,
         args.page_block_size = 0;
     }
 
+    args.is_gappy = false;
     args.cache_batch_idx = nullptr;
 
     args.seqstart_q_ptr = seqlens_q.data_ptr();