diff --git a/example/ck_tile/01_fmha/fmha_fwd.cpp b/example/ck_tile/01_fmha/fmha_fwd.cpp
index 28f7905734..7553ed0aaa 100644
--- a/example/ck_tile/01_fmha/fmha_fwd.cpp
+++ b/example/ck_tile/01_fmha/fmha_fwd.cpp
@@ -248,11 +248,13 @@ float fmha_fwd_dispatch(fmha_fwd_traits traits,
                         fmha_fwd_args args,
                         const ck_tile::stream_config& config)
 {
+#if CK_TILE_FMHA_FWD_SPLITKV_API
     if(1 < args.num_splits)
     {
         return fmha_fwd_splitkv(traits, args, config);
     }
     else
+#endif
     {
         return fmha_fwd(traits, args, config);
     }