Lines Matching refs:sbaij

8   Mat_MPISBAIJ   *sbaij = (Mat_MPISBAIJ *)mat->data;  in MatSetUpMultiply_MPISBAIJ()  local
9 Mat_SeqBAIJ *B = (Mat_SeqBAIJ *)sbaij->B->data; in MatSetUpMultiply_MPISBAIJ()
11 PetscInt bs = mat->rmap->bs, *stmp, mbs = sbaij->mbs, vec_size, nt; in MatSetUpMultiply_MPISBAIJ()
14 PetscMPIInt rank = sbaij->rank; in MatSetUpMultiply_MPISBAIJ()
15 PetscInt *owners = sbaij->rangebs, *ec_owner, k; in MatSetUpMultiply_MPISBAIJ()
23 PetscInt Nbs = sbaij->Nbs; in MatSetUpMultiply_MPISBAIJ()
82 for (j = 0; j < sbaij->size; j++) { in MatSetUpMultiply_MPISBAIJ()
102 PetscCall(PetscLayoutDestroy(&sbaij->B->cmap)); in MatSetUpMultiply_MPISBAIJ()
103 …eFromSizes(PetscObjectComm((PetscObject)sbaij->B), ec * mat->rmap->bs, ec * mat->rmap->bs, mat->rm… in MatSetUpMultiply_MPISBAIJ()
105 PetscCall(VecScatterDestroy(&sbaij->sMvctx)); in MatSetUpMultiply_MPISBAIJ()
107 PetscCall(VecCreateSeq(PETSC_COMM_SELF, ec * bs, &sbaij->lvec)); in MatSetUpMultiply_MPISBAIJ()
118 PetscCall(VecScatterCreate(gvec, from, sbaij->lvec, to, &sbaij->Mvctx)); in MatSetUpMultiply_MPISBAIJ()
121 sbaij->garray = garray; in MatSetUpMultiply_MPISBAIJ()
127 …VecCreateMPI(PetscObjectComm((PetscObject)mat), (mbs + ec) * bs, PETSC_DETERMINE, &sbaij->slvec0)); in MatSetUpMultiply_MPISBAIJ()
128 PetscCall(VecDuplicate(sbaij->slvec0, &sbaij->slvec1)); in MatSetUpMultiply_MPISBAIJ()
129 PetscCall(VecGetSize(sbaij->slvec0, &vec_size)); in MatSetUpMultiply_MPISBAIJ()
131 PetscCall(VecGetOwnershipRanges(sbaij->slvec0, &sowners)); in MatSetUpMultiply_MPISBAIJ()
151 PetscCall(VecScatterCreate(sbaij->slvec0, from, sbaij->slvec1, to, &sbaij->sMvctx)); in MatSetUpMultiply_MPISBAIJ()
152 PetscCall(VecScatterViewFromOptions(sbaij->sMvctx, (PetscObject)mat, "-matmult_vecscatter_view")); in MatSetUpMultiply_MPISBAIJ()
154 PetscCall(VecGetLocalSize(sbaij->slvec1, &nt)); in MatSetUpMultiply_MPISBAIJ()
155 PetscCall(VecGetArray(sbaij->slvec1, &ptr)); in MatSetUpMultiply_MPISBAIJ()
156 PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, bs * mbs, ptr, &sbaij->slvec1a)); in MatSetUpMultiply_MPISBAIJ()
157 …ay(PETSC_COMM_SELF, 1, nt - bs * mbs, PetscSafePointerPlusOffset(ptr, bs * mbs), &sbaij->slvec1b)); in MatSetUpMultiply_MPISBAIJ()
158 PetscCall(VecRestoreArray(sbaij->slvec1, &ptr)); in MatSetUpMultiply_MPISBAIJ()
160 PetscCall(VecGetArray(sbaij->slvec0, &ptr)); in MatSetUpMultiply_MPISBAIJ()
161 …ay(PETSC_COMM_SELF, 1, nt - bs * mbs, PetscSafePointerPlusOffset(ptr, bs * mbs), &sbaij->slvec0b)); in MatSetUpMultiply_MPISBAIJ()
162 PetscCall(VecRestoreArray(sbaij->slvec0, &ptr)); in MatSetUpMultiply_MPISBAIJ()