Lines Matching refs:blocksizes
10 PetscInt i, j, rows[2], *parts, cnt, N = 21, nblocks, *blocksizes; in main() local
70 PetscCall(PetscMalloc1(nblocks, &blocksizes)); in main()
73 blocksizes[i] = PetscMax(1, (PetscInt)(N * rand)); in main()
74 blocksizes[i] = PetscMin(blocksizes[i], N - cnt); in main()
75 cnt += blocksizes[i]; in main()
81 if (cnt < N) blocksizes[nblocks - 1] = N - cnt; in main()
83 PetscCall(BuildMatrix(parts, nblocks, blocksizes, &A)); in main()
84 PetscCall(PetscFree(blocksizes)); in main()
131 PetscErrorCode BuildMatrix(const PetscInt *parts, PetscInt nblocks, const PetscInt *blocksizes, Mat… in BuildMatrix() argument
142 PetscCall(MatSetValue(*A, cnt, cnt + blocksizes[i] - 1, 1.0, INSERT_VALUES)); in BuildMatrix()
143 PetscCall(MatSetValue(*A, cnt + blocksizes[i] - 1, cnt, 1.0, INSERT_VALUES)); in BuildMatrix()
144 cnt += blocksizes[i]; in BuildMatrix()