From d41ff70db5a599a1295449faf66bb103a0990c17 Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Fri, 26 Jul 2024 06:51:24 +0000 Subject: [PATCH] Enlarge rotary_dim limit (8 -> 16) --- 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 8b7020ebec..8b02e27322 100644 --- a/example/ck_tile/01_fmha/fmha_fwd.cpp +++ b/example/ck_tile/01_fmha/fmha_fwd.cpp @@ -409,9 +409,9 @@ bool run(const ck_tile::ArgParser& arg_parser) std::cerr << "rotary_dim should be less than or equal to head dim for q" << std::endl; return false; } - else if(!(rotary_dim % 8 == 0)) + else if(!(rotary_dim % 16 == 0)) { - std::cerr << "only rotary dimensions divisible by 8 are currently supported" << std::endl; + std::cerr << "only rotary dimensions divisible by 16 are currently supported" << std::endl; return false; }