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

Add softmax selection to two of the testing scripts

Browse Source
This commit is contained in:
Qianfeng Zhang
2026-02-05 14:56:31 +00:00
parent 0a8c5f523a
commit 8af5e26717
2 changed files with 22 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
example/ck_tile/18_hstu_attention/scripts/test_jagged_causal_mattn0_full0.sh
View File

@@ -1,14 +1,20 @@
#!/bin/bash
ndist=0
set +x
ndist=0
BUILD=build
USE_SOFTMAX=0
if [ $# -ge 1 ]; then
ndist=$1
USE_SOFTMAX=$1
fi
set +x
BUILD=build
EXE=$BUILD/bin/tile_example_hstu_attention
if [ $USE_SOFTMAX -eq 1 ]; then
EXE="$BUILD/bin/tile_example_hstu_attention -softmax=1"
else
EXE="$BUILD/bin/tile_example_hstu_attention"
fi
dtype="bf16"

16
example/ck_tile/18_hstu_attention/scripts/test_jagged_causal_mattn256_full256.sh
View File

@@ -1,14 +1,20 @@
#!/bin/bash
ndist=0
set +x
ndist=0
BUILD=build
USE_SOFTMAX=0
if [ $# -ge 1 ]; then
ndist=$1
USE_SOFTMAX=$1
fi
set +x
BUILD=build
EXE=$BUILD/bin/tile_example_hstu_attention
if [ $USE_SOFTMAX -eq 1 ]; then
EXE="$BUILD/bin/tile_example_hstu_attention -softmax=1"
else
EXE="$BUILD/bin/tile_example_hstu_attention"
fi
dtype="bf16"