Lines Matching refs:mb
42 PetscCall(PetscViewerASCIIPrintf(viewer, "block sizes: %d,%d\n", (int)a->mb, (int)a->nb)); in MatView_ScaLAPACK()
164 …PetscBLASInt xdesc[9], ydesc[9], x2desc[9], y2desc[9], mb, nb, lszx, lszy, zero = 0, one = 1, x… in MatMultXXXYYY_ScaLAPACK() local
170 PetscCall(PetscBLASIntCast(ranges[1], &mb)); /* x block size */ in MatMultXXXYYY_ScaLAPACK()
172 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(xdesc, &a->M, &one, &mb, &one, &zero, &zero,… in MatMultXXXYYY_ScaLAPACK()
181 lszx = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow); in MatMultXXXYYY_ScaLAPACK()
187 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(x2desc, &a->M, &one, &a->mb, &one, &zero, &z… in MatMultXXXYYY_ScaLAPACK()
214 PetscCall(PetscBLASIntCast(ranges[1], &mb)); /* y block size */ in MatMultXXXYYY_ScaLAPACK()
216 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(ydesc, &a->M, &one, &mb, &one, &zero, &zero,… in MatMultXXXYYY_ScaLAPACK()
220 lszy = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow); in MatMultXXXYYY_ScaLAPACK()
228 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(y2desc, &a->M, &one, &a->mb, &one, &zero, &z… in MatMultXXXYYY_ScaLAPACK()
435 PetscBLASInt j, ddesc[9], d2desc[9], mb, nb, lszd, zero = 0, one = 1, dlld, info; in MatGetDiagonal_ScaLAPACK() local
444 PetscCall(PetscBLASIntCast(ranges[1], &mb)); /* D block size */ in MatGetDiagonal_ScaLAPACK()
446 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(ddesc, &a->M, &one, &mb, &one, &zero, &zero,… in MatGetDiagonal_ScaLAPACK()
450 lszd = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow); in MatGetDiagonal_ScaLAPACK()
455 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(d2desc, &a->M, &one, &a->mb, &one, &zero, &z… in MatGetDiagonal_ScaLAPACK()
508 PetscBLASInt i, j, ddesc[9], d2desc[9], mb, nb, lszd, zero = 0, one = 1, dlld, info; in MatDiagonalScale_ScaLAPACK() local
546 PetscCall(PetscBLASIntCast(ranges[1], &mb)); /* D block size */ in MatDiagonalScale_ScaLAPACK()
548 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(ddesc, &a->M, &one, &mb, &one, &zero, &zero,… in MatDiagonalScale_ScaLAPACK()
552 lszd = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow); in MatDiagonalScale_ScaLAPACK()
557 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(d2desc, &a->M, &one, &a->mb, &one, &zero, &z… in MatDiagonalScale_ScaLAPACK()
643 b->mb = a->mb; in MatDuplicate_ScaLAPACK()
667 …PetscCall(MatCreateScaLAPACK(PetscObjectComm((PetscObject)A), a->nb, a->mb, a->N, a->M, a->csrc, a… in MatTranspose_ScaLAPACK()
700 …PetscCall(MatCreateScaLAPACK(PetscObjectComm((PetscObject)A), a->nb, a->mb, a->N, a->M, a->csrc, a… in MatHermitianTranspose_ScaLAPACK()
713 PetscBLASInt xdesc[9], x2desc[9], mb, lszx, zero = 0, one = 1, xlld, nrhs = 1, info; in MatSolve_ScaLAPACK() local
721 PetscCall(PetscBLASIntCast(ranges[1], &mb)); /* x block size */ in MatSolve_ScaLAPACK()
723 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(xdesc, &a->M, &one, &mb, &one, &zero, &zero,… in MatSolve_ScaLAPACK()
727 lszx = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow); in MatSolve_ScaLAPACK()
732 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(x2desc, &a->M, &one, &a->mb, &one, &zero, &z… in MatSolve_ScaLAPACK()
815 if (!a->pivots) PetscCall(PetscMalloc1(a->locr + a->mb, &a->pivots)); in MatLUFactor_ScaLAPACK()
886 …PetscCall(MatCreateScaLAPACK(PetscObjectComm((PetscObject)A), a->mb, a->nb, a->M, a->N, a->rsrc, a… in MatGetFactor_scalapack_scalapack()
952 nb = a->mb; in MatGetOwnershipIS_ScaLAPACK()
1294 a->locr = SCALAPACKnumroc_(&a->M, &a->mb, &a->grid->myrow, &a->rsrc, &a->grid->nprow); in MatSetUp_ScaLAPACK()
1302 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(a->desc, &a->M, &a->N, &a->mb, &a->nb, &a->r… in MatSetUp_ScaLAPACK()
1670 static PetscErrorCode MatScaLAPACKSetBlockSizes_ScaLAPACK(Mat A, PetscInt mb, PetscInt nb) in MatScaLAPACKSetBlockSizes_ScaLAPACK() argument
1676 …PetscCheck(mb >= 1 || mb == PETSC_DECIDE, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "mb %" PetscInt_FMT … in MatScaLAPACKSetBlockSizes_ScaLAPACK()
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()
1701 PetscErrorCode MatScaLAPACKSetBlockSizes(Mat A, PetscInt mb, PetscInt nb) in MatScaLAPACKSetBlockSizes() argument
1705 PetscValidLogicalCollectiveInt(A, mb, 2); in MatScaLAPACKSetBlockSizes()
1707 PetscTryMethod(A, "MatScaLAPACKSetBlockSizes_C", (Mat, PetscInt, PetscInt), (A, mb, nb)); in MatScaLAPACKSetBlockSizes()
1711 static PetscErrorCode MatScaLAPACKGetBlockSizes_ScaLAPACK(Mat A, PetscInt *mb, PetscInt *nb) in MatScaLAPACKGetBlockSizes_ScaLAPACK() argument
1716 if (mb) *mb = a->mb; in MatScaLAPACKGetBlockSizes_ScaLAPACK()
1741 PetscErrorCode MatScaLAPACKGetBlockSizes(Mat A, PetscInt *mb, PetscInt *nb) in MatScaLAPACKGetBlockSizes() argument
1745 PetscUseMethod(A, "MatScaLAPACKGetBlockSizes_C", (Mat, PetscInt *, PetscInt *), (A, mb, nb)); in MatScaLAPACKGetBlockSizes()
1841 a->mb = 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()
1897 PetscErrorCode MatCreateScaLAPACK(MPI_Comm comm, PetscInt mb, PetscInt nb, PetscInt M, PetscInt N, … in MatCreateScaLAPACK() argument
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()