Lines Matching refs:bs
9 PetscInt i, col[3], n = 5, bs = 1; in main() local
14 PetscCall(PetscOptionsGetInt(NULL, NULL, "-bs", &bs, NULL)); in main()
18 PetscCall(MatSetSizes(seqmat, (n + rank) * bs, PETSC_DECIDE, PETSC_DECIDE, n * bs)); in main()
20 PetscCall(MatSeqAIJSetPreallocation(seqmat, 3 * bs, NULL)); in main()
21 PetscCall(MatSeqBAIJSetPreallocation(seqmat, bs, 3, NULL)); in main()
23 if (bs == 1) { in main()
46 PetscCall(PetscMalloc3(bs * bs, &vals, bs, &rows, bs, &cols)); in main()
48 for (i = 0; i < bs * bs; i++) vals[i] = 2.0; in main()
49 for (i = 0; i < n * bs; i += bs) { in main()
50 for (j = 0; j < bs; j++) { in main()
54 PetscCall(MatSetValues(seqmat, bs, rows, bs, cols, vals, INSERT_VALUES)); in main()
57 for (i = 0; i < bs * bs; i++) vals[i] = -1.0; in main()
58 for (i = 0; i < (n - 1) * bs; i += bs) { in main()
59 for (j = 0; j < bs; j++) { in main()
61 cols[j] = i + bs + j; in main()
63 PetscCall(MatSetValues(seqmat, bs, rows, bs, cols, vals, INSERT_VALUES)); in main()