Function solveTriangleTester

test/solve_dense.cpp:121–171

template<typename T>
void solveTriangleTester(const int n, const int k, bool is_upper, double eps,
                         int targetDevice = -1) {
    if (targetDevice >= 0) setDevice(targetDevice);

    deviceGC();

    SUPPORTED_TYPE_CHECK(T);
    LAPACK_ENABLED_CHECK();

#if 1
    array A  = cpu_randu<T>(dim4(n, n));
    array X0 = cpu_randu<T>(dim4(n, k));
#else
    array A  = randu(n, n, (dtype)dtype_traits<T>::af_type);
    array X0 = randu(n, k, (dtype)dtype_traits<T>::af_type);
#endif

    array L, U, pivot;
    lu(L, U, pivot, A);

    array AT = is_upper ? U : L;
    array B0 = matmul(AT, X0);
    array X1;

    if (is_upper) {
        //! [ex_solve_upper]
        array X = solve(AT, B0, AF_MAT_UPPER);
        //! [ex_solve_upper]

        X1 = X;
    } else {
        //! [ex_solve_lower]
        array X = solve(AT, B0, AF_MAT_LOWER);
        //! [ex_solve_lower]

        X1 = X;
    }

    array B1 = matmul(AT, X1);

    ASSERT_NEAR(
        0,
        sum<typename dtype_traits<T>::base_type>(af::abs(real(B0 - B1))) /
            (n * k),
        eps);
    ASSERT_NEAR(
        0,
        sum<typename dtype_traits<T>::base_type>(af::abs(imag(B0 - B1))) /
            (n * k),
        eps);
}

template<typename T>
class Solve : public ::testing::Test {};