Searched defs:num_n_blocks (Results 1 – 2 of 2) sorted by relevance
262 inline int num_splits_heuristic(int batch_nheads_mblocks, int num_SMs, int num_n_blocks, int max_sp… in num_splits_heuristic()304 const int num_n_blocks = (max_seqlen_k + block_n - 1) / block_n; in set_params_splitkv() local
1567 const int num_n_blocks = (right_dim1 + NR - 1) / NR; in Compute() local