From b6c2f2f01dbcf83a4aa5d1ddcb55e57eb8ae9b6a Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Wed, 7 Aug 2024 15:22:57 +0000 Subject: [PATCH] Add missing group mode argument --- example/ck_tile/01_fmha/fmha_fwd.hpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/example/ck_tile/01_fmha/fmha_fwd.hpp b/example/ck_tile/01_fmha/fmha_fwd.hpp index cd05efa85b..8dbef4491e 100644 --- a/example/ck_tile/01_fmha/fmha_fwd.hpp +++ b/example/ck_tile/01_fmha/fmha_fwd.hpp @@ -515,7 +515,9 @@ auto fmha_fwd_appendkv_create_kargs_and_grids(fmha_fwd_appendkv_args args) args.nhead_stride_knew, args.nhead_stride_v, args.nhead_stride_vnew, + args.batch_stride_k, args.batch_stride_knew, + args.batch_stride_v, args.batch_stride_vnew); } else