Searched defs:sK (Results 1 – 2 of 2) sorted by relevance
145 Tensor sK = make_tensor(sQ.data() + (Kernel_traits::Share_Q_K_smem ? 0 : size(sQ)), in compute_attn_1rowblock() local592 Tensor sK = make_tensor(sQ.data() + size(sQ), typename Kernel_traits::SmemLayoutKV{}); in compute_attn_1rowblock_splitkv() local
163 Tensor sK = make_tensor(sdO.data() + size(sdO), typename Kernel_traits::SmemLayoutKV{}); in compute_dq_dk_dv_1colblock() local