1 static char help[] = "Tests MatTranspose() with MAT_REUSE_MATRIX and different nonzero pattern\n\n"; 2 3 #include <petscmat.h> 4 5 int main(int argc, char **argv) 6 { 7 Mat A, B; 8 PetscMPIInt size; 9 10 PetscCall(PetscInitialize(&argc, &argv, (char *)0, help)); 11 PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size)); 12 PetscCheck(size == 1, PETSC_COMM_WORLD, PETSC_ERR_WRONG_MPI_SIZE, "This is a uniprocessor example only!"); 13 PetscCall(MatCreateSeqAIJ(PETSC_COMM_SELF, 2, 2, 2, NULL, &A)); 14 PetscCall(MatSetValue(A, 0, 0, 1.0, INSERT_VALUES)); 15 PetscCall(MatSetValue(A, 0, 1, 2.0, INSERT_VALUES)); 16 PetscCall(MatSetValue(A, 1, 1, 4.0, INSERT_VALUES)); 17 PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY)); 18 PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY)); 19 PetscCall(MatView(A, PETSC_VIEWER_STDOUT_SELF)); 20 PetscCall(MatTranspose(A, MAT_INITIAL_MATRIX, &B)); 21 PetscCall(MatView(B, PETSC_VIEWER_STDOUT_SELF)); 22 PetscCall(MatSetOption(A, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_FALSE)); 23 PetscCall(MatSetValue(A, 1, 0, 3.0, INSERT_VALUES)); 24 PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY)); 25 PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY)); 26 PetscCall(MatView(A, PETSC_VIEWER_STDOUT_SELF)); 27 PetscCall(MatTranspose(A, MAT_REUSE_MATRIX, &B)); 28 PetscCall(MatView(B, PETSC_VIEWER_STDOUT_SELF)); 29 PetscCall(MatDestroy(&A)); 30 PetscCall(MatDestroy(&B)); 31 PetscCall(PetscFinalize()); 32 return 0; 33 } 34 35 /*TEST 36 37 test: 38 39 TEST*/ 40