|
Stratimikos Package Browser (Single Doxygen Collection)
Version of the Day
|
|
Stratimikos Package Browser (Single Doxygen Collection)
Version of the Day
|
00001 // @HEADER 00002 // *********************************************************************** 00003 // 00004 // Stratimikos: Thyra-based strategies for linear solvers 00005 // Copyright (2006) Sandia Corporation 00006 // 00007 // Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive 00008 // license for use of this work by or on behalf of the U.S. Government. 00009 // 00010 // This library is free software; you can redistribute it and/or modify 00011 // it under the terms of the GNU Lesser General Public License as 00012 // published by the Free Software Foundation; either version 2.1 of the 00013 // License, or (at your option) any later version. 00014 // 00015 // This library is distributed in the hope that it will be useful, but 00016 // WITHOUT ANY WARRANTY; without even the implied warranty of 00017 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 00018 // Lesser General Public License for more details. 00019 // 00020 // You should have received a copy of the GNU Lesser General Public 00021 // License along with this library; if not, write to the Free Software 00022 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 00023 // USA 00024 // Questions? Contact Roscoe A. Bartlett (rabartl@sandia.gov) 00025 // 00026 // *********************************************************************** 00027 // @HEADER 00028 */ 00029 00030 #include "test_single_aztecoo_thyra_solver.hpp" 00031 #include "Teuchos_GlobalMPISession.hpp" 00032 #include "Teuchos_CommandLineProcessor.hpp" 00033 #include "Teuchos_ParameterList.hpp" 00034 #include "az_aztec_defs.h" 00035 00036 struct MatrixTestPacket { 00037 MatrixTestPacket( 00038 std::string _matrixFile 00039 ,double _maxFwdError 00040 ,int _maxIters 00041 ,double _maxResid 00042 ,double _maxSolutionError 00043 ,double _maxSlackErrorFrac 00044 ,int _maxPrecIters 00045 ,double _maxPrecResid 00046 ,double _maxPrecSolutionError 00047 ,double _maxPrecSlackErrorFrac 00048 ) 00049 :matrixFile(_matrixFile) 00050 ,maxFwdError(_maxFwdError) 00051 ,maxIters(_maxIters) 00052 ,maxResid(_maxResid) 00053 ,maxSolutionError(_maxSolutionError) 00054 ,maxSlackErrorFrac(_maxSlackErrorFrac) 00055 ,maxPrecIters(_maxPrecIters) 00056 ,maxPrecResid(_maxPrecResid) 00057 ,maxPrecSolutionError(_maxPrecSolutionError) 00058 ,maxPrecSlackErrorFrac(_maxPrecSlackErrorFrac) 00059 {} 00060 std::string matrixFile; 00061 double maxFwdError; 00062 int maxIters; 00063 double maxResid; 00064 double maxSolutionError; 00065 double maxSlackErrorFrac; 00066 int maxPrecIters; 00067 double maxPrecResid; 00068 double maxPrecSolutionError; 00069 double maxPrecSlackErrorFrac; 00070 }; 00071 00072 int main(int argc, char* argv[]) 00073 { 00074 00075 Teuchos::GlobalMPISession mpiSession(&argc,&argv); 00076 00077 using Teuchos::CommandLineProcessor; 00078 00079 bool result, success = true; 00080 bool verbose = true; 00081 00082 Teuchos::FancyOStream out(Teuchos::rcp(&std::cout,false)); 00083 00084 try { 00085 00086 // 00087 // Read options from command-line 00088 // 00089 00090 std::string matrixDir = ""; 00091 int numRandomVectors = 1; 00092 bool showAllTests = false; 00093 bool showAllTestsDetails = false; 00094 bool dumpAll = false; 00095 std::string aztecOutputLevel = "freq"; 00096 int aztecOutputFreq = 0; 00097 00098 CommandLineProcessor clp(false); // Don't throw exceptions 00099 clp.setOption( "matrix-dir", &matrixDir, "Base directory for the test matrices" ); 00100 clp.setOption( "num-random-vectors", &numRandomVectors, "Number of times a test is performed with different random vectors." ); 00101 clp.setOption( "verbose", "quiet", &verbose, "Set if output is printed or not." ); 00102 clp.setOption( "show-all-tests", "no-show-all-tests", &showAllTests, "Set if all the tests are shown or not." ); 00103 clp.setOption( "show-all-tests-details", "no-show-all-tests-details", &showAllTestsDetails, "Set if all the details of the tests are shown or not." ); 00104 clp.setOption( "dump-all", "no-dump-all", &dumpAll, "Determines if vectors are printed or not." ); 00105 clp.setOption( "aztec-output-level", &aztecOutputLevel, "Aztec output level (freq,last,summary,warnings,all)" ); 00106 clp.setOption( "aztec-output-freq", &aztecOutputFreq, "Aztec output freqency (> 0)" ); 00107 CommandLineProcessor::EParseCommandLineReturn parse_return = clp.parse(argc,argv); 00108 if( parse_return != CommandLineProcessor::PARSE_SUCCESSFUL ) return parse_return; 00109 00110 TEUCHOS_TEST_FOR_EXCEPT( matrixDir == "" ); 00111 00112 // 00113 // Define the test matrices 00114 // 00115 00116 const int numTestMatrices = 9; 00117 00118 typedef MatrixTestPacket MTP; 00119 00120 // Set up the matices and the tolerances. 00121 // Note, we may need to adjust these for bad platforms ... 00122 const MTP testMatrices[numTestMatrices] = 00123 { 00124 MTP("bcsstk01.mtx" ,1e-12, 40 , 1e-4, 0.6, 1.0, 20 , 1e-10, 0.5, 1.0) 00125 ,MTP("bcsstk02.mtx" ,1e-12, 40 , 1e-3, 0.5, 1.0, 2 , 1e-10, 0.5, 1.0) 00126 ,MTP("bcsstk04.mtx" ,1e-12, 80 , 1e-4, 0.999990, 1.0, 40 , 1e-10, 0.999990, 1.0) 00127 ,MTP("Diagonal.mtx" ,1e-12, 4 , 1e-6, 1e-14, 1.0, 2 , 1e-10, 1e-14, 1.0) 00128 ,MTP("FourByFour.mtx" ,1e-12, 4 , 1e-6, 1e-14, 1.0, 2 , 1e-10, 1e-14, 1.0) 00129 ,MTP("KheadK.mtx" ,1e-12, 8 , 1e-6, 1e-14, 1.0, 2 , 1e-10, 1e-14, 1.0) 00130 ,MTP("KheadSorted.mtx" ,1e-12, 8 , 1e-6, 1e-14, 1.0, 2 , 1e-10, 1e-14, 1.0) 00131 ,MTP("nos1.mtx" ,1e-11, 200, 1e-4, 0.6, 1.0, 237, 1e-2, 5.0, 1.0) 00132 ,MTP("nos5.mtx" ,1e-12, 468, 1e-5, 0.5, 1.0, 468, 1e-10, 0.5, 1.0) 00133 }; 00134 // 00135 // Loop through all of the test matrices 00136 // 00137 for( int matrix_i = 0; matrix_i < numTestMatrices; ++matrix_i ) { 00138 const MatrixTestPacket 00139 mtp = testMatrices[matrix_i]; 00140 // 00141 // Do unpreconditioned and preconditioned solves 00142 // 00143 for( int prec_i = 0; prec_i < 2; ++prec_i ) { 00144 if(verbose) 00145 out << std::endl<<matrix_i<<":"<<prec_i<<": Testing, matrixFile=\'"<<mtp.matrixFile<<"\', "; 00146 bool testTranspose; 00147 double maxResid; 00148 double maxSolutionError; 00149 double maxSlackErrorFrac; 00150 Teuchos::ParameterList 00151 paramList("AztecOOLinearOpWithSolveFactory"); 00152 Teuchos::ParameterList 00153 &fwdSolvePL = paramList.sublist("Forward Solve"), 00154 &adjSolvePL = paramList.sublist("Adjoint Solve"); 00155 Teuchos::ParameterList 00156 &fwdAztecOOPL = fwdSolvePL.sublist("AztecOO"), 00157 &adjAztecOOPL = adjSolvePL.sublist("AztecOO"); 00158 if( aztecOutputLevel != "freq" ) { 00159 fwdAztecOOPL.set("Output Frequency",aztecOutputLevel); 00160 adjAztecOOPL.set("Output Frequency",aztecOutputLevel); 00161 } 00162 else { 00163 fwdAztecOOPL.set("Output Frequency",aztecOutputFreq); 00164 adjAztecOOPL.set("Output Frequency",aztecOutputFreq); 00165 } 00166 if(prec_i==0) { 00167 out << "no aztec preconditioning ... "; 00168 fwdAztecOOPL.set("Aztec Preconditioner","none"); 00169 testTranspose = true; 00170 fwdSolvePL.set("Max Iterations",mtp.maxIters); 00171 adjSolvePL.set("Max Iterations",mtp.maxIters); 00172 maxResid = mtp.maxResid; 00173 maxSolutionError = mtp.maxSolutionError; 00174 maxSlackErrorFrac = mtp.maxSlackErrorFrac; 00175 } 00176 else { 00177 out << "using aztec preconditioning ... "; 00178 fwdAztecOOPL.set("Aztec Preconditioner","ilu"); 00179 testTranspose = false; 00180 fwdSolvePL.set("Max Iterations",mtp.maxPrecIters); 00181 adjSolvePL.set("Max Iterations",mtp.maxPrecIters); 00182 maxResid = mtp.maxPrecResid; 00183 maxSolutionError = mtp.maxPrecSolutionError; 00184 maxSlackErrorFrac = mtp.maxPrecSlackErrorFrac; 00185 } 00186 std::ostringstream oss; 00187 Teuchos::FancyOStream fancy_oss(Teuchos::rcp(&oss,false)); 00188 result = 00189 Thyra::test_single_aztecoo_thyra_solver( 00190 matrixDir+"/"+mtp.matrixFile,testTranspose,numRandomVectors 00191 ,mtp.maxFwdError,maxResid,maxSolutionError 00192 ,showAllTestsDetails,dumpAll,¶mList,&fancy_oss 00193 ); 00194 if(!result) success = false; 00195 if(verbose) { 00196 if(result) { 00197 if(showAllTests) 00198 out << std::endl << oss.str(); 00199 else 00200 out << " : passed!\n"; 00201 } 00202 else { 00203 if(showAllTests) 00204 out << std::endl << oss.str(); 00205 else 00206 out << " : failed!\n"; 00207 } 00208 } 00209 } 00210 } 00211 00212 } 00213 catch( const std::exception &excpt ) { 00214 std::cerr << "*** Caught standard exception : " << excpt.what() << std::endl; 00215 success = false; 00216 } 00217 catch( ... ) { 00218 std::cerr << "*** Caught an unknown exception\n"; 00219 success = false; 00220 } 00221 00222 if (verbose) { 00223 if(success) out << "\nCongratulations! All of the tests checked out!\n"; 00224 else out << "\nOh no! At least one of the tests failed!\n"; 00225 } 00226 00227 return ( success ? 0 : 1 ); 00228 }
1.7.6.1
