Lines Matching refs:a
32 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatView_ScaLAPACK() local
42 PetscCall(PetscViewerASCIIPrintf(viewer, "block sizes: %d,%d\n", (int)a->mb, (int)a->nb)); in MatView_ScaLAPACK()
43 …iewerASCIIPrintf(viewer, "grid height=%d, grid width=%d\n", (int)a->grid->nprow, (int)a->grid->npc… in MatView_ScaLAPACK()
44 …wer, "coordinates of process owning first row and column: (%d,%d)\n", (int)a->rsrc, (int)a->csrc)); in MatView_ScaLAPACK()
45 …erASCIIPrintf(viewer, "dimension of largest local matrix: %d x %d\n", (int)a->locr, (int)a->locc)); in MatView_ScaLAPACK()
60 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatGetInfo_ScaLAPACK() local
66 isend[0] = a->lld * a->locc; /* locally allocated */ in MatGetInfo_ScaLAPACK()
67 isend[1] = a->locr * a->locc; /* used submatrix */ in MatGetInfo_ScaLAPACK()
93 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatSetOption_ScaLAPACK() local
105 a->roworiented = flg; in MatSetOption_ScaLAPACK()
115 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatSetValues_ScaLAPACK() local
118 PetscBool roworiented = a->roworiented; in MatSetValues_ScaLAPACK()
128 …Kinfog2l", SCALAPACKinfog2l_(&gridx, &gcidx, a->desc, &a->grid->nprow, &a->grid->npcol, &a->grid->… in MatSetValues_ScaLAPACK()
129 if (rsrc == a->grid->myrow && csrc == a->grid->mycol) { in MatSetValues_ScaLAPACK()
133 a->loc[lridx - 1 + (lcidx - 1) * a->lld] = vals[i * nc + j]; in MatSetValues_ScaLAPACK()
136 a->loc[lridx - 1 + (lcidx - 1) * a->lld] += vals[i * nc + j]; in MatSetValues_ScaLAPACK()
142 a->loc[lridx - 1 + (lcidx - 1) * a->lld] = vals[i + j * nr]; in MatSetValues_ScaLAPACK()
145 a->loc[lridx - 1 + (lcidx - 1) * a->lld] += vals[i + j * nr]; in MatSetValues_ScaLAPACK()
161 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatMultXXXYYY_ScaLAPACK() local
172 …CallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(xdesc, &a->M, &one, &mb, &one, &zero, &zero, &a->… in MatMultXXXYYY_ScaLAPACK()
177 …lBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(ydesc, &one, &a->N, &one, &nb, &zero, &zero, &a->gri… 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 …lBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(x2desc, &a->M, &one, &a->mb, &one, &zero, &zero, &a-… in MatMultXXXYYY_ScaLAPACK()
189 …AS("SCALAPACKdescinit", SCALAPACKdescinit_(y2desc, &one, &a->N, &one, &a->nb, &zero, &zero, &a->gr… in MatMultXXXYYY_ScaLAPACK()
193 …CallBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&a->M, &one, x, &one, &one, xdesc, x2d, &one, &one, x… in MatMultXXXYYY_ScaLAPACK()
196 …lBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&one, &a->N, y, &one, &one, ydesc, y2d, &one, &one, y2de… in MatMultXXXYYY_ScaLAPACK()
199 …hermitian) PetscCallBLAS("PBLASgemv", PBLASgemv_("C", &a->M, &a->N, &alpha, a->loc, &one, &one, a-… in MatMultXXXYYY_ScaLAPACK()
200 …else PetscCallBLAS("PBLASgemv", PBLASgemv_("T", &a->M, &a->N, &alpha, a->loc, &one, &one, a->desc,… in MatMultXXXYYY_ScaLAPACK()
203 …lBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&one, &a->N, y2d, &one, &one, y2desc, y, &one, &one, yde… in MatMultXXXYYY_ScaLAPACK()
211 …lBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(xdesc, &one, &a->N, &one, &nb, &zero, &zero, &a->gri… in MatMultXXXYYY_ScaLAPACK()
216 …CallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(ydesc, &a->M, &one, &mb, &one, &zero, &zero, &a->… 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 …AS("SCALAPACKdescinit", SCALAPACKdescinit_(x2desc, &one, &a->N, &one, &a->nb, &zero, &zero, &a->gr… in MatMultXXXYYY_ScaLAPACK()
228 …lBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(y2desc, &a->M, &one, &a->mb, &one, &zero, &zero, &a-… in MatMultXXXYYY_ScaLAPACK()
232 …lBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&one, &a->N, x, &one, &one, xdesc, x2d, &one, &one, x2de… in MatMultXXXYYY_ScaLAPACK()
235 …CallBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&a->M, &one, y, &one, &one, ydesc, y2d, &one, &one, y… in MatMultXXXYYY_ScaLAPACK()
238 …PetscCallBLAS("PBLASgemv", PBLASgemv_("N", &a->M, &a->N, &alpha, a->loc, &one, &one, a->desc, x2d,… in MatMultXXXYYY_ScaLAPACK()
241 …CallBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&a->M, &one, y2d, &one, &one, y2desc, y, &one, &one, … in MatMultXXXYYY_ScaLAPACK()
336 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatMatMultNumeric_ScaLAPACK() local
343 …PetscCallBLAS("PBLASgemm", PBLASgemm_("N", "N", &a->M, &b->N, &a->N, &sone, a->loc, &one, &one, a-… in MatMatMultNumeric_ScaLAPACK()
360 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatTransposeMatMultNumeric_ScaLAPACK() local
367 …PetscCallBLAS("PBLASgemm", PBLASgemm_("T", "N", &a->N, &b->N, &a->M, &sone, a->loc, &one, &one, a-… in MatTransposeMatMultNumeric_ScaLAPACK()
384 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatMatTransposeMultNumeric_ScaLAPACK() local
391 …PetscCallBLAS("PBLASgemm", PBLASgemm_("N", "T", &a->M, &b->M, &a->N, &sone, a->loc, &one, &one, a-… in MatMatTransposeMultNumeric_ScaLAPACK()
432 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatGetDiagonal_ScaLAPACK() local
446 …CallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(ddesc, &a->M, &one, &mb, &one, &zero, &zero, &a->… in MatGetDiagonal_ScaLAPACK()
450 lszd = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow); in MatGetDiagonal_ScaLAPACK()
455 …lBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(d2desc, &a->M, &one, &a->mb, &one, &zero, &zero, &a-… in MatGetDiagonal_ScaLAPACK()
459 for (j = 1; j <= a->M; j++) { in MatGetDiagonal_ScaLAPACK()
460 PetscCallBLAS("SCALAPACKelget", SCALAPACKelget_("R", " ", &v, a->loc, &j, &j, a->desc)); in MatGetDiagonal_ScaLAPACK()
465 …lBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&a->M, &one, d2d, &one, &one, d2desc, darray, &one, &one… in MatGetDiagonal_ScaLAPACK()
474 …lBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(ddesc, &one, &a->N, &one, &nb, &zero, &zero, &a->gri… in MatGetDiagonal_ScaLAPACK()
478 lszd = SCALAPACKnumroc_(&a->N, &a->nb, &a->grid->mycol, &a->csrc, &a->grid->npcol); in MatGetDiagonal_ScaLAPACK()
482 …AS("SCALAPACKdescinit", SCALAPACKdescinit_(d2desc, &one, &a->N, &one, &a->nb, &zero, &zero, &a->gr… in MatGetDiagonal_ScaLAPACK()
486 for (j = 1; j <= a->N; j++) { in MatGetDiagonal_ScaLAPACK()
487 PetscCallBLAS("SCALAPACKelget", SCALAPACKelget_("C", " ", &v, a->loc, &j, &j, a->desc)); in MatGetDiagonal_ScaLAPACK()
492 …AS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&one, &a->N, d2d, &one, &one, d2desc, darray, &one, &one, d… in MatGetDiagonal_ScaLAPACK()
504 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatDiagonalScale_ScaLAPACK() local
517 …lBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(ddesc, &one, &a->N, &one, &nb, &zero, &zero, &a->gri… in MatDiagonalScale_ScaLAPACK()
521 lszd = SCALAPACKnumroc_(&a->N, &a->nb, &a->grid->mycol, &a->csrc, &a->grid->npcol); in MatDiagonalScale_ScaLAPACK()
525 …AS("SCALAPACKdescinit", SCALAPACKdescinit_(d2desc, &one, &a->N, &one, &a->nb, &zero, &zero, &a->gr… in MatDiagonalScale_ScaLAPACK()
529 …lBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&one, &a->N, d, &one, &one, ddesc, d2d, &one, &one, d2de… 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()
536 for (j = 0; j < a->locc; j++) in MatDiagonalScale_ScaLAPACK()
537 for (i = 0; i < a->locr; i++) a->loc[i + j * a->lld] *= d2d[j]; in MatDiagonalScale_ScaLAPACK()
548 …CallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(ddesc, &a->M, &one, &mb, &one, &zero, &zero, &a->… in MatDiagonalScale_ScaLAPACK()
552 lszd = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow); in MatDiagonalScale_ScaLAPACK()
557 …lBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(d2desc, &a->M, &one, &a->mb, &one, &zero, &zero, &a-… in MatDiagonalScale_ScaLAPACK()
561 …CallBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&a->M, &one, d, &one, &one, ddesc, d2d, &one, &one, d… 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()
568 for (i = 0; i < a->locr; i++) in MatDiagonalScale_ScaLAPACK()
569 for (j = 0; j < a->locc; j++) a->loc[i + j * a->lld] *= d2d[i]; in MatDiagonalScale_ScaLAPACK()
577 static PetscErrorCode MatScale_ScaLAPACK(Mat X, PetscScalar a) in MatScale_ScaLAPACK() argument
584 PetscCallBLAS("BLASscal", BLASscal_(&n, &a, x->loc, &one)); in MatScale_ScaLAPACK()
620 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatCopy_ScaLAPACK() local
624 PetscCall(PetscArraycpy(b->loc, a->loc, a->lld * a->locc)); in MatCopy_ScaLAPACK()
633 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data, *b; in MatDuplicate_ScaLAPACK() local
641 b->M = a->M; in MatDuplicate_ScaLAPACK()
642 b->N = a->N; in MatDuplicate_ScaLAPACK()
643 b->mb = a->mb; in MatDuplicate_ScaLAPACK()
644 b->nb = a->nb; in MatDuplicate_ScaLAPACK()
645 b->rsrc = a->rsrc; in MatDuplicate_ScaLAPACK()
646 b->csrc = a->csrc; in MatDuplicate_ScaLAPACK()
649 if (op == MAT_COPY_VALUES) PetscCall(PetscArraycpy(b->loc, a->loc, a->lld * a->locc)); in MatDuplicate_ScaLAPACK()
656 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data, *b; in MatTranspose_ScaLAPACK() local
667 …cCall(MatCreateScaLAPACK(PetscObjectComm((PetscObject)A), a->nb, a->mb, a->N, a->M, a->csrc, a->rs… in MatTranspose_ScaLAPACK()
670 …PetscCallBLAS("PBLAStran", PBLAStran_(&a->N, &a->M, &sone, a->loc, &one, &one, a->desc, &zero, b->… in MatTranspose_ScaLAPACK()
682 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatConjugate_ScaLAPACK() local
686 for (i = 0; i < a->locr; i++) in MatConjugate_ScaLAPACK()
687 for (j = 0; j < a->locc; j++) a->loc[i + j * a->lld] = PetscConj(a->loc[i + j * a->lld]); in MatConjugate_ScaLAPACK()
693 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data, *b; in MatHermitianTranspose_ScaLAPACK() local
700 …cCall(MatCreateScaLAPACK(PetscObjectComm((PetscObject)A), a->nb, a->mb, a->N, a->M, a->csrc, a->rs… in MatHermitianTranspose_ScaLAPACK()
703 …PetscCallBLAS("PBLAStran", PBLAStran_(&a->N, &a->M, &sone, a->loc, &one, &one, a->desc, &zero, b->… in MatHermitianTranspose_ScaLAPACK()
710 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatSolve_ScaLAPACK() local
723 …CallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(xdesc, &a->M, &one, &mb, &one, &zero, &zero, &a->… in MatSolve_ScaLAPACK()
727 lszx = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow); in MatSolve_ScaLAPACK()
732 …lBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(x2desc, &a->M, &one, &a->mb, &one, &zero, &zero, &a-… in MatSolve_ScaLAPACK()
736 …CallBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&a->M, &one, x, &one, &one, xdesc, x2d, &one, &one, x… in MatSolve_ScaLAPACK()
741 …PetscCallBLAS("SCALAPACKgetrs", SCALAPACKgetrs_("N", &a->M, &nrhs, a->loc, &one, &one, a->desc, a-… in MatSolve_ScaLAPACK()
745 …PetscCallBLAS("SCALAPACKpotrs", SCALAPACKpotrs_("L", &a->M, &nrhs, a->loc, &one, &one, a->desc, x2… in MatSolve_ScaLAPACK()
753 …CallBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&a->M, &one, x2d, &one, &one, x2desc, x, &one, &one, … in MatSolve_ScaLAPACK()
770 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data, *x; in MatMatSolve_ScaLAPACK() local
791 …PetscCallBLAS("SCALAPACKgetrs", SCALAPACKgetrs_("N", &a->M, &x->N, a->loc, &one, &one, a->desc, a-… in MatMatSolve_ScaLAPACK()
795 …PetscCallBLAS("SCALAPACKpotrs", SCALAPACKpotrs_("L", &a->M, &x->N, a->loc, &one, &one, a->desc, x-… in MatMatSolve_ScaLAPACK()
811 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatLUFactor_ScaLAPACK() local
815 if (!a->pivots) PetscCall(PetscMalloc1(a->locr + a->mb, &a->pivots)); in MatLUFactor_ScaLAPACK()
816 …PetscCallBLAS("SCALAPACKgetrf", SCALAPACKgetrf_(&a->M, &a->N, a->loc, &one, &one, a->desc, a->pivo… in MatLUFactor_ScaLAPACK()
843 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatCholeskyFactor_ScaLAPACK() local
847 PetscCallBLAS("SCALAPACKpotrf", SCALAPACKpotrf_("L", &a->M, a->loc, &one, &one, a->desc, &info)); in MatCholeskyFactor_ScaLAPACK()
882 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatGetFactor_scalapack_scalapack() local
886 …cCall(MatCreateScaLAPACK(PetscObjectComm((PetscObject)A), a->mb, a->nb, a->M, a->N, a->rsrc, a->cs… in MatGetFactor_scalapack_scalapack()
907 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatNorm_ScaLAPACK() local
916 lwork = PetscMax(a->locr, a->locc); in MatNorm_ScaLAPACK()
924 lwork = PetscMax(a->locr, a->locc); in MatNorm_ScaLAPACK()
930 *nrm = SCALAPACKlange_(ntype, &a->M, &a->N, a->loc, &one, &one, a->desc, work); in MatNorm_ScaLAPACK()
937 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatZeroEntries_ScaLAPACK() local
940 PetscCall(PetscArrayzero(a->loc, a->lld * a->locc)); in MatZeroEntries_ScaLAPACK()
946 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatGetOwnershipIS_ScaLAPACK() local
951 n = a->locr; in MatGetOwnershipIS_ScaLAPACK()
952 nb = a->mb; in MatGetOwnershipIS_ScaLAPACK()
953 isrc = a->rsrc; in MatGetOwnershipIS_ScaLAPACK()
954 nproc = a->grid->nprow; in MatGetOwnershipIS_ScaLAPACK()
955 iproc = a->grid->myrow; in MatGetOwnershipIS_ScaLAPACK()
961 n = a->locc; in MatGetOwnershipIS_ScaLAPACK()
962 nb = a->nb; in MatGetOwnershipIS_ScaLAPACK()
963 isrc = a->csrc; in MatGetOwnershipIS_ScaLAPACK()
964 nproc = a->grid->npcol; in MatGetOwnershipIS_ScaLAPACK()
965 iproc = a->grid->mycol; in MatGetOwnershipIS_ScaLAPACK()
975 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatConvert_ScaLAPACK_Dense() local
1014 …llBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(bdesc, &a->M, &a->N, &bmb, &a->N, &zero, &zero, &a-… in MatConvert_ScaLAPACK_Dense()
1019 …LAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&a->M, &a->N, a->loc, &one, &one, a->desc, barray, &one, &… in MatConvert_ScaLAPACK_Dense()
1053 …llBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(bdesc, &a->M, &a->N, &bmb, &a->N, &zero, &zero, &a-… in MatConvert_ScaLAPACK_Dense()
1058 …LAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&a->M, &a->N, a->loc, &one, &one, a->desc, barray, &one, &… in MatConvert_ScaLAPACK_Dense()
1230 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatScaLAPACKSetPreallocation() local
1236 if (!a->lld) a->lld = a->locr; in MatScaLAPACKSetPreallocation()
1238 PetscCall(PetscFree(a->loc)); in MatScaLAPACKSetPreallocation()
1239 PetscCall(PetscIntMultError(a->lld, a->locc, &sz)); in MatScaLAPACKSetPreallocation()
1240 PetscCall(PetscCalloc1(sz, &a->loc)); in MatScaLAPACKSetPreallocation()
1248 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatDestroy_ScaLAPACK() local
1255 PetscCall(PetscFree(a->loc)); in MatDestroy_ScaLAPACK()
1256 PetscCall(PetscFree(a->pivots)); in MatDestroy_ScaLAPACK()
1277 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatSetUp_ScaLAPACK() local
1292 PetscCall(PetscBLASIntCast(A->rmap->N, &a->M)); in MatSetUp_ScaLAPACK()
1293 PetscCall(PetscBLASIntCast(A->cmap->N, &a->N)); in MatSetUp_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()
1296 a->lld = PetscMax(1, a->locr); in MatSetUp_ScaLAPACK()
1302 …ALAPACKdescinit", SCALAPACKdescinit_(a->desc, &a->M, &a->N, &a->mb, &a->nb, &a->rsrc, &a->csrc, &a… in MatSetUp_ScaLAPACK()
1321 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatAssemblyEnd_ScaLAPACK() local
1335 …Kinfog2l", SCALAPACKinfog2l_(&gridx, &gcidx, a->desc, &a->grid->nprow, &a->grid->npcol, &a->grid->… in MatAssemblyEnd_ScaLAPACK()
1336 …PetscCheck(rsrc == a->grid->myrow && csrc == a->grid->mycol, PetscObjectComm((PetscObject)A), PETS… in MatAssemblyEnd_ScaLAPACK()
1339 a->loc[lridx - 1 + (lcidx - 1) * a->lld] = val[i]; in MatAssemblyEnd_ScaLAPACK()
1342 a->loc[lridx - 1 + (lcidx - 1) * a->lld] += val[i]; in MatAssemblyEnd_ScaLAPACK()
1527 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)mat->data; in MatStashScatterBegin_ScaLAPACK() local
1550 …Kinfog2l", SCALAPACKinfog2l_(&gridx, &gcidx, a->desc, &a->grid->nprow, &a->grid->npcol, &a->grid->… in MatStashScatterBegin_ScaLAPACK()
1551 j = Cblacs_pnum(a->grid->ictxt, rsrc, csrc); in MatStashScatterBegin_ScaLAPACK()
1672 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatScaLAPACKSetBlockSizes_ScaLAPACK() local
1678 PetscCall(PetscBLASIntCast((mb == PETSC_DECIDE) ? DEFAULT_BLOCKSIZE : mb, &a->mb)); in MatScaLAPACKSetBlockSizes_ScaLAPACK()
1679 PetscCall(PetscBLASIntCast((nb == PETSC_DECIDE) ? a->mb : nb, &a->nb)); in MatScaLAPACKSetBlockSizes_ScaLAPACK()
1713 Mat_ScaLAPACK *a = (Mat_ScaLAPACK *)A->data; in MatScaLAPACKGetBlockSizes_ScaLAPACK() local
1716 if (mb) *mb = a->mb; in MatScaLAPACKGetBlockSizes_ScaLAPACK()
1717 if (nb) *nb = a->nb; in MatScaLAPACKGetBlockSizes_ScaLAPACK()
1775 Mat_ScaLAPACK *a; in MatCreate_ScaLAPACK() local
1794 PetscCall(PetscNew(&a)); in MatCreate_ScaLAPACK()
1795 A->data = (void *)a; in MatCreate_ScaLAPACK()
1840 a->grid = grid; in MatCreate_ScaLAPACK()
1841 a->mb = DEFAULT_BLOCKSIZE; in MatCreate_ScaLAPACK()
1842 a->nb = DEFAULT_BLOCKSIZE; in MatCreate_ScaLAPACK()
1847 a->mb = (PetscMPIInt)array[0]; in MatCreate_ScaLAPACK()
1848 a->nb = (k > 1) ? (PetscMPIInt)array[1] : a->mb; in MatCreate_ScaLAPACK()
1852 a->roworiented = PETSC_TRUE; in MatCreate_ScaLAPACK()
1899 Mat_ScaLAPACK *a; in MatCreateScaLAPACK() local
1912 a = (Mat_ScaLAPACK *)(*A)->data; in MatCreateScaLAPACK()
1913 PetscCall(PetscBLASIntCast(M, &a->M)); in MatCreateScaLAPACK()
1914 PetscCall(PetscBLASIntCast(N, &a->N)); in MatCreateScaLAPACK()
1915 PetscCall(PetscBLASIntCast((mb == PETSC_DECIDE) ? DEFAULT_BLOCKSIZE : mb, &a->mb)); in MatCreateScaLAPACK()
1916 PetscCall(PetscBLASIntCast((nb == PETSC_DECIDE) ? a->mb : nb, &a->nb)); in MatCreateScaLAPACK()
1917 PetscCall(PetscBLASIntCast(rsrc, &a->rsrc)); in MatCreateScaLAPACK()
1918 PetscCall(PetscBLASIntCast(csrc, &a->csrc)); in MatCreateScaLAPACK()