Searched defs:bidh (Results 1 – 3 of 3) sorted by relevance
66 const int bidh = blockIdx.z; in compute_dot_do_o() local149 const int bidh = blockIdx.z; in clear_dKVaccum() local193 const int bidh = blockIdx.z; in convert_dQ() local281 const int bidh = blockIdx.z; in convert_dKV() local
80 …oid compute_dq_dk_dv_1colblock(const Params ¶ms, const int bidb, const int bidh, const int n_b… in compute_dq_dk_dv_1colblock()794 const int bidh = blockIdx.y; in compute_dq_dk_dv() local820 const int bidh = blockIdx.z; 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 …compute_attn_1rowblock_splitkv(const Params ¶ms, const int bidb, const int bidh, const int m_b… in compute_attn_1rowblock_splitkv()1051 const int bidh = blockIdx.z; in compute_attn() local1072 const int bidh = Split ? blockIdx.z - bidb * params.h : blockIdx.z; in compute_attn_splitkv() local