From 9fbe96ab76423633c18c463272eb8727b645d2cd Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Fri, 22 May 2026 14:44:14 +0000 Subject: [PATCH] Update to the cross_attention test/bench scripts --- .../scripts/bench_cross_attention_with_sparsity.sh | 12 +++++++++++- .../scripts/test_cross_attention_with_sparsity.sh | 12 +++++++++++- 2 files changed, 22 insertions(+), 2 deletions(-) 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