shard_gdn_conv_rows

Function shard_gdn_conv_rows 

Source
pub fn shard_gdn_conv_rows(
    src: DevicePtr,
    dims: &TpGdnDims,
    d_conv: usize,
    gpu: &dyn GpuBackend,
) -> Result<(DevicePtr, usize, usize)>
Expand description

2026-09-25: Shard the depthwise conv1d BF16 weight [full_conv_dim, d_conv] to [local_conv_dim, d_conv]. Its channels are the QKV channels (one filter per channel), so it uses the same [Q|K|V] segments as the QKV in-projection.