1c4762a1bSJed Brown 2c4762a1bSJed Brown static char help[] = "Test saving SeqSBAIJ matrix that is missing diagonal entries."; 3c4762a1bSJed Brown 4c4762a1bSJed Brown #include <petscmat.h> 5c4762a1bSJed Brown 6*d71ae5a4SJacob Faibussowitsch int main(int argc, char **args) 7*d71ae5a4SJacob Faibussowitsch { 8c4762a1bSJed Brown Mat A; 9c4762a1bSJed Brown PetscInt bs = 3, m = 4, i, j, val = 10, row[2], col[3], rstart; 10c4762a1bSJed Brown PetscMPIInt size; 11c4762a1bSJed Brown PetscScalar x[6][9]; 12c4762a1bSJed Brown 13327415f7SBarry Smith PetscFunctionBeginUser; 149566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc, &args, (char *)0, help)); 159566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size)); 16be096a46SBarry Smith PetscCheck(size == 1, PETSC_COMM_WORLD, PETSC_ERR_WRONG_MPI_SIZE, "Test is only for sequential"); 179566063dSJacob Faibussowitsch PetscCall(MatCreateSeqSBAIJ(PETSC_COMM_SELF, bs, m * bs, m * bs, 1, NULL, &A)); 189566063dSJacob Faibussowitsch PetscCall(MatSetOption(A, MAT_IGNORE_LOWER_TRIANGULAR, PETSC_TRUE)); 19c4762a1bSJed Brown rstart = 0; 20c4762a1bSJed Brown 219371c9d4SSatish Balay row[0] = rstart + 0; 229371c9d4SSatish Balay row[1] = rstart + 2; 239371c9d4SSatish Balay col[0] = rstart + 0; 249371c9d4SSatish Balay col[1] = rstart + 1; 259371c9d4SSatish Balay col[2] = rstart + 3; 26c4762a1bSJed Brown for (i = 0; i < 6; i++) { 27c4762a1bSJed Brown for (j = 0; j < 9; j++) x[i][j] = (PetscScalar)val++; 28c4762a1bSJed Brown } 299566063dSJacob Faibussowitsch PetscCall(MatSetValuesBlocked(A, 2, row, 3, col, &x[0][0], INSERT_VALUES)); 309566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY)); 319566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY)); 329566063dSJacob Faibussowitsch PetscCall(MatView(A, PETSC_VIEWER_BINARY_WORLD)); 33c4762a1bSJed Brown 349566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A)); 359566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 36b122ec5aSJacob Faibussowitsch return 0; 37c4762a1bSJed Brown } 38