xref: /petsc/src/mat/tests/ex232.c (revision b122ec5aa1bd4469eb4e0673542fb7de3f411254)
1c4762a1bSJed Brown 
2c4762a1bSJed Brown /* tests MatSeqSBAIJSetPreallocationCSR() and MatMPISBAIJSetPreallocationCSR() */
3c4762a1bSJed Brown 
4c4762a1bSJed Brown #include <petsc.h>
5c4762a1bSJed Brown 
6c4762a1bSJed Brown int main(int argc, char **args)
7c4762a1bSJed Brown {
8c4762a1bSJed Brown   PetscInt       ia[3] = { 0, 2, 4};
9c4762a1bSJed Brown   PetscInt       ja[4] = { 0, 1, 0, 1};
10c4762a1bSJed Brown   PetscScalar    c[16] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 };
11c4762a1bSJed Brown   Mat            ssbaij;
12c4762a1bSJed Brown   Mat            msbaij;
13c4762a1bSJed Brown 
14*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscInitialize(&argc,&args,(char*)0,(char*)0));
15c4762a1bSJed Brown 
165f80ce2aSJacob Faibussowitsch   CHKERRQ(MatCreate(PETSC_COMM_SELF, &ssbaij));
175f80ce2aSJacob Faibussowitsch   CHKERRQ(MatCreate(PETSC_COMM_SELF, &msbaij));
185f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSetType(ssbaij, MATSEQSBAIJ));
195f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSetType(msbaij, MATMPISBAIJ));
205f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSetBlockSize(ssbaij, 2));
215f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSetSizes(ssbaij, 4, 4, 4, 4));
225f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSetBlockSize(msbaij, 2));
235f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSetSizes(msbaij, 4, 4, 4, 4));
245f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqSBAIJSetPreallocationCSR(ssbaij, 2, ia, ja, c));
255f80ce2aSJacob Faibussowitsch   CHKERRQ(MatMPISBAIJSetPreallocationCSR(msbaij, 2, ia, ja, c));
265f80ce2aSJacob Faibussowitsch   CHKERRQ(MatView(ssbaij, PETSC_VIEWER_STDOUT_(PETSC_COMM_SELF)));
275f80ce2aSJacob Faibussowitsch   CHKERRQ(MatView(msbaij, PETSC_VIEWER_STDOUT_(PETSC_COMM_SELF)));
285f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDestroy(&ssbaij));
295f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDestroy(&msbaij));
30*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscFinalize());
31*b122ec5aSJacob Faibussowitsch   return 0;
32c4762a1bSJed Brown }
33c4762a1bSJed Brown 
34c4762a1bSJed Brown /*TEST
35c4762a1bSJed Brown 
36c4762a1bSJed Brown    test:
37c4762a1bSJed Brown      filter: sed "s?\.??g"
38c4762a1bSJed Brown 
39c4762a1bSJed Brown TEST*/
40