Lines Matching refs:rbs
191 static PetscErrorCode MatSetBlockSizes_MPIAIJ(Mat M, PetscInt rbs, PetscInt cbs) in MatSetBlockSizes_MPIAIJ() argument
197 PetscCall(MatSetBlockSizes(mat->A, rbs, cbs)); in MatSetBlockSizes_MPIAIJ()
198 PetscCall(MatSetBlockSizes(mat->B, rbs, 1)); in MatSetBlockSizes_MPIAIJ()
5769 …= 0, aBn = a->B->cmap->n, row, *b_othi, *b_othj, *rvalues = NULL, *svalues = NULL, *cols, sbs, rbs; in MatGetBrowsOfAoCols_MPIAIJ() local
5796 …Private(ctx, PETSC_FALSE /*recv*/, &nrecvs, &rstarts, NULL /*indices not needed*/, &rprocs, &rbs)); in MatGetBrowsOfAoCols_MPIAIJ()
5806 …if (nrecvs) PetscCall(PetscMalloc1(rbs * (rstarts[nrecvs] - rstarts[0]), &rvalues)); /* rstarts ca… in MatGetBrowsOfAoCols_MPIAIJ()
5808 rowlen = rvalues + rstarts[i] * rbs; in MatGetBrowsOfAoCols_MPIAIJ()
5809 nrows = (rstarts[i + 1] - rstarts[i]) * rbs; /* num of indices to be received */ in MatGetBrowsOfAoCols_MPIAIJ()
5857 rowlen = rvalues + (rstarts[i] - rstarts[0]) * rbs; in MatGetBrowsOfAoCols_MPIAIJ()
5858 nrows = (rstarts[i + 1] - rstarts[i]) * rbs; /* num of rows to be received */ in MatGetBrowsOfAoCols_MPIAIJ()
5957 …(VecScatterRestoreRemoteOrdered_Private(ctx, PETSC_FALSE, &nrecvs, &rstarts, NULL, &rprocs, &rbs)); in MatGetBrowsOfAoCols_MPIAIJ()