Searched defs:n_block (Results 1 – 8 of 8) sorted by relevance
113 unsigned int n_block = args._cfg->outer_block_size; in compute_n_block() local135 unsigned int n_block = (scaled_l2_size - k_block_area) / (sizeof(Toi) * k_block); in compute_n_block() local
100 unsigned int n_block = args._cfg->outer_block_size; in compute_n_block() local
120 …unsigned int n_block = (((L2_size * 9) / 10) - (k_block * sizeof(Toi) * (strategy::out_width() + s… in compute_n_block() local
97 unsigned int n_block=0; member in arm_gemm::GemvPretransposed
367 unsigned int n_block = iceildiv(args._Nsize, columns_needed); local
145 const int n_block = blockIdx.x; in clear_dKVaccum() local277 const int n_block = blockIdx.x; in convert_dKV() local
80 …mpute_dq_dk_dv_1colblock(const Params ¶ms, const int bidb, const int bidh, const int n_block) { in compute_dq_dk_dv_1colblock()805 for (int n_block = n_block_max - 2; n_block > 0; n_block--) { in compute_dq_dk_dv() local823 …for (int n_block = blockIdx.x; n_block < (params.seqlen_k + Kernel_traits::kBlockN - 1) / Kernel_t… in compute_dq_dk_dv_seqk_parallel() local
249 int n_block = n_block_max - 1; in compute_attn_1rowblock() local712 for (int n_block = n_block_max - 1; n_block >= n_block_copy_min; n_block--) { in compute_attn_1rowblock_splitkv() local802 int n_block = n_block_max - 1; in compute_attn_1rowblock_splitkv() local