ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-01 04:07:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix issue

Browse Source
This commit is contained in:
ltqin
2026-01-29 09:46:19 +00:00
parent a9d85dfe16
commit fbae2aba20
2 changed files with 33 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
example/ck_tile/01_fmha/fmha_fwd.hpp
View File

@@ -1237,7 +1237,7 @@ auto fmha_batch_prefill_create_kargs_and_grids(fmha_batch_prefill_args args)
args.drop_seed_offset,
args.sink_ptr,
args.block_scale_size_kv,
args.block_scale_seqstart_k_ptr);
reinterpret_cast<const int32_t*>(args.block_scale_seqstart_k_ptr));
}
else
{ // create batch mode kernel arguments