From 12da00c3be029ddb2c484e339d3729515b8f616f Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Tue, 6 Aug 2024 03:20:29 +0000 Subject: [PATCH] Use 128 as minimus page_block_size --- example/ck_tile/01_fmha/fmha_fwd.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/example/ck_tile/01_fmha/fmha_fwd.cpp b/example/ck_tile/01_fmha/fmha_fwd.cpp index bd6c36603a..89a7857c23 100644 --- a/example/ck_tile/01_fmha/fmha_fwd.cpp +++ b/example/ck_tile/01_fmha/fmha_fwd.cpp @@ -441,9 +441,9 @@ bool run(const ck_tile::ArgParser& arg_parser) page_block_size = 0; } #endif - if(!(page_block_size % 256 == 0)) + if(!(page_block_size % 128 == 0)) { - std::cerr << "only paged-kvcache block size divisible by 256 are currently supported" + std::cerr << "only paged-kvcache block size divisible by 128 are currently supported" << std::endl; return false; }