Skip to main content

gemv_f32_parts

logicaffeine_compile::codegen_cuda::lane::kernels

Function gemv_f32_parts 

Source 
pub fn gemv_f32_parts(n: usize, k: usize, sms: usize) -> usize
Expand description

The warps an f32 GEMV (lk_gemv_f32) cuts each output row’s k into: the most, to eight, that keep 128 values a warp and give its n outputs no more than 16 warps a streaming multiprocessor of the chip’s sms.