Home
last modified time | relevance | path

Searched refs:Bbn (Results 1 – 2 of 2) sorted by relevance

/petsc/src/ksp/ksp/interface/
H A Ditfunc.c1204 PetscInt n1, N1, n2, N2, Bbn = PETSC_DECIDE; in KSPMatSolve_Private() local
1238 PetscCall(KSPGetMatSolveBatchSize(ksp, &Bbn)); in KSPMatSolve_Private()
1240 if (Bbn == PETSC_DECIDE) Bbn = N2; in KSPMatSolve_Private()
1241 …scCheck(Bbn >= 1, PetscObjectComm((PetscObject)ksp), PETSC_ERR_ARG_OUTOFRANGE, "KSPMatSolve() batc… in KSPMatSolve_Private()
1242 … PetscInt_FMT "\n", ((PetscObject)ksp)->type_name, ksp->transpose_solve ? " transpose" : "", Bbn)); in KSPMatSolve_Private()
1244 if (Bbn >= N2) { in KSPMatSolve_Private()
1256 for (n2 = 0; n2 < N2; n2 += Bbn) { in KSPMatSolve_Private()
1257 … PetscCall(MatDenseGetSubMatrix(B, PETSC_DECIDE, PETSC_DECIDE, n2, PetscMin(n2 + Bbn, N2), &vB)); in KSPMatSolve_Private()
1258 … PetscCall(MatDenseGetSubMatrix(X, PETSC_DECIDE, PETSC_DECIDE, n2, PetscMin(n2 + Bbn, N2), &vX)); in KSPMatSolve_Private()
/petsc/src/mat/impls/aij/mpi/
H A Dmpimatmatmult.c419 PetscInt Am = A->rmap->n, Bm = B->rmap->n, BN = B->cmap->N, Bbn, Bbn1, bs, numBb; in MatMatMultSymbolic_MPIAIJ_MPIDense() local
453 PetscCallMPI(MPIU_Allreduce(&Bbn1, &Bbn, 1, MPIU_INT, MPI_MAX, comm)); in MatMatMultSymbolic_MPIAIJ_MPIDense()
458 …tscOptionsInt("-matproduct_batch_size", "Number of columns in Bb", "MatProduct", Bbn, &Bbn, NULL)); in MatMatMultSymbolic_MPIAIJ_MPIDense()
460 Bbn = PetscMin(Bbn, BN); in MatMatMultSymbolic_MPIAIJ_MPIDense()
462 if (Bbn > 0 && Bbn < BN) { in MatMatMultSymbolic_MPIAIJ_MPIDense()
463 numBb = BN / Bbn; in MatMatMultSymbolic_MPIAIJ_MPIDense()
464 Bbn1 = BN - numBb * Bbn; in MatMatMultSymbolic_MPIAIJ_MPIDense()
468 …"use Bb, BN=%" PetscInt_FMT ", Bbn=%" PetscInt_FMT "; numBb=%" PetscInt_FMT "\n", BN, Bbn, numBb)); in MatMatMultSymbolic_MPIAIJ_MPIDense()
477 PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, nz, Bbn, NULL, &contents->workB)); in MatMatMultSymbolic_MPIAIJ_MPIDense()