Lines Matching refs:bs

7   PetscInt bs, nz, *i, *j;  member
11 PetscInt bs, n, N, Nghosts, *ghosts; member
18 PetscInt *globals, *sd_nnz, *so_nnz, rstart, bs, i; in DMCreateMatrix_Sliced() local
24 bs = slice->bs; in DMCreateMatrix_Sliced()
26 PetscCall(MatSetSizes(*J, slice->n * bs, slice->n * bs, PETSC_DETERMINE, PETSC_DETERMINE)); in DMCreateMatrix_Sliced()
27 PetscCall(MatSetBlockSize(*J, bs)); in DMCreateMatrix_Sliced()
29 PetscCall(MatSeqBAIJSetPreallocation(*J, bs, slice->d_nz, slice->d_nnz)); in DMCreateMatrix_Sliced()
30 …PetscCall(MatMPIBAIJSetPreallocation(*J, bs, slice->d_nz, slice->d_nnz, slice->o_nz, slice->o_nnz)… in DMCreateMatrix_Sliced()
36 if (bs == 1) { in DMCreateMatrix_Sliced()
40 PetscCall(MatSeqAIJSetPreallocation(*J, slice->d_nz * bs, NULL)); in DMCreateMatrix_Sliced()
41 PetscCall(MatMPIAIJSetPreallocation(*J, slice->d_nz * bs, NULL, slice->o_nz * bs, NULL)); in DMCreateMatrix_Sliced()
44 PetscCall(PetscMalloc2(slice->n * bs, &sd_nnz, (!!slice->o_nnz) * slice->n * bs, &so_nnz)); in DMCreateMatrix_Sliced()
45 for (i = 0; i < slice->n * bs; i++) { in DMCreateMatrix_Sliced()
46bs] - 1) * (slice->ofill ? slice->ofill->i[i % bs + 1] - slice->ofill->i[i % bs] : bs) + (slice->d… in DMCreateMatrix_Sliced()
47 …o_nnz) so_nnz[i] = slice->o_nnz[i / bs] * (slice->ofill ? slice->ofill->i[i % bs + 1] - slice->ofi… in DMCreateMatrix_Sliced()
49 PetscCall(MatSeqAIJSetPreallocation(*J, slice->d_nz * bs, sd_nnz)); in DMCreateMatrix_Sliced()
50 PetscCall(MatMPIAIJSetPreallocation(*J, slice->d_nz * bs, sd_nnz, slice->o_nz * bs, so_nnz)); in DMCreateMatrix_Sliced()
58 for (i = 0; i < slice->n; i++) globals[i] = rstart / bs + i; in DMCreateMatrix_Sliced()
61 …PetscCall(ISLocalToGlobalMappingCreate(PETSC_COMM_SELF, bs, slice->n + slice->Nghosts, globals, PE… in DMCreateMatrix_Sliced()
85 PetscErrorCode DMSlicedSetGhosts(DM dm, PetscInt bs, PetscInt nlocal, PetscInt Nghosts, const Petsc… in DMSlicedSetGhosts() argument
94 slice->bs = bs; in DMSlicedSetGhosts()
140 static PetscErrorCode DMSlicedSetBlockFills_Private(PetscInt bs, const PetscInt *fill, DMSlicedBloc… in DMSlicedSetBlockFills_Private() argument
149 for (i = 0, nz = 0; i < bs * bs; i++) in DMSlicedSetBlockFills_Private()
151 PetscCall(PetscMalloc3(1, &f, bs + 1, &fi, nz, &fj)); in DMSlicedSetBlockFills_Private()
152 f->bs = bs; in DMSlicedSetBlockFills_Private()
156 for (i = 0, nz = 0; i < bs; i++) { in DMSlicedSetBlockFills_Private()
158 for (j = 0; j < bs; j++) in DMSlicedSetBlockFills_Private()
159 if (fill[i * bs + j]) fj[nz++] = j; in DMSlicedSetBlockFills_Private()
191 PetscCall(DMSlicedSetBlockFills_Private(slice->bs, dfill, &slice->dfill)); in DMSlicedSetBlockFills()
192 PetscCall(DMSlicedSetBlockFills_Private(slice->bs, ofill, &slice->ofill)); in DMSlicedSetBlockFills()
217 …PetscCall(VecCreateGhostBlock(PetscObjectComm((PetscObject)dm), slice->bs, slice->n * slice->bs, P… in DMCreateGlobalVector_Sliced()
301 PetscErrorCode DMSlicedCreate(MPI_Comm comm, PetscInt bs, PetscInt nlocal, PetscInt Nghosts, const … in DMSlicedCreate() argument
307 PetscCall(DMSlicedSetGhosts(*dm, bs, nlocal, Nghosts, ghosts)); in DMSlicedCreate()