From 2126d4d88db1a3ad4b6f70aada76e58ecb01e175 Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Wed, 24 Jul 2024 06:35:53 +0000 Subject: [PATCH] Add append-kv smoke tests --- .../ck_tile/01_fmha/script/smoke_test_fwd.sh | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) 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 779e8d09ee..b2c7149d2f 100755 --- a/example/ck_tile/01_fmha/script/smoke_test_fwd.sh +++ b/example/ck_tile/01_fmha/script/smoke_test_fwd.sh @@ -6,6 +6,7 @@ KNAME=1 export CK_WARMUP=0 export CK_REPEAT=1 +ENABLE_APPENDKV_TEST=0 COMMON_ARGS='-v=1 -warmup=0 -repeat=1' # mode=0 @@ -51,3 +52,21 @@ done done done set +x + +if [ $ENABLE_APPENDKV_TEST -eq 1 ]; then +for mode in 1 0 ; do +for s in $(seq 63 1 65) ; do +for s_k in 65 127 128 129 ; do +for s_k_new in 0 63 64 65 ; do +for ri in 0 1 ; do +for rdim in $(seq 0 8 16) ; do + +$EXE -prec="fp16" -mode=$mode -b=3 -h=2 -d=128 -s=$s -s_k=$s_k -s_k_new=$s_k_new -rotary_dim=$rdim -rotary_interleaved=$ri -iperm=1 -operm=1 -kname=1 $COMMON_ARGS + +done +done +done +done +done +done +fi \ No newline at end of file