diff --git a/example/ck_tile/01_fmha/CMakeLists.txt b/example/ck_tile/01_fmha/CMakeLists.txt index 75c05f3b20..cbad6acda3 100644 --- a/example/ck_tile/01_fmha/CMakeLists.txt +++ b/example/ck_tile/01_fmha/CMakeLists.txt @@ -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 diff --git a/script/cmake-ck-dev.sh b/script/cmake-ck-dev.sh index 7cf9f5affe..9a76569d24 100755 --- a/script/cmake-ck-dev.sh +++ b/script/cmake-ck-dev.sh @@ -41,5 +41,5 @@ cmake -D CMAKE_VERBOSE_MAKEFILE:BOOL=ON \ -D USE_BITINT_EXTENSION_INT4=OFF \ -G Ninja \ -$REST_ARGS \ +$@ \ ${MY_PROJECT_SOURCE}