From d099819657e253a2fdbcfa2a6b91f27ad98fc10e Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Fri, 24 Apr 2026 07:36:41 +0000 Subject: [PATCH] Renaming the test_hstu_attention_seqlen_kv.sh to test_hstu_cross_attention.sh --- ...ion_seqlen_kv.sh => test_hstu_cross_attention.sh} | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) rename example/ck_tile/18_hstu_attention/scripts/{test_hstu_attention_seqlen_kv.sh => test_hstu_cross_attention.sh} (94%) diff --git a/example/ck_tile/18_hstu_attention/scripts/test_hstu_attention_seqlen_kv.sh b/example/ck_tile/18_hstu_attention/scripts/test_hstu_cross_attention.sh similarity index 94% rename from example/ck_tile/18_hstu_attention/scripts/test_hstu_attention_seqlen_kv.sh rename to example/ck_tile/18_hstu_attention/scripts/test_hstu_cross_attention.sh index d3bdca3808..647b72c074 100644 --- a/example/ck_tile/18_hstu_attention/scripts/test_hstu_attention_seqlen_kv.sh +++ b/example/ck_tile/18_hstu_attention/scripts/test_hstu_cross_attention.sh @@ -1,7 +1,17 @@ #!/bin/bash BUILD=build -EXE="$BUILD/bin/tile_example_hstu_attention" + +USE_SOFTMAX=0 +if [ $# -ge 1 ]; then + USE_SOFTMAX=$1 +fi + +if [ $USE_SOFTMAX -eq 1 ]; then + EXE="$BUILD/bin/tile_example_hstu_attention -softmax=1" +else + EXE="$BUILD/bin/tile_example_hstu_attention" +fi for T in "fp16" "bf16"; do set -x