ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-15 19:44:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update to the cross_attention test/bench scripts

Browse Source
This commit is contained in:
Qianfeng Zhang
2026-05-22 14:44:14 +00:00
parent 24329f15d1
commit 9fbe96ab76
2 changed files with 22 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
example/ck_tile/18_hstu_attention/scripts/bench_cross_attention_with_sparsity.sh
View File

@@ -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

12
example/ck_tile/18_hstu_attention/scripts/test_cross_attention_with_sparsity.sh
View File

@@ -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