Function test_gemm

ciphercore-base/src/evaluators/simple_evaluator.rs:3476–3554

()

    #[test]
    fn test_gemm() {
        || -> Result<()> {
            gemm_helper(
                array_type(vec![2, 3], UINT32),
                array_type(vec![3, 3], UINT32),
                array!([[1, 2, 3], [4, 5, 6]]).into_raw_vec(),
                array!([[7, 8, 9], [10, 11, 12], [13, 14, 15]]).into_raw_vec(),
                array!([[50, 68, 86], [122, 167, 212]]).into_raw_vec(),
            )?;
            gemm_helper(
                array_type(vec![2, 2, 2], UINT32),
                array_type(vec![2, 3, 2], UINT32),
                vec![1, 2, 3, 4, 5, 6, 7, 8],
                vec![10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120],
                vec![
                    50, 110, 170, 110, 250, 390, 830, 1050, 1270, 1130, 1430, 1730,
                ],
            )?;

            gemm_helper(
                array_type(vec![2, 127], BIT),
                array_type(vec![3, 127], BIT),
                vec![1; 2 * 127],
                vec![1; 3 * 127],
                vec![1, 1, 1, 1, 1, 1],
            )?;

            {
                let mut arr0 = vec![1; 127];
                arr0.extend(vec![0; 127]);

                let mut arr1 = vec![1; 127];
                arr1.extend(vec![0; 127]);
                arr1.extend(vec![1; 127]);

                gemm_helper(
                    array_type(vec![2, 127], BIT),
                    array_type(vec![3, 127], BIT),
                    arr0,
                    arr1,
                    vec![1, 0, 1, 0, 0, 0],
                )?;
            }
            gemm_helper(
                array_type(vec![2, 3], BIT),
                array_type(vec![3, 3], BIT),
                array!([[1, 0, 1], [0, 1, 1]]).into_raw_vec(),
                array!([[1, 1, 1], [0, 1, 0], [1, 1, 0]]).into_raw_vec(),
                vec![0, 0, 1, 0, 1, 1],
            )?;

            gemm_helper(
                array_type(vec![2, 9], BIT),
                array_type(vec![3, 9], BIT),
                array!([[1, 0, 1, 0, 1, 0, 1, 0, 1], [0, 1, 0, 1, 0, 0, 0, 1, 0]]).into_raw_vec(),
                array!([
                    [1, 0, 1, 0, 1, 0, 1, 0, 1],
                    [0, 1, 0, 1, 0, 1, 0, 1, 0],