From 18a3834fb4e5daebf24b70b6fa5e82b9ebe3d8c2 Mon Sep 17 00:00:00 2001
From: "PoYen, Chen" <PoYen.Chen@amd.com>
Date: Mon, 8 Jul 2024 10:27:32 +0000
Subject: [PATCH] Set num_splits=1 if split-kv is not supported

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

diff --git a/example/ck_tile/01_fmha/fmha_fwd.cpp b/example/ck_tile/01_fmha/fmha_fwd.cpp
index 33492765c1..0e7ac5061b 100644
--- a/example/ck_tile/01_fmha/fmha_fwd.cpp
+++ b/example/ck_tile/01_fmha/fmha_fwd.cpp
@@ -370,6 +370,7 @@ bool run(const ck_tile::ArgParser& arg_parser)
     if(num_splits != 1)
     {
         std::cerr << "split-kv is not supported. ignoring the 'num_splits' option" << std::endl;
+        num_splits = 1;
     }
 #endif