Actual source code: ex115.c
2: static char help[] = "Test memory leak when user switches from one matrix type to another\n\n";
4: #include petscmat.h
8: int main(int argc,char **args)
9: {
10: Mat C=0;
11: PetscScalar v;
12: PetscInt Ii,J,Istart,Iend;
14: PetscInt i,j,m = 3,n = 2;
15: PetscMPIInt size,rank;
16: PetscInt solve_count;
17: const MatType type;
19: PetscInitialize(&argc,&args,(char *)0,help);
20: PetscOptionsGetInt(PETSC_NULL,"-m",&m,PETSC_NULL);
21: MPI_Comm_rank(PETSC_COMM_WORLD,&rank);
22: MPI_Comm_size(PETSC_COMM_WORLD,&size);
23: n = 2*size;
25: MatCreate(PETSC_COMM_WORLD,&C);
26: MatSetSizes(C,PETSC_DECIDE,PETSC_DECIDE,m*n,m*n);
27: MatSetFromOptions(C);
29: for (solve_count=0; solve_count<3; solve_count++){
30: if (solve_count == 1){
31: MatSetType(C,MATSBAIJ);
32: MatSetOption(C,MAT_IGNORE_LOWER_TRIANGULAR,PETSC_TRUE);
33: } else {
34: MatSetType(C,MATMPIDENSE);
35: }
36:
37: MatGetOwnershipRange(C,&Istart,&Iend);
38: for (Ii=Istart; Ii<Iend; Ii++) {
39: v = -1.0; i = Ii/n; j = Ii - i*n;
40: if (i>0) {J = Ii - n; MatSetValues(C,1,&Ii,1,&J,&v,ADD_VALUES);}
41: if (i<m-1) {J = Ii + n; MatSetValues(C,1,&Ii,1,&J,&v,ADD_VALUES);}
42: if (j>0) {J = Ii - 1; MatSetValues(C,1,&Ii,1,&J,&v,ADD_VALUES);}
43: if (j<n-1) {J = Ii + 1; MatSetValues(C,1,&Ii,1,&J,&v,ADD_VALUES);}
44: v = 4.0; MatSetValues(C,1,&Ii,1,&Ii,&v,ADD_VALUES);
45: }
46: MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);
47: MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);
48: MatGetType(C,&type);
49: PetscPrintf(PETSC_COMM_WORLD," case %D, matrix type: %s\n",solve_count,type);
50: }
52: /* Free work space. */
53: MatDestroy(C);
54: PetscFinalize();
55: return 0;
56: }