43 …rASCIIPrintf(viewer, "grid height=%d, grid width=%d\n", (int)a->grid->nprow, (int)a->grid->npcol));  in MatView_ScaLAPACK()
128 …, SCALAPACKinfog2l_(&gridx, &gcidx, a->desc, &a->grid->nprow, &a->grid->npcol, &a->grid->myrow, &a…  in MatSetValues_ScaLAPACK()
129       if (rsrc == a->grid->myrow && csrc == a->grid->mycol) {  in MatSetValues_ScaLAPACK()
172 …nit", SCALAPACKdescinit_(xdesc, &a->M, &one, &mb, &one, &zero, &zero, &a->grid->ictxcol, &xlld, &i…  in MatMultXXXYYY_ScaLAPACK()
177 …nit", SCALAPACKdescinit_(ydesc, &one, &a->N, &one, &nb, &zero, &zero, &a->grid->ictxrow, &ylld, &i…  in MatMultXXXYYY_ScaLAPACK()
181     lszx = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow);  in MatMultXXXYYY_ScaLAPACK()
182     lszy = SCALAPACKnumroc_(&a->N, &a->nb, &a->grid->mycol, &a->csrc, &a->grid->npcol);  in MatMultXXXYYY_ScaLAPACK()
187 …, SCALAPACKdescinit_(x2desc, &a->M, &one, &a->mb, &one, &zero, &zero, &a->grid->ictxt, &xlld, &inf…  in MatMultXXXYYY_ScaLAPACK()
189 …, SCALAPACKdescinit_(y2desc, &one, &a->N, &one, &a->nb, &zero, &zero, &a->grid->ictxt, &ylld, &inf…  in MatMultXXXYYY_ScaLAPACK()
193 …, SCALAPACKgemr2d_(&a->M, &one, x, &one, &one, xdesc, x2d, &one, &one, x2desc, &a->grid->ictxcol));  in MatMultXXXYYY_ScaLAPACK()
196 …, SCALAPACKgemr2d_(&one, &a->N, y, &one, &one, ydesc, y2d, &one, &one, y2desc, &a->grid->ictxrow));  in MatMultXXXYYY_ScaLAPACK()
203 …, SCALAPACKgemr2d_(&one, &a->N, y2d, &one, &one, y2desc, y, &one, &one, ydesc, &a->grid->ictxrow));  in MatMultXXXYYY_ScaLAPACK()
211 …nit", SCALAPACKdescinit_(xdesc, &one, &a->N, &one, &nb, &zero, &zero, &a->grid->ictxrow, &xlld, &i…  in MatMultXXXYYY_ScaLAPACK()
216 …nit", SCALAPACKdescinit_(ydesc, &a->M, &one, &mb, &one, &zero, &zero, &a->grid->ictxcol, &ylld, &i…  in MatMultXXXYYY_ScaLAPACK()
220     lszy = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow);  in MatMultXXXYYY_ScaLAPACK()
221     lszx = SCALAPACKnumroc_(&a->N, &a->nb, &a->grid->mycol, &a->csrc, &a->grid->npcol);  in MatMultXXXYYY_ScaLAPACK()
226 …, SCALAPACKdescinit_(x2desc, &one, &a->N, &one, &a->nb, &zero, &zero, &a->grid->ictxt, &xlld, &inf…  in MatMultXXXYYY_ScaLAPACK()
228 …, SCALAPACKdescinit_(y2desc, &a->M, &one, &a->mb, &one, &zero, &zero, &a->grid->ictxt, &ylld, &inf…  in MatMultXXXYYY_ScaLAPACK()
232 …, SCALAPACKgemr2d_(&one, &a->N, x, &one, &one, xdesc, x2d, &one, &one, x2desc, &a->grid->ictxrow));  in MatMultXXXYYY_ScaLAPACK()
235 …, SCALAPACKgemr2d_(&a->M, &one, y, &one, &one, ydesc, y2d, &one, &one, y2desc, &a->grid->ictxcol));  in MatMultXXXYYY_ScaLAPACK()
241 …, SCALAPACKgemr2d_(&a->M, &one, y2d, &one, &one, y2desc, y, &one, &one, ydesc, &a->grid->ictxcol));  in MatMultXXXYYY_ScaLAPACK()
446 …nit", SCALAPACKdescinit_(ddesc, &a->M, &one, &mb, &one, &zero, &zero, &a->grid->ictxcol, &dlld, &i…  in MatGetDiagonal_ScaLAPACK()
450     lszd = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow);  in MatGetDiagonal_ScaLAPACK()
455 …, SCALAPACKdescinit_(d2desc, &a->M, &one, &a->mb, &one, &zero, &zero, &a->grid->ictxt, &dlld, &inf…  in MatGetDiagonal_ScaLAPACK()
465 …LAPACKgemr2d_(&a->M, &one, d2d, &one, &one, d2desc, darray, &one, &one, ddesc, &a->grid->ictxcol));  in MatGetDiagonal_ScaLAPACK()
474 …nit", SCALAPACKdescinit_(ddesc, &one, &a->N, &one, &nb, &zero, &zero, &a->grid->ictxrow, &dlld, &i…  in MatGetDiagonal_ScaLAPACK()
478     lszd = SCALAPACKnumroc_(&a->N, &a->nb, &a->grid->mycol, &a->csrc, &a->grid->npcol);  in MatGetDiagonal_ScaLAPACK()
482 …, SCALAPACKdescinit_(d2desc, &one, &a->N, &one, &a->nb, &zero, &zero, &a->grid->ictxt, &dlld, &inf…  in MatGetDiagonal_ScaLAPACK()
492 …LAPACKgemr2d_(&one, &a->N, d2d, &one, &one, d2desc, darray, &one, &one, ddesc, &a->grid->ictxrow));  in MatGetDiagonal_ScaLAPACK()
517 …nit", SCALAPACKdescinit_(ddesc, &one, &a->N, &one, &nb, &zero, &zero, &a->grid->ictxrow, &dlld, &i…  in MatDiagonalScale_ScaLAPACK()
521     lszd = SCALAPACKnumroc_(&a->N, &a->nb, &a->grid->mycol, &a->csrc, &a->grid->npcol);  in MatDiagonalScale_ScaLAPACK()
525 …, SCALAPACKdescinit_(d2desc, &one, &a->N, &one, &a->nb, &zero, &zero, &a->grid->ictxt, &dlld, &inf…  in MatDiagonalScale_ScaLAPACK()
529 …, SCALAPACKgemr2d_(&one, &a->N, d, &one, &one, ddesc, d2d, &one, &one, d2desc, &a->grid->ictxrow));  in MatDiagonalScale_ScaLAPACK()
532     if (!a->grid->myrow) Cdgebs2d(a->grid->ictxt, "C", " ", 1, lszd, d2d, dlld);  in MatDiagonalScale_ScaLAPACK()
533     else Cdgebr2d(a->grid->ictxt, "C", " ", 1, lszd, d2d, dlld, 0, a->grid->mycol);  in MatDiagonalScale_ScaLAPACK()
548 …nit", SCALAPACKdescinit_(ddesc, &a->M, &one, &mb, &one, &zero, &zero, &a->grid->ictxcol, &dlld, &i…  in MatDiagonalScale_ScaLAPACK()
552     lszd = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow);  in MatDiagonalScale_ScaLAPACK()
557 …, SCALAPACKdescinit_(d2desc, &a->M, &one, &a->mb, &one, &zero, &zero, &a->grid->ictxt, &dlld, &inf…  in MatDiagonalScale_ScaLAPACK()
561 …, SCALAPACKgemr2d_(&a->M, &one, d, &one, &one, ddesc, d2d, &one, &one, d2desc, &a->grid->ictxcol));  in MatDiagonalScale_ScaLAPACK()
564     if (!a->grid->mycol) Cdgebs2d(a->grid->ictxt, "R", " ", lszd, 1, d2d, dlld);  in MatDiagonalScale_ScaLAPACK()
565     else Cdgebr2d(a->grid->ictxt, "R", " ", lszd, 1, d2d, dlld, a->grid->myrow, 0);  in MatDiagonalScale_ScaLAPACK()
723 …nit", SCALAPACKdescinit_(xdesc, &a->M, &one, &mb, &one, &zero, &zero, &a->grid->ictxcol, &xlld, &i…  in MatSolve_ScaLAPACK()
727   lszx = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow);  in MatSolve_ScaLAPACK()
732 …, SCALAPACKdescinit_(x2desc, &a->M, &one, &a->mb, &one, &zero, &zero, &a->grid->ictxt, &xlld, &inf…  in MatSolve_ScaLAPACK()
736 …, SCALAPACKgemr2d_(&a->M, &one, x, &one, &one, xdesc, x2d, &one, &one, x2desc, &a->grid->ictxcol));  in MatSolve_ScaLAPACK()
753 …, SCALAPACKgemr2d_(&a->M, &one, x2d, &one, &one, x2desc, x, &one, &one, xdesc, &a->grid->ictxcol));  in MatSolve_ScaLAPACK()
954     nproc = a->grid->nprow;  in MatGetOwnershipIS_ScaLAPACK()
955     iproc = a->grid->myrow;  in MatGetOwnershipIS_ScaLAPACK()
964     nproc = a->grid->npcol;  in MatGetOwnershipIS_ScaLAPACK()
965     iproc = a->grid->mycol;  in MatGetOwnershipIS_ScaLAPACK()
1014 …", SCALAPACKdescinit_(bdesc, &a->M, &a->N, &bmb, &a->N, &zero, &zero, &a->grid->ictxcol, &lld, &in…  in MatConvert_ScaLAPACK_Dense()
1019 …Kgemr2d_(&a->M, &a->N, a->loc, &one, &one, a->desc, barray, &one, &one, bdesc, &a->grid->ictxcol));  in MatConvert_ScaLAPACK_Dense()
1053 …", SCALAPACKdescinit_(bdesc, &a->M, &a->N, &bmb, &a->N, &zero, &zero, &a->grid->ictxcol, &lld, &in…  in MatConvert_ScaLAPACK_Dense()
1058 …Kgemr2d_(&a->M, &a->N, a->loc, &one, &one, a->desc, barray, &one, &one, bdesc, &a->grid->ictxcol));  in MatConvert_ScaLAPACK_Dense()
1126 …", SCALAPACKdescinit_(adesc, &b->M, &b->N, &amb, &b->N, &zero, &zero, &b->grid->ictxcol, &lld, &in…  in MatConvert_Dense_ScaLAPACK()
1130 …Kgemr2d_(&b->M, &b->N, aarray, &one, &one, adesc, b->loc, &one, &one, b->desc, &b->grid->ictxcol));  in MatConvert_Dense_ScaLAPACK()
1249   Mat_ScaLAPACK_Grid *grid;  in MatDestroy_ScaLAPACK()  local
1258   PetscCallMPI(MPI_Comm_get_attr(icomm, Petsc_ScaLAPACK_keyval, (void **)&grid, &iflg));  in MatDestroy_ScaLAPACK()
1259   if (--grid->grid_refct == 0) {  in MatDestroy_ScaLAPACK()
1260     Cblacs_gridexit(grid->ictxt);  in MatDestroy_ScaLAPACK()
1261     Cblacs_gridexit(grid->ictxrow);  in MatDestroy_ScaLAPACK()
1262     Cblacs_gridexit(grid->ictxcol);  in MatDestroy_ScaLAPACK()
1263     PetscCall(PetscFree(grid));  in MatDestroy_ScaLAPACK()
1294   a->locr = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow);  in MatSetUp_ScaLAPACK()
1295   a->locc = SCALAPACKnumroc_(&a->N, &a->nb, &a->grid->mycol, &a->csrc, &a->grid->npcol);  in MatSetUp_ScaLAPACK()
1302 …Kdescinit_(a->desc, &a->M, &a->N, &a->mb, &a->nb, &a->rsrc, &a->csrc, &a->grid->ictxt, &a->lld, &i…  in MatSetUp_ScaLAPACK()
1335 …, SCALAPACKinfog2l_(&gridx, &gcidx, a->desc, &a->grid->nprow, &a->grid->npcol, &a->grid->myrow, &a…  in MatAssemblyEnd_ScaLAPACK()
1336 …PetscCheck(rsrc == a->grid->myrow && csrc == a->grid->mycol, PetscObjectComm((PetscObject)A), PETS…  in MatAssemblyEnd_ScaLAPACK()
1550 …, SCALAPACKinfog2l_(&gridx, &gcidx, a->desc, &a->grid->nprow, &a->grid->npcol, &a->grid->myrow, &a…  in MatStashScatterBegin_ScaLAPACK()
1551       j = Cblacs_pnum(a->grid->ictxt, rsrc, csrc);  in MatStashScatterBegin_ScaLAPACK()
1778   Mat_ScaLAPACK_Grid *grid;  in MatCreate_ScaLAPACK()  local
1804   PetscCallMPI(MPI_Comm_get_attr(icomm, Petsc_ScaLAPACK_keyval, (void **)&grid, &iflg));  in MatCreate_ScaLAPACK()
1806     PetscCall(PetscNew(&grid));  in MatCreate_ScaLAPACK()
1809     PetscCall(PetscBLASIntCast(PetscSqrtReal((PetscReal)size) + 0.001, &grid->nprow));  in MatCreate_ScaLAPACK()
1812 …PetscCall(PetscOptionsInt("-mat_scalapack_grid_height", "Grid Height", "None", grid->nprow, &optv1…  in MatCreate_ScaLAPACK()
1815       PetscCall(PetscBLASIntCast(optv1, &grid->nprow));  in MatCreate_ScaLAPACK()
1819     if (size % grid->nprow) grid->nprow = 1; /* cannot use a squarish grid, use a 1d grid */  in MatCreate_ScaLAPACK()
1820     grid->npcol = size / grid->nprow;  in MatCreate_ScaLAPACK()
1821     PetscCall(PetscBLASIntCast(grid->nprow, &nprow));  in MatCreate_ScaLAPACK()
1822     PetscCall(PetscBLASIntCast(grid->npcol, &npcol));  in MatCreate_ScaLAPACK()
1823     grid->ictxt = Csys2blacs_handle(icomm);  in MatCreate_ScaLAPACK()
1824     Cblacs_gridinit(&grid->ictxt, "R", nprow, npcol);  in MatCreate_ScaLAPACK()
1825     Cblacs_gridinfo(grid->ictxt, &nprow, &npcol, &myrow, &mycol);  in MatCreate_ScaLAPACK()
1826     grid->grid_refct = 1;  in MatCreate_ScaLAPACK()
1827     grid->nprow      = nprow;  in MatCreate_ScaLAPACK()
1828     grid->npcol      = npcol;  in MatCreate_ScaLAPACK()
1829     grid->myrow      = myrow;  in MatCreate_ScaLAPACK()
1830     grid->mycol      = mycol;  in MatCreate_ScaLAPACK()
1832     grid->ictxrow = Csys2blacs_handle(icomm);  in MatCreate_ScaLAPACK()
1833     Cblacs_gridinit(&grid->ictxrow, "R", 1, size);  in MatCreate_ScaLAPACK()
1834     grid->ictxcol = Csys2blacs_handle(icomm);  in MatCreate_ScaLAPACK()
1835     Cblacs_gridinit(&grid->ictxcol, "R", size, 1);  in MatCreate_ScaLAPACK()
1836     PetscCallMPI(MPI_Comm_set_attr(icomm, Petsc_ScaLAPACK_keyval, (void *)grid));  in MatCreate_ScaLAPACK()
1838   } else grid->grid_refct++;  in MatCreate_ScaLAPACK()
1840   a->grid = grid;  in MatCreate_ScaLAPACK()