xref: /petsc/src/mat/tests/ex182.c (revision 6c2b77d522d8aa5c8b27f04fddd7150d0d6755fb) 
1c4762a1bSJed Brown static char help[] = "Tests using MatShift() to create a constant diagonal matrix\n\n";
2c4762a1bSJed Brown 
3c4762a1bSJed Brown #include <petscmat.h>
4c4762a1bSJed Brown 
5d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv)
6d71ae5a4SJacob Faibussowitsch {
7c4762a1bSJed Brown   Mat           A, F;
8c4762a1bSJed Brown   MatFactorInfo info;
9c4762a1bSJed Brown   PetscInt      m = 10;
10c4762a1bSJed Brown   IS            perm;
11c4762a1bSJed Brown   PetscMPIInt   size;
12c4762a1bSJed Brown   PetscBool     issbaij;
13c4762a1bSJed Brown 
14327415f7SBarry Smith   PetscFunctionBeginUser;
159566063dSJacob Faibussowitsch   PetscCall(PetscInitialize(&argc, &argv, (char *)0, help));
169566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size));
17c4762a1bSJed Brown 
189566063dSJacob Faibussowitsch   PetscCall(MatCreate(PETSC_COMM_WORLD, &A));
199566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(A, PETSC_DECIDE, PETSC_DECIDE, m, m));
209566063dSJacob Faibussowitsch   PetscCall(MatSetFromOptions(A));
219566063dSJacob Faibussowitsch   PetscCall(MatSetUp(A));
229566063dSJacob Faibussowitsch   PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY));
239566063dSJacob Faibussowitsch   PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY));
24c4762a1bSJed Brown 
259566063dSJacob Faibussowitsch   PetscCall(MatShift(A, 1.0));
26c4762a1bSJed Brown 
279566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)A, MATSEQSBAIJ, &issbaij));
28c4762a1bSJed Brown   if (size == 1 && !issbaij) {
299566063dSJacob Faibussowitsch     PetscCall(MatGetFactor(A, MATSOLVERPETSC, MAT_FACTOR_LU, &F));
309566063dSJacob Faibussowitsch     PetscCall(MatFactorInfoInitialize(&info));
319566063dSJacob Faibussowitsch     PetscCall(ISCreateStride(PETSC_COMM_SELF, m, 0, 1, &perm));
329566063dSJacob Faibussowitsch     PetscCall(MatLUFactorSymbolic(F, A, perm, perm, &info));
339566063dSJacob Faibussowitsch     PetscCall(MatLUFactorNumeric(F, A, &info));
349566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&F));
359566063dSJacob Faibussowitsch     PetscCall(ISDestroy(&perm));
36c4762a1bSJed Brown   }
379566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&A));
389566063dSJacob Faibussowitsch   PetscCall(PetscFinalize());
39b122ec5aSJacob Faibussowitsch   return 0;
40c4762a1bSJed Brown }
41c4762a1bSJed Brown 
42c4762a1bSJed Brown /*TEST
43c4762a1bSJed Brown 
44c4762a1bSJed Brown    test:
45dfd57a17SPierre Jolivet       requires: defined(PETSC_USE_INFO)
46c4762a1bSJed Brown       args: -info
47c4762a1bSJed Brown       filter: grep malloc | sort -b
48c4762a1bSJed Brown 
49c4762a1bSJed Brown    test:
50c4762a1bSJed Brown       suffix: 2
51c4762a1bSJed Brown       nsize: 2
52dfd57a17SPierre Jolivet       requires: defined(PETSC_USE_INFO)
53c4762a1bSJed Brown       args: -info ex182info
54c4762a1bSJed Brown       filter: grep -h malloc "ex182info.0" | sort -b
55c4762a1bSJed Brown 
56c4762a1bSJed Brown    test:
57c4762a1bSJed Brown       suffix: 3
58dfd57a17SPierre Jolivet       requires: defined(PETSC_USE_INFO)
59c4762a1bSJed Brown       args: -info -mat_type baij
60c4762a1bSJed Brown       filter: grep malloc | sort -b
61c4762a1bSJed Brown 
62c4762a1bSJed Brown    test:
63c4762a1bSJed Brown       suffix: 4
64c4762a1bSJed Brown       nsize: 2
65dfd57a17SPierre Jolivet       requires: defined(PETSC_USE_INFO)
66c4762a1bSJed Brown       args: -info ex182info -mat_type baij
67c4762a1bSJed Brown       filter: grep -h malloc "ex182info.1" | sort -b
68c4762a1bSJed Brown 
69c4762a1bSJed Brown    test:
70c4762a1bSJed Brown       suffix: 5
71dfd57a17SPierre Jolivet       requires: defined(PETSC_USE_INFO)
72c4762a1bSJed Brown       args: -info -mat_type sbaij
73c4762a1bSJed Brown       filter: grep malloc | sort  -b
74c4762a1bSJed Brown 
75c4762a1bSJed Brown    test:
76c4762a1bSJed Brown       suffix: 6
77c4762a1bSJed Brown       nsize: 2
78dfd57a17SPierre Jolivet       requires: defined(PETSC_USE_INFO)
79c4762a1bSJed Brown       args: -info ex182info -mat_type sbaij
80c4762a1bSJed Brown       filter: grep -h malloc "ex182info.0" | sort -b
81c4762a1bSJed Brown 
82c20d7725SJed Brown    test:
83c20d7725SJed Brown      suffix: 7
84c20d7725SJed Brown      nsize: 1
85dfd57a17SPierre Jolivet      requires: defined(PETSC_USE_INFO)
86c20d7725SJed Brown      args: -info ex182info
87c20d7725SJed Brown      filter: grep -h malloc "ex182info.0" | grep -v Running | sort -b
88c20d7725SJed Brown 
89c20d7725SJed Brown    test:
90c20d7725SJed Brown      suffix: 8
91c20d7725SJed Brown      nsize: 2
92dfd57a17SPierre Jolivet      requires: defined(PETSC_USE_INFO)
93c20d7725SJed Brown      args: -info ex182info:mat
94c20d7725SJed Brown      filter: grep -h malloc "ex182info.1" | sort -b
95c20d7725SJed Brown 
96c20d7725SJed Brown    test:
97c20d7725SJed Brown      suffix: 9
98c20d7725SJed Brown      nsize: 1
99*6c2b77d5SStefano Zampini      requires: defined(PETSC_USE_INFO) !defined(PETSC_HAVE_THREADSAFETY)
100c20d7725SJed Brown      args: -info ex182info:sys
101d6837840SMatthew G. Knepley      filter: grep -h -ve Running -ve MPI_Comm -ve Initialize -ve communicator -ve HostName -ve PetscDetermineInitialFPTrap -ve libpetscbamg "ex182info.0" | sort -b
102c20d7725SJed Brown 
103c20d7725SJed Brown    test:
104c20d7725SJed Brown      suffix: 10
105c20d7725SJed Brown      nsize: 1
106dfd57a17SPierre Jolivet      requires: defined(PETSC_USE_INFO)
107c20d7725SJed Brown      args: -info :~sys
108c20d7725SJed Brown      filter: grep -h malloc | sort -b
109c20d7725SJed Brown 
110c20d7725SJed Brown    test:
111c20d7725SJed Brown      suffix: 11
112c20d7725SJed Brown      nsize: 2
113dfd57a17SPierre Jolivet      requires: defined(PETSC_USE_INFO)
114c20d7725SJed Brown      args: -info :~sys,mat
115c20d7725SJed Brown      filter: sort -b
116c20d7725SJed Brown 
117c20d7725SJed Brown    test:
118c20d7725SJed Brown      suffix: 12
119c20d7725SJed Brown      nsize: 2
120*6c2b77d5SStefano Zampini      requires: defined(PETSC_USE_INFO) !defined(PETSC_HAVE_THREADSAFETY)
121c20d7725SJed Brown      args: -info ex182info:sys,mat
122d6837840SMatthew G. Knepley      filter: grep -h -ve Running -ve MPI_Comm -ve Initialize -ve communicator -ve HostName -ve PetscDetermineInitialFPTrap -ve libpetscbamg "ex182info.1" | sort -b
123c20d7725SJed Brown 
124c20d7725SJed Brown    test:
125c20d7725SJed Brown      suffix: 13
126c20d7725SJed Brown      nsize: 2
127dfd57a17SPierre Jolivet      requires: defined(PETSC_USE_INFO)
128c20d7725SJed Brown      args: -info ex182info:mat:~self
129c20d7725SJed Brown      filter: grep -h "ex182info.1" | sort -b
130c20d7725SJed Brown 
131c20d7725SJed Brown    test:
132c20d7725SJed Brown      suffix: 14
133c20d7725SJed Brown      nsize: 2
134dfd57a17SPierre Jolivet      requires: defined(PETSC_USE_INFO)
135c20d7725SJed Brown      args: -info ex182info::~self
136aba4c478SBarry Smith      filter: grep -h -ve Running -ve MPI_Comm -ve Initialize -ve communicator -ve HostName -ve PetscDetermineInitialFPTrap "ex182info.1" | sort -b
137c20d7725SJed Brown 
138c20d7725SJed Brown    test:
139c20d7725SJed Brown      suffix: 15
140c20d7725SJed Brown      nsize: 2
141*6c2b77d5SStefano Zampini      requires: defined(PETSC_USE_INFO) !defined(PETSC_HAVE_THREADSAFETY)
142c20d7725SJed Brown      args: -info ex182info::self
143d6837840SMatthew G. Knepley      filter: grep -h -ve Running -ve MPI_Comm -ve Initialize -ve communicator -ve HostName -ve PetscDetermineInitialFPTrap -ve libpetscbamg "ex182info.1" | sort -b
144c20d7725SJed Brown 
145c4762a1bSJed Brown TEST*/
146