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

Remove options

Browse Source
This commit is contained in:
PoYen, Chen
2024-08-16 10:22:44 +00:00
parent f2b3620511
commit 095819a387
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
example/ck_tile/01_fmha/script/smoke_test_fwd.sh
View File

@@ -7,7 +7,6 @@ KNAME=1
export CK_WARMUP=0
export CK_REPEAT=1
MODE=(0)
P_DROP=(0.0)
NUM_SPLITS=(1)
PAGE_BLOCK_SIZE=(0)
@@ -38,13 +37,12 @@ if [[ ($TEST_SPLITKV -eq 1) || ($TEST_APPENDKV -eq 1)]] ; then
NUM_SPLITS+=(2 3)
PAGE_BLOCK_SIZE+=(128)
else
MODE+=(1)
P_DROP+=(0.2)
fi
run_fp16_bf16_tests() {
for prec in "fp16" "bf16" ; do
for mode in "${MODE[@]}" ; do
for mode in 1 0 ; do
for perm in 0 1 ; do
for vlayout in "r" "c" ; do
for hdim in 32 64 128 256 ; do