10 #include "Epetra_MpiComm.h" 12 #include "Epetra_SerialComm.h" 14 #include "Epetra_Map.h" 15 #include "Epetra_Vector.h" 16 #include "Epetra_Time.h" 17 #include "Epetra_Util.h" 18 #include "Epetra_CrsMatrix.h" 21 #include "Teuchos_ParameterList.hpp" 25 const Epetra_MultiVector& x,
26 const Epetra_MultiVector& b,
27 const Epetra_MultiVector& x_exact)
29 std::vector<double> Norm;
30 int NumVectors = x.NumVectors();
31 Norm.resize(NumVectors);
32 Epetra_MultiVector Ax(x);
33 A.Multiply(
false,x,Ax);
34 Ax.Update(1.0,b,-1.0);
36 bool TestPassed =
false;
37 double TotalNorm = 0.0;
38 for (
int i = 0 ; i < NumVectors ; ++i) {
41 if (A.Comm().MyPID() == 0)
42 std::cout <<
"||Ax - b|| = " << TotalNorm << std::endl;
43 if (TotalNorm < 1e-5 )
48 Ax.Update (1.0,x,-1.0,x_exact,0.0);
50 for (
int i = 0 ; i < NumVectors ; ++i) {
53 if (A.Comm().MyPID() == 0)
54 std::cout <<
"||Ax - b|| = " << TotalNorm << std::endl;
55 if (TotalNorm < 1e-5 )
64 int main(
int argc,
char *argv[]) {
67 MPI_Init(&argc, &argv);
68 Epetra_MpiComm Comm(MPI_COMM_WORLD);
70 Epetra_SerialComm Comm;
73 int NumGlobalElements = 1000;
80 int* part =
new int[NumGlobalElements];
82 if (Comm.MyPID() == 0) {
85 for(
int i=0 ; i<NumGlobalElements ; ++i ) {
86 unsigned int r = Util.RandomInt();
87 part[i] = r%(Comm.NumProc());
91 Comm.Broadcast(part,NumGlobalElements,0);
94 int NumMyElements = 0;
95 for (
int i = 0 ; i < NumGlobalElements ; ++i) {
96 if (part[i] == Comm.MyPID())
101 int* MyGlobalElements =
new int[NumMyElements];
103 for (
int i = 0 ; i < NumGlobalElements ; ++i) {
104 if (part[i] == Comm.MyPID() )
105 MyGlobalElements[count++] = i;
108 Epetra_Map Map(NumGlobalElements,NumMyElements,MyGlobalElements,
117 Epetra_CrsMatrix Matrix(Copy,Map,NumGlobalElements);
119 int* Indices =
new int[NumGlobalElements];
120 double* Values =
new double[NumGlobalElements];
122 for (
int i = 0 ; i < NumGlobalElements ; ++i)
125 for (
int i = 0 ; i < NumMyElements ; ++i) {
126 int iGlobal = MyGlobalElements[i];
127 for (
int jGlobal = 0 ; jGlobal < NumGlobalElements ; ++jGlobal) {
128 if (iGlobal == jGlobal)
129 Values[jGlobal] = 1.0 * (NumGlobalElements + 1 ) *
130 (NumGlobalElements + 1);
131 else if (iGlobal > jGlobal)
132 Values[jGlobal] = 1.0*(jGlobal+1);
134 Values[jGlobal] = 1.0*(iGlobal+1);
136 Matrix.InsertGlobalValues(MyGlobalElements[i],
137 NumGlobalElements, Values, Indices);
141 Matrix.FillComplete();
143 delete [] MyGlobalElements;
152 Epetra_MultiVector x(Map,NumVectors);
153 Epetra_MultiVector x_exact(Map,NumVectors);
154 Epetra_MultiVector b(Map,NumVectors);
156 Matrix.Multiply(
false,x_exact,b);
162 Epetra_LinearProblem Problem;
165 Problem.SetOperator(&A);
169 Teuchos::ParameterList ParamList ;
170 ParamList.set(
"MaxProcs", Comm.NumProc());
int SymbolicFactorization()
Performs SymbolicFactorization on the matrix A.
Amesos_Dscpack: An object-oriented wrapper for Dscpack.
Amesos_TestRowMatrix: a class to test Epetra_RowMatrix based codes.
bool CheckError(const Epetra_RowMatrix &A, const Epetra_MultiVector &x, const Epetra_MultiVector &b, const Epetra_MultiVector &x_exact)
int main(int argc, char *argv[])
#define AMESOS_CHK_ERR(a)
int SetParameters(Teuchos::ParameterList &ParameterList)
Updates internal variables.
int NumericFactorization()
Performs NumericFactorization on the matrix A.
#define EPETRA_CHK_ERR(xxx)
int Solve()
Solves A X = B (or AT x = B)