1 2 /* tests MatSeqSBAIJSetPreallocationCSR() and MatMPISBAIJSetPreallocationCSR() */ 3 4 #include <petsc.h> 5 6 int main(int argc, char **args) 7 { 8 PetscInt ia[3] = {0, 2, 4}; 9 PetscInt ja[4] = {0, 1, 0, 1}; 10 PetscScalar c[16] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}; 11 Mat ssbaij; 12 Mat msbaij; 13 14 PetscFunctionBeginUser; 15 PetscCall(PetscInitialize(&argc, &args, (char *)0, (char *)0)); 16 17 PetscCall(MatCreate(PETSC_COMM_SELF, &ssbaij)); 18 PetscCall(MatCreate(PETSC_COMM_SELF, &msbaij)); 19 PetscCall(MatSetType(ssbaij, MATSEQSBAIJ)); 20 PetscCall(MatSetType(msbaij, MATMPISBAIJ)); 21 PetscCall(MatSetBlockSize(ssbaij, 2)); 22 PetscCall(MatSetSizes(ssbaij, 4, 4, 4, 4)); 23 PetscCall(MatSetBlockSize(msbaij, 2)); 24 PetscCall(MatSetSizes(msbaij, 4, 4, 4, 4)); 25 PetscCall(MatSeqSBAIJSetPreallocationCSR(ssbaij, 2, ia, ja, c)); 26 PetscCall(MatMPISBAIJSetPreallocationCSR(msbaij, 2, ia, ja, c)); 27 PetscCall(MatView(ssbaij, PETSC_VIEWER_STDOUT_(PETSC_COMM_SELF))); 28 PetscCall(MatView(msbaij, PETSC_VIEWER_STDOUT_(PETSC_COMM_SELF))); 29 PetscCall(MatDestroy(&ssbaij)); 30 PetscCall(MatDestroy(&msbaij)); 31 PetscCall(PetscFinalize()); 32 return 0; 33 } 34 35 /*TEST 36 37 test: 38 filter: sed "s?\.??g" 39 40 TEST*/ 41