Searched defs:kBlockM (Results 1 – 5 of 5) sorted by relevance
52 DEFINE_FLASH_FORWARD_KERNEL(flash_fwd_splitkv_combine_kernel, int kBlockM, int Log_max_splits, bool… in DEFINE_FLASH_FORWARD_KERNEL()141 …constexpr static int kBlockM = Kernel_traits::kHeadDim % 128 == 0 ? 4 : (Kernel_traits::kHeadDim %… in run_flash_splitkv_fwd() local166 constexpr static int kBlockM = 64; // Fixed for all head dimensions in run_mha_fwd_splitkv_dispatch() local
70 constexpr int kBlockM = Kernel_traits::kBlockM; in compute_dot_do_o() local197 constexpr int kBlockM = Kernel_traits::kBlockM; in convert_dQ() local
68 static constexpr int kBlockM = kBlockM_; member184 static constexpr int kBlockM = kBlockM_; member
92 constexpr int kBlockM = Kernel_traits::kBlockM; in compute_dq_dk_dv_1colblock() local513 …if (m_block * kBlockM < (n_block + 1) * kBlockN + binfo.actual_seqlen_q - binfo.actual_seqlen_k - … in compute_dq_dk_dv_1colblock() local
41 constexpr int kBlockM = Kernel_traits::kBlockM; in compute_attn_1rowblock() local491 constexpr int kBlockM = Kernel_traits::kBlockM; in compute_attn_1rowblock_splitkv() local