Function validate_spgemm_shapes 

pub fn validate_spgemm_shapes(
    a_shape: [usize; 2],
    b_shape: [usize; 2],
) -> Result<([usize; 2], usize)>

Validate CSR matrix dimensions for SpGEMM