From 02375d4cf021dadf60893f01d43c6dc43329874e Mon Sep 17 00:00:00 2001 From: aska-0096 Date: Thu, 25 Sep 2025 06:49:04 +0000 Subject: [PATCH] revert changes in smoke test. disable v col-major fmha fwd tests in gtest --- .../ck_tile/01_fmha/script/smoke_test_fwd.sh | 19 ------------------- test/ck_tile/fmha/test_fmha_fwd_bf16.cpp | 2 +- test/ck_tile/fmha/test_fmha_fwd_fp16.cpp | 2 +- test/ck_tile/fmha/test_fmha_fwd_fp8.cpp | 2 +- 4 files changed, 3 insertions(+), 22 deletions(-) diff --git a/example/ck_tile/01_fmha/script/smoke_test_fwd.sh b/example/ck_tile/01_fmha/script/smoke_test_fwd.sh index 322a2e3ad5..afd0c728c6 100755 --- a/example/ck_tile/01_fmha/script/smoke_test_fwd.sh +++ b/example/ck_tile/01_fmha/script/smoke_test_fwd.sh @@ -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() { diff --git a/test/ck_tile/fmha/test_fmha_fwd_bf16.cpp b/test/ck_tile/fmha/test_fmha_fwd_bf16.cpp index fbc6449a6a..30bfbb2500 100644 --- a/test/ck_tile/fmha/test_fmha_fwd_bf16.cpp +++ b/test/ck_tile/fmha/test_fmha_fwd_bf16.cpp @@ -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"; diff --git a/test/ck_tile/fmha/test_fmha_fwd_fp16.cpp b/test/ck_tile/fmha/test_fmha_fwd_fp16.cpp index abc2c44726..ed4473497d 100644 --- a/test/ck_tile/fmha/test_fmha_fwd_fp16.cpp +++ b/test/ck_tile/fmha/test_fmha_fwd_fp16.cpp @@ -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"; diff --git a/test/ck_tile/fmha/test_fmha_fwd_fp8.cpp b/test/ck_tile/fmha/test_fmha_fwd_fp8.cpp index b99c304d1f..30e0b0d00d 100644 --- a/test/ck_tile/fmha/test_fmha_fwd_fp8.cpp +++ b/test/ck_tile/fmha/test_fmha_fwd_fp8.cpp @@ -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";