Function ssm_rowwise_w_bf16_bytes_for 

pub fn ssm_rowwise_w_bf16_bytes_for(
    config: &ModelConfig,
    rowwise_enabled: bool,
) -> usize

ssm_rowwise_w_bf16_bytes with the lever passed in, so the sizing arithmetic is pinnable at exact integers without touching the process environment.