Searched defs:bidb (Results 1 – 3 of 3) sorted by relevance
64 const int bidb = blockIdx.y; in compute_dot_do_o() local147 const int bidb = blockIdx.y; in clear_dKVaccum() local191 const int bidb = blockIdx.y; in convert_dQ() local279 const int bidb = blockIdx.y; in convert_dKV() local
80 inline __device__ void compute_dq_dk_dv_1colblock(const Params ¶ms, const int bidb, const int b… in compute_dq_dk_dv_1colblock()791 const int bidb = blockIdx.x; in compute_dq_dk_dv() local818 const int bidb = blockIdx.y; in compute_dq_dk_dv_seqk_parallel() local
29 inline __device__ void compute_attn_1rowblock(const Params ¶ms, const int bidb, const int bidh,… in compute_attn_1rowblock()479 inline __device__ void compute_attn_1rowblock_splitkv(const Params ¶ms, const int bidb, const i… in compute_attn_1rowblock_splitkv()1049 const int bidb = blockIdx.y; in compute_attn() local1070 const int bidb = Split ? blockIdx.z / params.h : blockIdx.y; in compute_attn_splitkv() local