cublas_bf16_proj_dense

metrale_model_layers::layers::ops

Function cublas_bf16_proj_dense 

Source 
pub fn cublas_bf16_proj_dense(
    act: DevicePtr,
    weight_bf16: DevicePtr,
    out: DevicePtr,
    m: u32,
    n: u32,
    k: u32,
    stream: u64,
) -> Result<()>
Expand description

2026-09-25: Route a projection out[M,N] = act[M,K] @ weightᵀ through cuBLASLt BF16 for a weight that is already BF16 [N,K].