From 82f3b3d0a0aea8a8e5c9feafdbe081eabd7e81b6 Mon Sep 17 00:00:00 2001
From: "PoYen, Chen" <PoYen.Chen@amd.com>
Date: Mon, 8 Jul 2024 06:40:18 +0000
Subject: [PATCH] Conditionally add call to fmha_fwd_splitkv()

---
 example/ck_tile/01_fmha/fmha_fwd.cpp | 2 ++
 1 file changed, 2 insertions(+)

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);
     }