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,none = -1.0;
12: PetscInt Ii,J,ldim,low,high,iglobal,Istart,Iend;
14: PetscInt i,j,m = 3,n = 2;
15: PetscMPIInt size,rank;
16: PetscInt solve_count;
17: 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,MATSBAIJMUMPS);
32: MatSetType(C,MATSBAIJSPOOLES);
33: MatSetOption(C,MAT_IGNORE_LOWER_TRIANGULAR);
34: } else {
35: MatSetType(C,MATPLAPACK);
36: //MatSetType(C,MATAIJSPOOLES);
37: //MatSetType(C,MATAIJMUMPS);
38: }
39:
40: MatGetOwnershipRange(C,&Istart,&Iend);
41: for (Ii=Istart; Ii<Iend; Ii++) {
42: v = -1.0; i = Ii/n; j = Ii - i*n;
43: if (i>0) {J = Ii - n; MatSetValues(C,1,&Ii,1,&J,&v,ADD_VALUES);}
44: if (i<m-1) {J = Ii + n; MatSetValues(C,1,&Ii,1,&J,&v,ADD_VALUES);}
45: if (j>0) {J = Ii - 1; MatSetValues(C,1,&Ii,1,&J,&v,ADD_VALUES);}
46: if (j<n-1) {J = Ii + 1; MatSetValues(C,1,&Ii,1,&J,&v,ADD_VALUES);}
47: v = 4.0; MatSetValues(C,1,&Ii,1,&Ii,&v,ADD_VALUES);
48: }
49: MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);
50: MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);
51: MatGetType(C,&type);
52: PetscPrintf(PETSC_COMM_WORLD," case %D, matrix type: %s\n",solve_count,type);
53: }
55: /* Free work space. */
56: MatDestroy(C);
57: PetscFinalize();
58: return 0;
59: }