From e83c3c7fa03e0396ba400d3a3ff84f252685f517 Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Tue, 16 Jul 2024 06:54:37 +0000 Subject: [PATCH] Add constraint to the rotary_dim option --- example/ck_tile/01_fmha/fmha_fwd.cpp | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/example/ck_tile/01_fmha/fmha_fwd.cpp b/example/ck_tile/01_fmha/fmha_fwd.cpp index 7672b405d8..b6054f6285 100644 --- a/example/ck_tile/01_fmha/fmha_fwd.cpp +++ b/example/ck_tile/01_fmha/fmha_fwd.cpp @@ -400,6 +400,11 @@ 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)) + { + std::cerr << "only rotary dimensions divisible by 8 are currently supported" << std::endl; + return false; + } const bool is_rotary_interleaved = arg_parser.get_bool("rotary_interleaved");