ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 10:09:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Renaming the test_hstu_attention_seqlen_kv.sh to test_hstu_cross_attention.sh

Browse Source
This commit is contained in:
Qianfeng Zhang
2026-04-24 07:36:41 +00:00
parent 0b6bbe45d6
commit d099819657
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
example/ck_tile/18_hstu_attention/scripts/test_hstu_attention_seqlen_kv.sh → example/ck_tile/18_hstu_attention/scripts/test_hstu_cross_attention.sh
View File

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