void FinalizeQD()
{
fpu_fix_end( &::oldControlWord );
}
void unfix() {fpu_fix_end(&old_cw);}
int main(int argc, char*argv[])
{
unsigned int old_cw;
fpu_fix_start(&old_cw);
Teuchos::Time timer("total");
timer.start();
Teuchos::GlobalMPISession mpisess(&argc,&argv,&std::cout);
Tpetra::DefaultPlatform::DefaultPlatformType& platform = Tpetra::DefaultPlatform::getDefaultPlatform();
Teuchos::RCP<const Teuchos::Comm<int> > comm = platform.getComm();
typedef dd_real Scalar;
typedef int LO; //LocalOrdinal
typedef int GO; //GlobalOrdinal
typedef Tpetra::DefaultPlatform::DefaultPlatformType::NodeType Node;
typedef Tpetra::MultiVector<Scalar,LO,GO,Node> TMV;
typedef Tpetra::Operator<Scalar,LO,GO,Node> TOP;
typedef Belos::LinearProblem<Scalar,TMV,TOP> BLinProb;
typedef Belos::SolverManager<Scalar,TMV,TOP> BSolverMgr;
//Just get one parameter from the command-line: the name of an xml file
//to get parameters from.
std::string xml_file("calore1_mm.xml");
process_command_line(argc, argv, xml_file);
//Read the contents of the xml file into a ParameterList. That parameter list
//should specify a matrix-file and optionally which Belos solver to use, and
//which Ifpack2 preconditioner to use, etc. If there are sublists of parameters
//for Belos and Ifpack2, those will be passed to the respective destinations
//from within the build_problem and build_solver functions.
std::cout << "Every proc reading parameters from xml_file: "
<< xml_file << std::endl;
Teuchos::ParameterList test_params =
Teuchos::ParameterXMLFileReader(xml_file).getParameters();
//The build_problem function is located in build_problem.hpp.
//Note that build_problem calls build_precond and sets a preconditioner on the
//linear-problem, if a preconditioner is specified.
Teuchos::RCP<BLinProb> problem = build_problem<Scalar,LO,GO,Node>(test_params, comm);
//The build_solver function is located in build_solver.hpp:
Teuchos::RCP<BSolverMgr> solver = build_solver<Scalar,TMV,TOP>(test_params, problem);
Belos::ReturnType ret = solver->solve();
if (comm->getRank() == 0) {
std::cout << "Converged in " << solver->getNumIters() << " iterations." << std::endl;
}
//Next compute residual vector and then 2-norm of residual:
Teuchos::RCP<TMV> R = Teuchos::rcp(new TMV(*problem->getRHS()));
problem->computeCurrResVec(&*R, &*problem->getLHS(), &*problem->getRHS());
Teuchos::Array<Teuchos::ScalarTraits<Scalar>::magnitudeType> norms(R->getNumVectors());
R->norm2(norms);
if (norms.size() < 1) {
throw std::runtime_error("ERROR: norms.size()==0 indicates R->getNumVectors()==0.");
}
if (comm->getRank() == 0) {
std::cout << "2-Norm of 0th residual vec: " << norms[0] << std::endl;
}
//If the xml file specified a number of iterations to expect, then we will
//use that as a test pass/fail criteria.
if (test_params.isParameter("expectNumIters")) {
int expected_iters = 0;
Ifpack2::getParameter(test_params, "expectNumIters", expected_iters);
int actual_iters = solver->getNumIters();
if (ret == Belos::Converged && actual_iters <= expected_iters && norms[0] < 1.e-7) {
if (comm->getRank() == 0) {
std::cout << "End Result: TEST PASSED" << std::endl;
}
}
else {
if (comm->getRank() == 0) {
std::cout << "Actual iters("<<actual_iters
<<") != expected number of iterations ("
<<expected_iters<<"), or resid-norm(" << norms[0] << ") >= 1.e-7"<<std::endl;
}
}
}
fpu_fix_end(&old_cw);
timer.stop();
if (comm->getRank() == 0) {
std::cout << "proc 0 total program time: " << timer.totalElapsedTime()
<< std::endl;
}
return 0;
}
void f_fpu_fix_end(unsigned int *old_cw) {
fpu_fix_end(old_cw);
}