Lines Matching refs:bs

10   PetscInt     bs = mat->rmap->bs;  in MatGetMultiProcBlock_MPIBAIJ()  local
22 PetscCall(MatSetBlockSizes(*subMat, mat->rmap->bs, mat->cmap->bs)); in MatGetMultiProcBlock_MPIBAIJ()
25 PetscCall(PetscLayoutSetBlockSize((*subMat)->rmap, mat->rmap->bs)); in MatGetMultiProcBlock_MPIBAIJ()
26 PetscCall(PetscLayoutSetBlockSize((*subMat)->cmap, mat->cmap->bs)); in MatGetMultiProcBlock_MPIBAIJ()
40 PetscCall(PetscCalloc1(aij->B->cmap->n / bs, &garrayCMap)); in MatGetMultiProcBlock_MPIBAIJ()
41 for (i = 0; i < aij->B->cmap->n / bs; i++) { in MatGetMultiProcBlock_MPIBAIJ()
45 if ((col >= mat->cmap->range[rank] / bs) && (col < mat->cmap->range[rank + 1] / bs)) { in MatGetMultiProcBlock_MPIBAIJ()
46 garrayCMap[i] = (((*subMat)->cmap->range[subRank] - mat->cmap->range[rank]) / bs + col + 1); in MatGetMultiProcBlock_MPIBAIJ()
54 PetscCall(PetscCalloc1(aij->B->rmap->n / bs, &nnz)); in MatGetMultiProcBlock_MPIBAIJ()
55 for (i = 0; i < aij->B->rmap->n / bs; i++) { in MatGetMultiProcBlock_MPIBAIJ()
60 PetscCall(MatMPIBAIJSetPreallocation(*(subMat), bs, 0, NULL, 0, nnz)); in MatGetMultiProcBlock_MPIBAIJ()
80 PetscCall(PetscMalloc3(bs, &newbRow, bs, &newbCol, bs * bs, &vals)); in MatGetMultiProcBlock_MPIBAIJ()
81 for (i = 0; i < aij->B->rmap->n / bs; i++) { in MatGetMultiProcBlock_MPIBAIJ()
82 newRow = (*subMat)->rmap->range[subCommRank] + i * bs; in MatGetMultiProcBlock_MPIBAIJ()
87 newCol *= bs; in MatGetMultiProcBlock_MPIBAIJ()
88 for (k = 0; k < bs; k++) { in MatGetMultiProcBlock_MPIBAIJ()
93 aijBvals = aijB->a + j * bs * bs; in MatGetMultiProcBlock_MPIBAIJ()
94 for (k1 = 0; k1 < bs; k1++) { in MatGetMultiProcBlock_MPIBAIJ()
95 for (k = 0; k < bs; k++) vals[k1 + k * bs] = *aijBvals++; in MatGetMultiProcBlock_MPIBAIJ()
97 PetscCall(MatSetValues(*subMat, bs, newbRow, bs, newbCol, vals, INSERT_VALUES)); in MatGetMultiProcBlock_MPIBAIJ()