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