Skip to main content

attn_shared_padded_heads

logicaffeine_compile::codegen_cuda::lane::kernels

Function attn_shared_padded_heads 

Source 
pub fn attn_shared_padded_heads(g: usize) -> usize
Expand description

The heads lk_attention_shared pads its g heads to: whole warps of four.