Skip to main content

axpy_pitch

logicaffeine_compile::codegen_cuda::lane::kernels

Function axpy_pitch 

Source 
pub fn axpy_pitch(ty: GgmlType) -> usize
Expand description

The bytes of a row the staged per-head matmul holds in shared memory (lk_axpy_pitch): the units of AXPY_W sub-blocks and one more, 15 bytes of alignment slack, rounded to an odd number of words.