Function TEST

test/blas.cpp:240–263

float batch_tol = 1E-2;
TEST(MatrixMultiply, Batched) {
    const int M  = 512;
    const int K  = 512;
    const int N  = 10;
    const int D2 = 2;
    const int D3 = 3;
    for (int d3 = 1; d3 <= D3; d3 *= D3) {
        for (int d2 = 1; d2 <= D2; d2 *= D2) {
            array a = randu(M, K, d2, d3);
            array b = randu(K, N, d2, d3);
            array c = matmul(a, b);

            for (int j = 0; j < d3; j++) {
                for (int i = 0; i < d2; i++) {
                    array a_ij = a(span, span, i, j);
                    array b_ij = b(span, span, i, j);
                    array c_ij = c(span, span, i, j);
                    array res  = matmul(a_ij, b_ij);
                    ASSERT_ARRAYS_NEAR(c_ij, res, batch_tol);
                }
            }
        }
    }
}

#undef DEVICE_ITERATE