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

revert changes in smoke test. disable v col-major fmha fwd tests in gtest

Browse Source
This commit is contained in:
aska-0096
2025-09-25 06:49:04 +00:00
parent 51953569f2
commit 02375d4cf0
4 changed files with 3 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -86,25 +86,6 @@ run_fp16_bf16_tests() {
done ; done ; done ; done ; done
done ; done ; done ; done ; done
for prec in "fp16" "bf16" ; do
for mode in 1 0 ; do
for perm in 0 1 ; do
for hdim in 96 160 192 ; do
for bias in "n" "e" "a" ; do
# run_exe -prec=$prec -mode=$mode -b=1 -h=1 -d=$hdim -s=1024 -bias=$bias -p_drop=$p_drop -lse=$lse -iperm=$perm -operm=$perm -num_splits=$num_splits -page_block_size=$page_block_size -kname=$KNAME $COMMON_ARGS
run_exe -prec=$prec -mode=$mode -b=2 -h=2 -h_k=1 -d=16 -d_v=$hdim -s=55 -s_k=256 -bias=$bias -iperm=$perm -operm=$perm -kname=$KNAME $COMMON_ARGS
run_exe -prec=$prec -mode=$mode -b=1 -h=3 -d=$hdim -s=100 -s_k=51 -bias=$bias -iperm=$perm -operm=$perm -kname=$KNAME $COMMON_ARGS
run_exe -prec=$prec -mode=$mode -b=2 -h=1 -d=16 -d_v=$hdim -s=99 -s_k=256 -bias=$bias -iperm=$perm -operm=$perm -mask=1 -kname=$KNAME $COMMON_ARGS
run_exe -prec=$prec -mode=$mode -b=1 -h=2 -h_k=1 -d=$hdim -s=1024 -s_k=256 -bias=$bias -iperm=$perm -operm=$perm -mask=2 -kname=$KNAME $COMMON_ARGS
run_exe -prec=$prec -mode=$mode -b=2 -h=1 -d=$hdim -d_v=24 -s=3 -s_k=99 -bias=$bias -iperm=$perm -operm=$perm -mask=2 -kname=$KNAME $COMMON_ARGS
run_exe -prec=$prec -mode=$mode -b=3 -h=2 -h_k=1 -d=$hdim -s=200 -s_k=520 -bias=$bias -iperm=$perm -operm=$perm -mask=t:128,30 -kname=$KNAME $COMMON_ARGS
run_exe -prec=$prec -mode=$mode -b=2 -h=1 -d=$hdim -s=99 -s_k=32 -bias=$bias -iperm=$perm -operm=$perm -mask=b:4,35 -kname=$KNAME $COMMON_ARGS
run_exe -prec=$prec -mode=$mode -b=1 -h=2 -h_k=1 -d=$hdim -s=33 -s_k=0 -bias=$bias -iperm=$perm -operm=$perm -mask=2 -kname=$KNAME $COMMON_ARGS
run_exe -prec=$prec -mode=$mode -b=1 -h=2 -h_k=1 -d=$hdim -s=1 -s_k=10 -s_kpad=32 -bias=$bias -iperm=$perm -operm=$perm -mask=2 -kname=$KNAME $COMMON_ARGS
done ; done ; done ; done ; done
}
run_fp8_tests() {

2
test/ck_tile/fmha/test_fmha_fwd_bf16.cpp
View File

@@ -32,7 +32,7 @@ const auto AppendKVHDimValues =
const auto ModeValues = Values(mode_enum::batch, mode_enum::group);
const auto IsVRowmajorValues = Values(false, true);
const auto IsVRowmajorValues = Values(true);
const bool squant = false;
const std::string init_method = "uf";

2
test/ck_tile/fmha/test_fmha_fwd_fp16.cpp
View File

@@ -32,7 +32,7 @@ const auto AppendKVHDimValues =
const auto ModeValues = Values(mode_enum::batch, mode_enum::group);
const auto IsVRowmajorValues = Values(false, true);
const auto IsVRowmajorValues = Values(true);
const bool squant = false;
const std::string init_method = "uf";

2
test/ck_tile/fmha/test_fmha_fwd_fp8.cpp
View File

@@ -26,7 +26,7 @@ const auto AppendKVHDimValues = Values(std::tuple{64, -1}, std::tuple{128, -1});
// There are no fp8 instances with seqlen padding (mode_enum::group requires it)
const auto ModeValues = Values(mode_enum::batch);
const auto IsVRowmajorValues = Values(false);
const auto IsVRowmajorValues = Values(true);
const auto squant = true;
const std::string init_method = "uf";