1 static const char help[] = "Test MatGetLocalSubMatrix() with multiple levels of nesting.\n\n"; 2 3 #include <petscmat.h> 4 5 int main(int argc, char *argv[]) 6 { 7 IS is0a,is0b,is0,is1,isl0a,isl0b,isl0,isl1; 8 Mat A,Aexplicit; 9 PetscBool usenest; 10 PetscMPIInt rank,size; 11 PetscInt i,j; 12 13 PetscCall(PetscInitialize(&argc,&argv,NULL,help)); 14 PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD,&rank)); 15 PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size)); 16 17 { 18 PetscInt ix0a[1],ix0b[1],ix0[2],ix1[1]; 19 20 ix0a[0] = rank*2+0; 21 ix0b[0] = rank*2+1; 22 ix0[0] = rank*3+0; ix0[1] = rank*3+1; 23 ix1[0] = rank*3+2; 24 PetscCall(ISCreateGeneral(PETSC_COMM_WORLD,1,ix0a,PETSC_COPY_VALUES,&is0a)); 25 PetscCall(ISCreateGeneral(PETSC_COMM_WORLD,1,ix0b,PETSC_COPY_VALUES,&is0b)); 26 PetscCall(ISCreateGeneral(PETSC_COMM_WORLD,2,ix0,PETSC_COPY_VALUES,&is0)); 27 PetscCall(ISCreateGeneral(PETSC_COMM_WORLD,1,ix1,PETSC_COPY_VALUES,&is1)); 28 } 29 { 30 PetscCall(ISCreateStride(PETSC_COMM_SELF,6,0,1,&isl0)); 31 PetscCall(ISCreateStride(PETSC_COMM_SELF,3,0,1,&isl0a)); 32 PetscCall(ISCreateStride(PETSC_COMM_SELF,3,3,1,&isl0b)); 33 PetscCall(ISCreateStride(PETSC_COMM_SELF,3,6,1,&isl1)); 34 } 35 36 usenest = PETSC_FALSE; 37 PetscCall(PetscOptionsGetBool(NULL,NULL,"-nest",&usenest,NULL)); 38 if (usenest) { 39 ISLocalToGlobalMapping l2g; 40 PetscInt l2gind[3]; 41 Mat B[9]; 42 43 l2gind[0] = (rank-1+size)%size; l2gind[1] = rank; l2gind[2] = (rank+1)%size; 44 PetscCall(ISLocalToGlobalMappingCreate(PETSC_COMM_WORLD,1,3,l2gind,PETSC_COPY_VALUES,&l2g)); 45 for (i=0; i<9; i++) { 46 PetscCall(MatCreateAIJ(PETSC_COMM_WORLD,1,1,PETSC_DECIDE,PETSC_DECIDE,PETSC_DECIDE,NULL,PETSC_DECIDE,NULL,&B[i])); 47 PetscCall(MatSetUp(B[i])); 48 PetscCall(MatSetLocalToGlobalMapping(B[i],l2g,l2g)); 49 } 50 { 51 IS isx[2]; 52 Mat Bx00[4],Bx01[2],Bx10[2]; 53 Mat B00,B01,B10; 54 55 isx[0] = is0a; isx[1] = is0b; 56 Bx00[0] = B[0]; Bx00[1] = B[1]; Bx00[2] = B[3]; Bx00[3] = B[4]; 57 Bx01[0] = B[2]; Bx01[1] = B[5]; 58 Bx10[0] = B[6]; Bx10[1] = B[7]; 59 60 PetscCall(MatCreateNest(PETSC_COMM_WORLD,2,isx,2,isx,Bx00,&B00)); 61 PetscCall(MatSetUp(B00)); 62 PetscCall(MatCreateNest(PETSC_COMM_WORLD,2,isx,1,NULL,Bx01,&B01)); 63 PetscCall(MatSetUp(B01)); 64 PetscCall(MatCreateNest(PETSC_COMM_WORLD,1,NULL,2,isx,Bx10,&B10)); 65 PetscCall(MatSetUp(B10)); 66 { 67 Mat By[4]; 68 IS isy[2]; 69 70 By[0] = B00; By[1] = B01; By[2] = B10; By[3] = B[8]; 71 isy[0] = is0; isy[1] = is1; 72 73 PetscCall(MatCreateNest(PETSC_COMM_WORLD,2,isy,2,isy,By,&A)); 74 PetscCall(MatSetUp(A)); 75 } 76 PetscCall(MatDestroy(&B00)); 77 PetscCall(MatDestroy(&B01)); 78 PetscCall(MatDestroy(&B10)); 79 } 80 for (i=0; i<9; i++) PetscCall(MatDestroy(&B[i])); 81 PetscCall(ISLocalToGlobalMappingDestroy(&l2g)); 82 } else { 83 ISLocalToGlobalMapping l2g; 84 PetscInt l2gind[9]; 85 for (i=0; i<3; i++) for (j=0; j<3; j++) l2gind[3*i+j] = ((rank-1+j+size) % size)*3 + i; 86 PetscCall(ISLocalToGlobalMappingCreate(PETSC_COMM_WORLD,1,9,l2gind,PETSC_COPY_VALUES,&l2g)); 87 PetscCall(MatCreateAIJ(PETSC_COMM_WORLD,3,3,PETSC_DECIDE,PETSC_DECIDE,PETSC_DECIDE,NULL,PETSC_DECIDE,NULL,&A)); 88 PetscCall(MatSetLocalToGlobalMapping(A,l2g,l2g)); 89 PetscCall(ISLocalToGlobalMappingDestroy(&l2g)); 90 } 91 92 { 93 Mat A00,A11,A0a0a,A0a0b; 94 PetscCall(MatGetLocalSubMatrix(A,isl0,isl0,&A00)); 95 PetscCall(MatGetLocalSubMatrix(A,isl1,isl1,&A11)); 96 PetscCall(MatGetLocalSubMatrix(A00,isl0a,isl0a,&A0a0a)); 97 PetscCall(MatGetLocalSubMatrix(A00,isl0a,isl0b,&A0a0b)); 98 99 PetscCall(MatSetValueLocal(A0a0a,0,0,100*rank+1,ADD_VALUES)); 100 PetscCall(MatSetValueLocal(A0a0a,0,1,100*rank+2,ADD_VALUES)); 101 PetscCall(MatSetValueLocal(A0a0a,2,2,100*rank+9,ADD_VALUES)); 102 103 PetscCall(MatSetValueLocal(A0a0b,1,1,100*rank+50+5,ADD_VALUES)); 104 105 PetscCall(MatSetValueLocal(A11,0,0,1000*(rank+1)+1,ADD_VALUES)); 106 PetscCall(MatSetValueLocal(A11,1,2,1000*(rank+1)+6,ADD_VALUES)); 107 108 PetscCall(MatRestoreLocalSubMatrix(A00,isl0a,isl0a,&A0a0a)); 109 PetscCall(MatRestoreLocalSubMatrix(A00,isl0a,isl0b,&A0a0b)); 110 PetscCall(MatRestoreLocalSubMatrix(A,isl0,isl0,&A00)); 111 PetscCall(MatRestoreLocalSubMatrix(A,isl1,isl1,&A11)); 112 } 113 PetscCall(MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY)); 114 PetscCall(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY)); 115 116 PetscCall(MatComputeOperator(A,MATAIJ,&Aexplicit)); 117 PetscCall(MatView(Aexplicit,PETSC_VIEWER_STDOUT_WORLD)); 118 119 PetscCall(MatDestroy(&A)); 120 PetscCall(MatDestroy(&Aexplicit)); 121 PetscCall(ISDestroy(&is0a)); 122 PetscCall(ISDestroy(&is0b)); 123 PetscCall(ISDestroy(&is0)); 124 PetscCall(ISDestroy(&is1)); 125 PetscCall(ISDestroy(&isl0a)); 126 PetscCall(ISDestroy(&isl0b)); 127 PetscCall(ISDestroy(&isl0)); 128 PetscCall(ISDestroy(&isl1)); 129 PetscCall(PetscFinalize()); 130 return 0; 131 } 132 133 /*TEST 134 135 test: 136 nsize: 3 137 138 test: 139 suffix: nest 140 nsize: 3 141 args: -nest 142 143 TEST*/ 144