Searched defs:kBlockN (Results 1 – 5 of 5) sorted by relevance
153 constexpr int kBlockN = Kernel_traits::kBlockN; in clear_dKVaccum() local285 constexpr int kBlockN = Kernel_traits::kBlockN; in convert_dKV() local
69 static constexpr int kBlockN = kBlockN_; member185 static constexpr int kBlockN = kBlockN_; member
170 constexpr static int kBlockN = Headdim <= 64 ? 256 : (Headdim <= 128 ? 128 : 64); in run_mha_fwd_splitkv_dispatch() local
42 constexpr int kBlockN = Kernel_traits::kBlockN; in compute_attn_1rowblock() local492 constexpr int kBlockN = Kernel_traits::kBlockN; in compute_attn_1rowblock_splitkv() local1182 constexpr int kBlockN = kNThreads / kBlockM; in combine_attn_seqk_parallel() local
93 constexpr int kBlockN = Kernel_traits::kBlockN; in compute_dq_dk_dv_1colblock() local