sparselu.cpp 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. // This file is part of Eigen, a lightweight C++ template library
  2. // for linear algebra.
  3. //
  4. // Copyright (C) 2012 Désiré Nuentsa-Wakam <desire.nuentsa_wakam@inria.fr>
  5. //
  6. // This Source Code Form is subject to the terms of the Mozilla
  7. // Public License v. 2.0. If a copy of the MPL was not distributed
  8. // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
  9. // SparseLU solve does not accept column major matrices for the destination.
  10. // However, as expected, the generic check_sparse_square_solving routines produces row-major
  11. // rhs and destination matrices when compiled with EIGEN_DEFAULT_TO_ROW_MAJOR
  12. #ifdef EIGEN_DEFAULT_TO_ROW_MAJOR
  13. #undef EIGEN_DEFAULT_TO_ROW_MAJOR
  14. #endif
  15. #include "sparse_solver.h"
  16. #include <Eigen/SparseLU>
  17. #include <unsupported/Eigen/SparseExtra>
  18. template<typename T> void test_sparselu_T()
  19. {
  20. SparseLU<SparseMatrix<T, ColMajor> /*, COLAMDOrdering<int>*/ > sparselu_colamd; // COLAMDOrdering is the default
  21. SparseLU<SparseMatrix<T, ColMajor>, AMDOrdering<int> > sparselu_amd;
  22. SparseLU<SparseMatrix<T, ColMajor, long int>, NaturalOrdering<long int> > sparselu_natural;
  23. check_sparse_square_solving(sparselu_colamd, 300, 100000, true);
  24. check_sparse_square_solving(sparselu_amd, 300, 10000, true);
  25. check_sparse_square_solving(sparselu_natural, 300, 2000, true);
  26. check_sparse_square_abs_determinant(sparselu_colamd);
  27. check_sparse_square_abs_determinant(sparselu_amd);
  28. check_sparse_square_determinant(sparselu_colamd);
  29. check_sparse_square_determinant(sparselu_amd);
  30. }
  31. void test_sparselu()
  32. {
  33. CALL_SUBTEST_1(test_sparselu_T<float>());
  34. CALL_SUBTEST_2(test_sparselu_T<double>());
  35. CALL_SUBTEST_3(test_sparselu_T<std::complex<float> >());
  36. CALL_SUBTEST_4(test_sparselu_T<std::complex<double> >());
  37. }