1 static char help[] = "Test MatZeroEntries() on unassembled matrices \n\n"; 2 3 #include <petscmat.h> 4 5 int main(int argc, char **args) 6 { 7 Mat A; 8 PetscInt N = 32; 9 MPI_Comm comm; 10 PetscErrorCode ierr; 11 12 ierr = PetscInitialize(&argc, &args, (char*) 0, help);if (ierr) return ierr; 13 comm = PETSC_COMM_WORLD; 14 CHKERRQ(PetscOptionsGetInt(NULL,NULL, "-N", &N, NULL)); 15 CHKERRQ(MatCreate(comm, &A)); 16 CHKERRQ(MatSetSizes(A, PETSC_DECIDE, PETSC_DECIDE, N, N)); 17 CHKERRQ(MatSetFromOptions(A)); 18 CHKERRQ(MatSeqAIJSetPreallocation(A, 3, NULL)); 19 CHKERRQ(MatMPIAIJSetPreallocation(A, 3, NULL, 2, NULL)); 20 CHKERRQ(MatZeroEntries(A)); 21 CHKERRQ(MatDestroy(&A)); 22 ierr = PetscFinalize(); 23 return ierr; 24 } 25 26 /*TEST 27 test: 28 requires: kokkos_kernels 29 nsize: {{1 2}} 30 output_file: output/ex252_1.out 31 args: -mat_type aijkokkos 32 TEST*/ 33