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

bwd filter

Browse Source
This commit is contained in:
Ding, Yi
2025-08-18 02:07:03 +00:00
parent 75c4b9372f
commit b85daba2a3
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
example/ck_tile/01_fmha/CMakeLists.txt
View File

@@ -35,8 +35,8 @@ set(FMHA_BWD_CODE_GEN_COMMON_ARGS
${CMAKE_CURRENT_LIST_DIR}/generate.py
--api bwd
--receipt 3
--optdim 32,64,128,256
# --filter fmha_bwd_dot...@fmha_bwd_convert...@fmha_bwd...
--optdim 128
--filter fmha_bwd_dot_do_o_d128_bf16_batch_o2_ps@*@fmha_bwd_d128_bf16_batch_*_npad_nbias_ndbias_nmask_ndropout_ndeterministic_trload
)
# generate a list of kernels, but not actually emit files at config sta