From 4db9e47cd55c54b296455cc97439b31d37552652 Mon Sep 17 00:00:00 2001 From: Haocong WANG Date: Tue, 16 Sep 2025 15:07:10 +0800 Subject: [PATCH] [CK_TILE] fix bug when iperm =0 in fmha fwd (#2820) * fix bug when iperm =0 in fmha fwd * Disable f8 fmha smoke test until fix pr merged --------- Co-authored-by: Po Yen Chen [ROCm/composable_kernel commit: 59cb9064821de461c237736ae61691e96a07572d] --- example/ck_tile/01_fmha/script/smoke_test_fwd.sh | 2 +- include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/example/ck_tile/01_fmha/script/smoke_test_fwd.sh b/example/ck_tile/01_fmha/script/smoke_test_fwd.sh index 3913a0d5c2..dda3943454 100755 --- a/example/ck_tile/01_fmha/script/smoke_test_fwd.sh +++ b/example/ck_tile/01_fmha/script/smoke_test_fwd.sh @@ -97,7 +97,7 @@ run_fp16_appendkv_tests() { set -x run_fp16_bf16_tests -run_fp8_tests +# run_fp8_tests if [ $TEST_APPENDKV -eq 1 ] ; then run_fp16_appendkv_tests diff --git a/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp b/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp index 9d848dfd7a..6405ca50df 100644 --- a/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp +++ b/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp @@ -1868,7 +1868,7 @@ struct FmhaFwdKernel const auto v_dram_naive = make_naive_tensor_view( data, // will update this pointer if using paged-kvcache make_tuple(length, kargs.hdim_v), - make_tuple(kargs.hdim_v, 1), + make_tuple(kargs.stride_v, 1), number{}, number<1>{});