gpu_concat_rows

metrale_model_layers::weight_map::fp8_lut

Function gpu_concat_rows 

Source 
pub fn gpu_concat_rows(
    a: &DenseWeight,
    a_rows: usize,
    b: &DenseWeight,
    b_rows: usize,
    k: usize,
    gpu: &dyn GpuBackend,
) -> Result<DenseWeight>
Expand description

2026-09-25: GPU-concatenate two BF16 matrices row-wise: [A; B] → [A_rows + B_rows, K], in a new buffer. Both inputs must be contiguous with the same K.