1c4762a1bSJed Brown static char help[] = "Tests MatSetValuesBlockedStencil() in 3d.\n\n"; 2c4762a1bSJed Brown 3c4762a1bSJed Brown #include <petscmat.h> 4c4762a1bSJed Brown #include <petscdm.h> 5c4762a1bSJed Brown #include <petscdmda.h> 6c4762a1bSJed Brown 7d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv) 8d71ae5a4SJacob Faibussowitsch { 9c4762a1bSJed Brown PetscInt M = 3, N = 4, P = 2, s = 1, w = 2, i, m = PETSC_DECIDE, n = PETSC_DECIDE, p = PETSC_DECIDE; 10c4762a1bSJed Brown DM da; 11c4762a1bSJed Brown Mat mat; 12c4762a1bSJed Brown DMDAStencilType stencil_type = DMDA_STENCIL_BOX; 13c4762a1bSJed Brown PetscBool flg = PETSC_FALSE; 14c4762a1bSJed Brown MatStencil idx[2], idy[2]; 15c4762a1bSJed Brown PetscScalar *values; 16c4762a1bSJed Brown 17327415f7SBarry Smith PetscFunctionBeginUser; 18c8025a54SPierre Jolivet PetscCall(PetscInitialize(&argc, &argv, NULL, help)); 199566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-M", &M, NULL)); 209566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-N", &N, NULL)); 219566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-P", &P, NULL)); 229566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-m", &m, NULL)); 239566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-n", &n, NULL)); 249566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-p", &p, NULL)); 259566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-s", &s, NULL)); 269566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-w", &w, NULL)); 279566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetBool(NULL, NULL, "-star", &flg, NULL)); 28c4762a1bSJed Brown if (flg) stencil_type = DMDA_STENCIL_STAR; 29c4762a1bSJed Brown 30c4762a1bSJed Brown /* Create distributed array and get vectors */ 319566063dSJacob Faibussowitsch PetscCall(DMDACreate3d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, stencil_type, M, N, P, m, n, p, w, s, 0, 0, 0, &da)); 329566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(da)); 339566063dSJacob Faibussowitsch PetscCall(DMSetUp(da)); 349566063dSJacob Faibussowitsch PetscCall(DMCreateMatrix(da, &mat)); 3592e3ebb5SRichard Tran Mills PetscCall(MatSetFromOptions(mat)); 36c4762a1bSJed Brown 379371c9d4SSatish Balay idx[0].i = 1; 389371c9d4SSatish Balay idx[0].j = 1; 399371c9d4SSatish Balay idx[0].k = 0; 409371c9d4SSatish Balay idx[1].i = 2; 419371c9d4SSatish Balay idx[1].j = 1; 429371c9d4SSatish Balay idx[1].k = 0; 439371c9d4SSatish Balay idy[0].i = 1; 449371c9d4SSatish Balay idy[0].j = 2; 459371c9d4SSatish Balay idy[0].k = 0; 469371c9d4SSatish Balay idy[1].i = 2; 479371c9d4SSatish Balay idy[1].j = 2; 489371c9d4SSatish Balay idy[1].k = 0; 499566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(2 * 2 * w * w, &values)); 50c4762a1bSJed Brown for (i = 0; i < 2 * 2 * w * w; i++) values[i] = i; 519566063dSJacob Faibussowitsch PetscCall(MatSetValuesBlockedStencil(mat, 2, idx, 2, idy, values, INSERT_VALUES)); 529566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(mat, MAT_FINAL_ASSEMBLY)); 539566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(mat, MAT_FINAL_ASSEMBLY)); 54c4762a1bSJed Brown 5592e3ebb5SRichard Tran Mills flg = PETSC_FALSE; 5692e3ebb5SRichard Tran Mills PetscCall(PetscOptionsGetBool(NULL, NULL, "-test_matdiagonalscalelocal", &flg, NULL)); 5792e3ebb5SRichard Tran Mills if (flg) { 58*7c590497SRichard Tran Mills Vec vec; 5992e3ebb5SRichard Tran Mills PetscInt size; 6092e3ebb5SRichard Tran Mills PetscMPIInt rank; 6192e3ebb5SRichard Tran Mills 6292e3ebb5SRichard Tran Mills PetscCall(DMGetLocalVector(da, &vec)); 63*7c590497SRichard Tran Mills PetscCall(VecGetLocalSize(vec, &size)); 6492e3ebb5SRichard Tran Mills PetscCall(PetscFree(values)); 65*7c590497SRichard Tran Mills PetscCall(VecGetArrayWrite(vec, &values)); 6692e3ebb5SRichard Tran Mills PetscCallMPI(MPI_Comm_rank(MPI_COMM_WORLD, &rank)); 6792e3ebb5SRichard Tran Mills for (i = 0; i < size; i++) values[i] = (PetscScalar)(rank + 1); 68*7c590497SRichard Tran Mills PetscCall(VecRestoreArrayWrite(vec, &values)); 69*7c590497SRichard Tran Mills PetscCall(MatDiagonalScaleLocal(mat, vec)); 7092e3ebb5SRichard Tran Mills PetscCall(DMRestoreLocalVector(da, &vec)); 7192e3ebb5SRichard Tran Mills } 7292e3ebb5SRichard Tran Mills 73c4762a1bSJed Brown /* Free memory */ 749566063dSJacob Faibussowitsch PetscCall(PetscFree(values)); 759566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mat)); 769566063dSJacob Faibussowitsch PetscCall(DMDestroy(&da)); 779566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 78b122ec5aSJacob Faibussowitsch return 0; 79c4762a1bSJed Brown } 8092e3ebb5SRichard Tran Mills 8192e3ebb5SRichard Tran Mills /*TEST 8292e3ebb5SRichard Tran Mills 8392e3ebb5SRichard Tran Mills test: 8492e3ebb5SRichard Tran Mills suffix: baij 8592e3ebb5SRichard Tran Mills nsize: 2 8692e3ebb5SRichard Tran Mills args: -mat_type baij 8792e3ebb5SRichard Tran Mills output_file: output/empty.out 8892e3ebb5SRichard Tran Mills 8992e3ebb5SRichard Tran Mills test: 9092e3ebb5SRichard Tran Mills suffix: aij 9192e3ebb5SRichard Tran Mills nsize: 2 9292e3ebb5SRichard Tran Mills args: -mat_type aij 9392e3ebb5SRichard Tran Mills output_file: output/empty.out 9492e3ebb5SRichard Tran Mills 9592e3ebb5SRichard Tran Mills test: 9692e3ebb5SRichard Tran Mills suffix: baij-diagonalscalelocal 9792e3ebb5SRichard Tran Mills nsize: 2 9892e3ebb5SRichard Tran Mills args: -mat_type baij -test_matdiagonalscalelocal 9992e3ebb5SRichard Tran Mills output_file: output/empty.out 10092e3ebb5SRichard Tran Mills 10192e3ebb5SRichard Tran Mills test: 10292e3ebb5SRichard Tran Mills suffix: aij-diagonalscalelocal 10392e3ebb5SRichard Tran Mills nsize: 2 10492e3ebb5SRichard Tran Mills args: -mat_type aij -test_matdiagonalscalelocal 10592e3ebb5SRichard Tran Mills output_file: output/empty.out 10692e3ebb5SRichard Tran Mills 10792e3ebb5SRichard Tran Mills TEST*/ 108