void test() {
extern void testFlips();
extern void testLastFlipCountGenerator();
extern void testUtils();
extern void testSearch();
extern void testSolve();
extern void testHash();
printCompileType();
testFlips();
testUtils();
testLastFlipCountGenerator();
testSearch();
testSolve();
testHash();
}
int main(int argc, char *argv[])
{
int nConstraints = 10;
int nRHS = 7;
int n = 100;
double alpha = 1.0;
double beta = 0.0;
double gamma = 2.0;
double scale = 1.0;
int ierr = 0;
double reltol = 1.0e-9;
double abstol = 1.0e-9;
alpha = alpha / scale;
try {
bool verbose = false;
// Check for verbose output
if (argc>1)
if (argv[1][0]=='-' && argv[1][1]=='v')
verbose = true;
// Create parameter list
Teuchos::RCP<Teuchos::ParameterList> paramList =
Teuchos::rcp(new Teuchos::ParameterList);
// Create LOCA sublist
Teuchos::ParameterList& locaParamsList = paramList->sublist("LOCA");
// Create the constraints list
Teuchos::ParameterList& constraintsList =
locaParamsList.sublist("Constraints");
constraintsList.set("Bordered Solver Method", "Bordering");
// Create the "Solver" parameters sublist to be used with NOX Solvers
Teuchos::ParameterList& nlParams = paramList->sublist("NOX");
Teuchos::ParameterList& nlPrintParams = nlParams.sublist("Printing");
if (verbose)
nlPrintParams.set("Output Information",
NOX::Utils::Error +
NOX::Utils::Details +
NOX::Utils::OuterIteration +
NOX::Utils::InnerIteration +
NOX::Utils::Warning +
NOX::Utils::TestDetails +
NOX::Utils::StepperIteration +
NOX::Utils::StepperDetails);
else
nlPrintParams.set("Output Information", NOX::Utils::Error);
// Create LAPACK factory
Teuchos::RCP<LOCA::Abstract::Factory> lapackFactory =
Teuchos::rcp(new LOCA::LAPACK::Factory);
// Create global data object
globalData = LOCA::createGlobalData(paramList, lapackFactory);
// Create parsed parameter list
parsedParams =
Teuchos::rcp(new LOCA::Parameter::SublistParser(globalData));
parsedParams->parseSublists(paramList);
// Set up the problem interface
ChanProblemInterface chan(globalData, n, alpha, beta, scale);
LOCA::ParameterVector p;
p.addParameter("alpha",alpha);
p.addParameter("beta",beta);
p.addParameter("gamma",gamma);
p.addParameter("scale",scale);
// Create a group which uses that problem interface. The group will
// be initialized to contain the default initial guess for the
// specified problem.
grp = Teuchos::rcp(new LOCA::LAPACK::Group(globalData, chan));
grp->setParams(p);
// Create Jacobian operator
op = Teuchos::rcp(new LOCA::BorderedSolver::JacobianOperator(grp));
// Change initial guess to a random vector
Teuchos::RCP<NOX::Abstract::Vector> xnew = grp->getX().clone();
xnew->random();
grp->setX(*xnew);
// Create the constraints object & constraint param IDs list
constraints = Teuchos::rcp(new LinearConstraint(nConstraints, p, *xnew));
Teuchos::RCP< std::vector<int> > constraintParamIDs =
Teuchos::rcp(new std::vector<int>(1));
(*constraintParamIDs)[0] = p.getIndex("alpha");
// Create bordering solver
bordering
= globalData->locaFactory->createBorderedSolverStrategy(
parsedParams,
parsedParams->getSublist("Constraints"));
// Change strategy to LAPACK Direct Solve
constraintsList.set("Bordered Solver Method",
"LAPACK Direct Solve");
// Create direct solver
direct
= globalData->locaFactory->createBorderedSolverStrategy(
parsedParams,
parsedParams->getSublist("Constraints"));
// Check some statistics on the solution
testCompare = Teuchos::rcp(new NOX::TestCompare(
globalData->locaUtils->out(),
*(globalData->locaUtils)));
// Evaluate blocks
grp->computeF();
grp->computeJacobian();
// A
A = grp->getX().createMultiVector(nConstraints);
A->random();
// B
constraints->setX(grp->getX());
B = grp->getX().createMultiVector(nConstraints);
B->random();
constraints->setDgDx(*B);
constraints->computeConstraints();
constraints->computeDX();
// C
C =
Teuchos::rcp(new NOX::Abstract::MultiVector::DenseMatrix(nConstraints,
nConstraints));
C->random();
// Set up left- and right-hand sides
F = grp->getX().createMultiVector(nRHS);
F->random();
G = Teuchos::rcp(new NOX::Abstract::MultiVector::DenseMatrix(nConstraints,
nRHS));
G->random();
X_bordering = F->clone(NOX::ShapeCopy);
Y_bordering =
Teuchos::rcp(new NOX::Abstract::MultiVector::DenseMatrix(nConstraints,
nRHS));
X_direct = F->clone(NOX::ShapeCopy);
Y_direct =
Teuchos::rcp(new NOX::Abstract::MultiVector::DenseMatrix(nConstraints,
nRHS));
std::string testName;
// Test all nonzero
testName = "Testing all nonzero";
ierr += testSolve(false, false, false, false, false,
reltol, abstol, testName);
// Test A = 0
testName = "Testing A=0";
ierr += testSolve(true, false, false, false, false,
reltol, abstol, testName);
// Test B = 0
testName = "Testing B=0";
ierr += testSolve(false, true, false, false, false,
reltol, abstol, testName);
// Test C = 0
testName = "Testing C=0";
ierr += testSolve(false, false, true, false, false,
reltol, abstol, testName);
// Test F = 0
testName = "Testing F=0";
ierr += testSolve(false, false, false, true, false,
reltol, abstol, testName);
// Test G = 0
testName = "Testing G=0";
ierr += testSolve(false, false, false, false, true,
reltol, abstol, testName);
// Test A,B = 0
testName = "Testing A,B=0";
ierr += testSolve(true, true, false, false, false,
reltol, abstol, testName);
// Test A,F = 0
testName = "Testing A,F=0";
ierr += testSolve(true, false, false, true, false,
reltol, abstol, testName);
// Test A,G = 0
testName = "Testing A,G=0";
ierr += testSolve(true, false, false, false, true,
reltol, abstol, testName);
// Test B,F = 0
testName = "Testing B,F=0";
ierr += testSolve(false, true, false, true, false,
reltol, abstol, testName);
// Test B,G = 0
testName = "Testing B,G=0";
ierr += testSolve(false, true, false, false, true,
reltol, abstol, testName);
// Test C,F = 0
testName = "Testing C,F=0";
ierr += testSolve(false, false, true, true, false,
reltol, abstol, testName);
// Test C,G = 0
testName = "Testing C,G=0";
ierr += testSolve(false, false, true, false, true,
reltol, abstol, testName);
// Test F,G = 0
testName = "Testing F,G=0";
ierr += testSolve(false, false, false, true, true,
reltol, abstol, testName);
// Test A,B,F = 0
testName = "Testing A,B,F=0";
ierr += testSolve(true, true, false, true, false,
reltol, abstol, testName);
// Test A,B,G = 0
testName = "Testing A,B,G=0";
ierr += testSolve(true, true, false, false, true,
reltol, abstol, testName);
// Test A,F,G = 0
testName = "Testing A,F,G=0";
ierr += testSolve(true, false, false, true, true,
reltol, abstol, testName);
// Test B,F,G = 0
testName = "Testing B,F,G=0";
ierr += testSolve(false, true, false, true, true,
reltol, abstol, testName);
// Test C,F,G = 0
testName = "Testing C,F,G=0";
ierr += testSolve(false, false, true, true, true,
reltol, abstol, testName);
// Test A,B,F,G = 0
testName = "Testing A,B,F,G=0";
ierr += testSolve(true, true, false, true, true,
reltol, abstol, testName);
LOCA::destroyGlobalData(globalData);
}
catch (std::exception& e) {
std::cout << e.what() << std::endl;
ierr = 1;
}
catch (const char *s) {
std::cout << s << std::endl;
ierr = 1;
}
catch (...) {
std::cout << "Caught unknown exception!" << std::endl;
ierr = 1;
}
if (ierr == 0)
std::cout << "All tests passed!" << std::endl;
else
std::cout << ierr << " test(s) failed!" << std::endl;
return ierr;
}
int main(int argc, char *argv[])
{
int nConstraints = 10;
int nRHS = 7;
double left_bc = 0.0;
double right_bc = 1.0;
double nonlinear_factor = 1.0;
int ierr = 0;
double reltol = 1.0e-8;
double abstol = 1.0e-8;
double lstol = 1.0e-11;
int MyPID = 0;
try {
// Initialize MPI
#ifdef HAVE_MPI
MPI_Init(&argc,&argv);
#endif
// Create a communicator for Epetra objects
#ifdef HAVE_MPI
Epetra_MpiComm Comm( MPI_COMM_WORLD );
#else
Epetra_SerialComm Comm;
#endif
// Get the total number of processors
MyPID = Comm.MyPID();
int NumProc = Comm.NumProc();
bool verbose = false;
// Check for verbose output
if (argc>1)
if (argv[1][0]=='-' && argv[1][1]=='v')
verbose = true;
// Get the number of elements from the command line
int NumGlobalElements = 0;
if ((argc > 2) && (verbose))
NumGlobalElements = atoi(argv[2]) + 1;
else if ((argc > 1) && (!verbose))
NumGlobalElements = atoi(argv[1]) + 1;
else
NumGlobalElements = 101;
// The number of unknowns must be at least equal to the
// number of processors.
if (NumGlobalElements < NumProc) {
std::cout << "numGlobalBlocks = " << NumGlobalElements
<< " cannot be < number of processors = " << NumProc << std::endl;
exit(1);
}
// Seed the random number generator in Teuchos. We create random
// bordering matrices and it is possible different processors might generate
// different matrices. By setting the seed, this shouldn't happen.
Teuchos::ScalarTraits<double>::seedrandom(12345);
// Create parameter list
Teuchos::RCP<Teuchos::ParameterList> paramList =
Teuchos::rcp(new Teuchos::ParameterList);
// Create LOCA sublist
Teuchos::ParameterList& locaParamsList = paramList->sublist("LOCA");
// Create the constraints list
Teuchos::ParameterList& constraintsList =
locaParamsList.sublist("Constraints");
constraintsList.set("Bordered Solver Method", "Bordering");
// Create the "Solver" parameters sublist to be used with NOX Solvers
Teuchos::ParameterList& nlParams = paramList->sublist("NOX");
Teuchos::ParameterList& nlPrintParams = nlParams.sublist("Printing");
nlPrintParams.set("MyPID", MyPID);
if (verbose)
nlPrintParams.set("Output Information",
NOX::Utils::Error +
NOX::Utils::Details +
NOX::Utils::LinearSolverDetails +
NOX::Utils::OuterIteration +
NOX::Utils::InnerIteration +
NOX::Utils::Warning +
NOX::Utils::TestDetails +
NOX::Utils::StepperIteration +
NOX::Utils::StepperDetails);
else
nlPrintParams.set("Output Information", NOX::Utils::Error);
// Create the "Direction" sublist for the "Line Search Based" solver
Teuchos::ParameterList& dirParams = nlParams.sublist("Direction");
Teuchos::ParameterList& newParams = dirParams.sublist("Newton");
Teuchos::ParameterList& lsParams = newParams.sublist("Linear Solver");
lsParams.set("Aztec Solver", "GMRES");
lsParams.set("Max Iterations", 100);
lsParams.set("Tolerance", lstol);
if (verbose)
lsParams.set("Output Frequency", 1);
else
lsParams.set("Output Frequency", 0);
lsParams.set("Scaling", "None");
lsParams.set("Preconditioner", "Ifpack");
//lsParams.set("Preconditioner", "AztecOO");
//lsParams.set("Jacobian Operator", "Matrix-Free");
//lsParams.set("Preconditioner Operator", "Finite Difference");
lsParams.set("Aztec Preconditioner", "ilut");
//lsParams.set("Overlap", 2);
//lsParams.set("Fill Factor", 2.0);
//lsParams.set("Drop Tolerance", 1.0e-12);
lsParams.set("Max Age Of Prec", -2);
// Create the FiniteElementProblem class. This creates all required
// Epetra objects for the problem and allows calls to the
// function (RHS) and Jacobian evaluation routines.
Tcubed_FiniteElementProblem Problem(NumGlobalElements, Comm);
// Get the vector from the Problem
Epetra_Vector& soln = Problem.getSolution();
// Initialize Solution
soln.PutScalar(0.0);
// Create and initialize the parameter vector
LOCA::ParameterVector pVector;
pVector.addParameter("Nonlinear Factor",nonlinear_factor);
pVector.addParameter("Left BC", left_bc);
pVector.addParameter("Right BC", right_bc);
// Create the interface between the test problem and the nonlinear solver
// This is created by the user using inheritance of the abstract base
// class:
Teuchos::RCP<Problem_Interface> interface =
Teuchos::rcp(new Problem_Interface(Problem));
Teuchos::RCP<LOCA::Epetra::Interface::Required> iReq = interface;
Teuchos::RCP<NOX::Epetra::Interface::Jacobian> iJac = interface;
// Create the Epetra_RowMatrixfor the Jacobian/Preconditioner
Teuchos::RCP<Epetra_RowMatrix> Amat =
Teuchos::rcp(&Problem.getJacobian(),false);
// Create the linear systems
Teuchos::RCP<NOX::Epetra::LinearSystemAztecOO> linsys =
Teuchos::rcp(new NOX::Epetra::LinearSystemAztecOO(nlPrintParams,
lsParams, iReq, iJac,
Amat, soln));
// Create the loca vector
NOX::Epetra::Vector locaSoln(soln);
// Create Epetra factory
Teuchos::RCP<LOCA::Abstract::Factory> epetraFactory =
Teuchos::rcp(new LOCA::Epetra::Factory);
// Create global data object
globalData = LOCA::createGlobalData(paramList, epetraFactory);
// Create parsed parameter list
parsedParams =
Teuchos::rcp(new LOCA::Parameter::SublistParser(globalData));
parsedParams->parseSublists(paramList);
// Create the Group
grp = Teuchos::rcp(new LOCA::Epetra::Group(globalData, nlPrintParams,
iReq, locaSoln,
linsys, pVector));
// Create Jacobian operator
op = Teuchos::rcp(new LOCA::BorderedSolver::JacobianOperator(grp));
// Change initial guess to a random vector
Teuchos::RCP<NOX::Abstract::Vector> xnew =
grp->getX().clone();
xnew->random();
grp->setX(*xnew);
// Create the constraints object & constraint param IDs list
constraints =
Teuchos::rcp(new LinearConstraint(nConstraints, LOCA::ParameterVector(),
locaSoln));
// Create bordering solver
bordering
= globalData->locaFactory->createBorderedSolverStrategy(
parsedParams,
parsedParams->getSublist("Constraints"));
// Change strategy to Householder
constraintsList.set("Bordered Solver Method",
"Householder");
// Create householder solver
householder
= globalData->locaFactory->createBorderedSolverStrategy(
parsedParams,
parsedParams->getSublist("Constraints"));
// Check some statistics on the solution
testCompare = Teuchos::rcp(new NOX::TestCompare(
globalData->locaUtils->out(),
*(globalData->locaUtils)));
// Evaluate blocks
grp->computeF();
grp->computeJacobian();
// A
A = grp->getX().createMultiVector(nConstraints);
A->random();
// B
constraints->setX(grp->getX());
B = grp->getX().createMultiVector(nConstraints);
B->random();
constraints->setDgDx(*B);
constraints->computeConstraints();
constraints->computeDX();
// C
C = Teuchos::rcp(new NOX::Abstract::MultiVector::DenseMatrix(nConstraints,
nConstraints));
C->random();
// Set up left- and right-hand sides
F = grp->getX().createMultiVector(nRHS);
F->random();
G = Teuchos::rcp(new NOX::Abstract::MultiVector::DenseMatrix(nConstraints,
nRHS));
G->random();
X_bordering = F->clone(NOX::ShapeCopy);
Y_bordering =
Teuchos::rcp(new NOX::Abstract::MultiVector::DenseMatrix(nConstraints,
nRHS));
X_householder = F->clone(NOX::ShapeCopy);
Y_householder =
Teuchos::rcp(new NOX::Abstract::MultiVector::DenseMatrix(nConstraints,
nRHS));
std::string testName;
// Test all nonzero
testName = "Testing all nonzero";
ierr += testSolve(false, false, false, false, false,
reltol, abstol, testName);
// Test A = 0
testName = "Testing A=0";
ierr += testSolve(true, false, false, false, false,
reltol, abstol, testName);
// Test B = 0
testName = "Testing B=0";
ierr += testSolve(false, true, false, false, false,
reltol, abstol, testName);
// Test C = 0
testName = "Testing C=0";
ierr += testSolve(false, false, true, false, false,
reltol, abstol, testName);
// Test F = 0
testName = "Testing F=0";
ierr += testSolve(false, false, false, true, false,
reltol, abstol, testName);
// Test G = 0
testName = "Testing G=0";
ierr += testSolve(false, false, false, false, true,
reltol, abstol, testName);
// Test A,B = 0
testName = "Testing A,B=0";
ierr += testSolve(true, true, false, false, false,
reltol, abstol, testName);
// Test A,F = 0
testName = "Testing A,F=0";
ierr += testSolve(true, false, false, true, false,
reltol, abstol, testName);
// Test A,G = 0
testName = "Testing A,G=0";
ierr += testSolve(true, false, false, false, true,
reltol, abstol, testName);
// Test B,F = 0
testName = "Testing B,F=0";
ierr += testSolve(false, true, false, true, false,
reltol, abstol, testName);
// Test B,G = 0
testName = "Testing B,G=0";
ierr += testSolve(false, true, false, false, true,
reltol, abstol, testName);
// Test C,F = 0
testName = "Testing C,F=0";
ierr += testSolve(false, false, true, true, false,
reltol, abstol, testName);
// Test C,G = 0
testName = "Testing C,G=0";
ierr += testSolve(false, false, true, false, true,
reltol, abstol, testName);
// Test F,G = 0
testName = "Testing F,G=0";
ierr += testSolve(false, false, false, true, true,
reltol, abstol, testName);
// Test A,B,F = 0
testName = "Testing A,B,F=0";
ierr += testSolve(true, true, false, true, false,
reltol, abstol, testName);
// Test A,B,G = 0
testName = "Testing A,B,G=0";
ierr += testSolve(true, true, false, false, true,
reltol, abstol, testName);
// Test A,F,G = 0
testName = "Testing A,F,G=0";
ierr += testSolve(true, false, false, true, true,
reltol, abstol, testName);
// Test B,F,G = 0
testName = "Testing B,F,G=0";
ierr += testSolve(false, true, false, true, true,
reltol, abstol, testName);
// Test C,F,G = 0
testName = "Testing C,F,G=0";
ierr += testSolve(false, false, true, true, true,
reltol, abstol, testName);
// Test A,B,F,G = 0
testName = "Testing A,B,F,G=0";
ierr += testSolve(true, true, false, true, true,
reltol, abstol, testName);
LOCA::destroyGlobalData(globalData);
}
catch (std::exception& e) {
std::cout << e.what() << std::endl;
ierr = 1;
}
catch (const char *s) {
std::cout << s << std::endl;
ierr = 1;
}
catch (...) {
std::cout << "Caught unknown exception!" << std::endl;
ierr = 1;
}
if (MyPID == 0) {
if (ierr == 0)
std::cout << "All tests passed!" << std::endl;
else
std::cout << ierr << " test(s) failed!" << std::endl;
}
#ifdef HAVE_MPI
MPI_Finalize() ;
#endif
return ierr;
}
