pub fn dense_gemm_m16_bf16_n64(
gpu: &dyn GpuBackend,
kernel: KernelHandle,
input: DevicePtr,
weight: &DenseWeight,
output: DevicePtr,
m: u32,
n: u32,
k: u32,
a_row_stride: u32,
c_row_stride: u32,
stream: u64,
) -> Result<()>Expand description
N_TILE=64 twin of dense_gemm_m16_bf16 — identical arguments and
identical per-output arithmetic, ceil(n/64) CTAs instead of ceil(n/32).