From 0fb487ad6329cdf0254c2ddf47a21c8ee22cf1fd Mon Sep 17 00:00:00 2001 From: rocking Date: Tue, 29 Jul 2025 13:44:10 +0800 Subject: [PATCH] set default optdim (#2580) [ROCm/composable_kernel commit: 01642ca8b15ffed4e15e0b9f58d7b0fb19b6f2b7] --- example/ck_tile/01_fmha/CMakeLists.txt | 2 ++ 1 file changed, 2 insertions(+) diff --git a/example/ck_tile/01_fmha/CMakeLists.txt b/example/ck_tile/01_fmha/CMakeLists.txt index 1b004ec100..bd03aee924 100644 --- a/example/ck_tile/01_fmha/CMakeLists.txt +++ b/example/ck_tile/01_fmha/CMakeLists.txt @@ -28,12 +28,14 @@ string(REPLACE ";" "," FMHA_FWD_APIS "${FMHA_FWD_ENABLE_APIS}") set(FMHA_FWD_CODE_GEN_COMMON_ARGS ${CMAKE_CURRENT_LIST_DIR}/generate.py --api ${FMHA_FWD_APIS} + --optdim 32,64,128,256 # --filter fmha_fwd... ) 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... )