From d52278a5efbaff8f412f33e591976ac4061cb049 Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Fri, 16 Aug 2024 18:23:55 +0000 Subject: [PATCH] Add more case for appendkv --- example/ck_tile/01_fmha/script/smoke_test_fwd.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 6d844e7bad..54dc106090 100755 --- a/example/ck_tile/01_fmha/script/smoke_test_fwd.sh +++ b/example/ck_tile/01_fmha/script/smoke_test_fwd.sh @@ -82,7 +82,7 @@ run_fp8_tests() { run_fp16_appendkv_tests() { for s in $(seq 63 1 65) ; do for s_k in 65 129 ; do - for s_knew in 64 $s_k ; do + for s_knew in 0 64 $s_k ; do for hdim in 32 64 128 256 ; do for ri in 0 1 ; do for rdim in 0 16 32 $hdim ; do