Lines Matching refs:nb
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
175 PetscCall(PetscBLASIntCast(ranges[1], &nb)); /* y block size */ in MatMultXXXYYY_ScaLAPACK()
177 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(ydesc, &one, &a->N, &one, &nb, &zero, &zero,… in MatMultXXXYYY_ScaLAPACK()
182 lszy = SCALAPACKnumroc_(&a->N, &a->nb, &a->grid->mycol, &a->csrc, &a->grid->npcol); in MatMultXXXYYY_ScaLAPACK()
189 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(y2desc, &one, &a->N, &one, &a->nb, &zero, &z… in MatMultXXXYYY_ScaLAPACK()
209 PetscCall(PetscBLASIntCast(ranges[1], &nb)); /* x block size */ in MatMultXXXYYY_ScaLAPACK()
211 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(xdesc, &one, &a->N, &one, &nb, &zero, &zero,… in MatMultXXXYYY_ScaLAPACK()
221 lszx = SCALAPACKnumroc_(&a->N, &a->nb, &a->grid->mycol, &a->csrc, &a->grid->npcol); in MatMultXXXYYY_ScaLAPACK()
226 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(x2desc, &one, &a->N, &one, &a->nb, &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
472 PetscCall(PetscBLASIntCast(ranges[1], &nb)); /* D block size */ in MatGetDiagonal_ScaLAPACK()
474 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(ddesc, &one, &a->N, &one, &nb, &zero, &zero,… in MatGetDiagonal_ScaLAPACK()
478 lszd = SCALAPACKnumroc_(&a->N, &a->nb, &a->grid->mycol, &a->csrc, &a->grid->npcol); in MatGetDiagonal_ScaLAPACK()
482 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(d2desc, &one, &a->N, &one, &a->nb, &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
515 PetscCall(PetscBLASIntCast(ranges[1], &nb)); /* D block size */ in MatDiagonalScale_ScaLAPACK()
517 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(ddesc, &one, &a->N, &one, &nb, &zero, &zero,… in MatDiagonalScale_ScaLAPACK()
521 lszd = SCALAPACKnumroc_(&a->N, &a->nb, &a->grid->mycol, &a->csrc, &a->grid->npcol); in MatDiagonalScale_ScaLAPACK()
525 …PetscCallBLAS("SCALAPACKdescinit", SCALAPACKdescinit_(d2desc, &one, &a->N, &one, &a->nb, &zero, &z… in MatDiagonalScale_ScaLAPACK()
644 b->nb = a->nb; 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()
886 …PetscCall(MatCreateScaLAPACK(PetscObjectComm((PetscObject)A), a->mb, a->nb, a->M, a->N, a->rsrc, a… in MatGetFactor_scalapack_scalapack()
947 PetscInt i, n, nb, isrc, nproc, iproc, *idx; in MatGetOwnershipIS_ScaLAPACK() local
952 nb = a->mb; in MatGetOwnershipIS_ScaLAPACK()
957 …for (i = 0; i < n; i++) idx[i] = nproc * nb * (i / nb) + i % nb + ((nproc + iproc - isrc) % nproc)… in MatGetOwnershipIS_ScaLAPACK()
962 nb = a->nb; in MatGetOwnershipIS_ScaLAPACK()
967 …for (i = 0; i < n; i++) idx[i] = nproc * nb * (i / nb) + i % nb + ((nproc + iproc - isrc) % nproc)… in MatGetOwnershipIS_ScaLAPACK()
1295 a->locc = SCALAPACKnumroc_(&a->N, &a->nb, &a->grid->mycol, &a->csrc, &a->grid->npcol); 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
1677 …PetscCheck(nb >= 1 || nb == PETSC_DECIDE, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "nb %" PetscInt_FMT … 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
1706 PetscValidLogicalCollectiveInt(A, nb, 3); 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
1717 if (nb) *nb = a->nb; 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()
1842 a->nb = DEFAULT_BLOCKSIZE; 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
1916 PetscCall(PetscBLASIntCast((nb == PETSC_DECIDE) ? a->mb : nb, &a->nb)); in MatCreateScaLAPACK()