Only use vllm kcache layout in group mode

example/ck_tile/01_fmha/fmha_fwd.cpp

@@ -770,9 +770,10 @@ bool run(const ck_tile::ArgParser& arg_parser)
 
     q_buf.ToDevice(q_host.data());
 
-    if(0 < page_block_size)
+    if(mode == mode_enum::group && 0 < page_block_size)
     {
-        if (!(i_perm && !is_v_rowmajor)) {
+        if(!(i_perm && !is_v_rowmajor))
+        {
             std::cerr << "make sure input layout is correct" << std::endl;
             return false;
         }

include/ck_tile/ops/fmha/kernel/fmha_fwd_splitkv_kernel.hpp

@@ -636,7 +636,7 @@ struct FmhaFwdSplitKVKernel
 
         const auto make_k_dram = [&](const KDataType* data, index_t height) {
             auto k_dram_naive = [&] {
-                if constexpr(kIsPagedKV)
+                if constexpr(kIsGroupMode && kIsPagedKV)
                 {
                     constexpr index_t vector_size = 16 / sizeof(KDataType);
                     // (hdim_q/vector_size, page_block_size, vector_size)