diff --git a/example/ck_tile/18_hstu_attention/scripts/bench_cross_attention_with_sparsity.sh b/example/ck_tile/18_hstu_attention/scripts/bench_cross_attention_with_sparsity.sh
index d49d06324a..fbb57d63a9 100755
--- a/example/ck_tile/18_hstu_attention/scripts/bench_cross_attention_with_sparsity.sh
+++ b/example/ck_tile/18_hstu_attention/scripts/bench_cross_attention_with_sparsity.sh
@@ -1,7 +1,17 @@
 #!/bin/bash
 
 BUILD=build
-EXE="$BUILD/bin/tile_example_hstu_attention -v=0 -perf=1"
+
+USE_SOFTMAX=0
+if [ $# -ge 1 ]; then
+    USE_SOFTMAX=$1
+fi
+
+if [ $USE_SOFTMAX -eq 1 ]; then
+    EXE="$BUILD/bin/tile_example_hstu_attention -v=0 -perf=1 -softmax=1"
+else
+    EXE="$BUILD/bin/tile_example_hstu_attention -v=0 -perf=1"
+fi
 
 set -x 
 
diff --git a/example/ck_tile/18_hstu_attention/scripts/test_cross_attention_with_sparsity.sh b/example/ck_tile/18_hstu_attention/scripts/test_cross_attention_with_sparsity.sh
index 06058f4921..ddbd325896 100755
--- a/example/ck_tile/18_hstu_attention/scripts/test_cross_attention_with_sparsity.sh
+++ b/example/ck_tile/18_hstu_attention/scripts/test_cross_attention_with_sparsity.sh
@@ -1,7 +1,17 @@
 #!/bin/bash
 
 BUILD=build
-EXE="$BUILD/bin/tile_example_hstu_attention -v=1"
+
+USE_SOFTMAX=0
+if [ $# -ge 1 ]; then
+    USE_SOFTMAX=$1
+fi
+
+if [ $USE_SOFTMAX -eq 1 ]; then
+    EXE="$BUILD/bin/tile_example_hstu_attention -v=1 -softmax=1"
+else
+    EXE="$BUILD/bin/tile_example_hstu_attention -v=1"
+fi
 
 set -x