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 025bd087a5..b8478fcd9d 100644 --- a/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp +++ b/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp @@ -1797,7 +1797,7 @@ struct FmhaFwdKernel o_acc_element_func, // o_acc_element_func mask, position_encoding, - scale_s, + kargs.scale_s, variant, variant_params, block_indices,