| /petsc/src/mat/tutorials/ |
| H A D | ex4.c | 13 PetscInt n = 5, m = 5, *dnnz, *onnz, i, rstart, rend, M, N; in main() local
18 PetscCall(PetscMalloc2(m, &dnnz, m, &onnz)); in main()
21 onnz[i] = 1; in main()
23 …eateAIJ(comm, m, n, PETSC_DETERMINE, PETSC_DETERMINE, PETSC_DECIDE, dnnz, PETSC_DECIDE, onnz, &A)); in main()
26 PetscCall(PetscFree2(dnnz, onnz)); in main()
|
| H A D | ex4f.F90 | 10 PetscInt, pointer, dimension(:) :: dnnz, onnz
17 allocate (onnz(0:m - 1))
21 onnz(i) = 1
24 …MM_WORLD, m, n, PETSC_DETERMINE, PETSC_DETERMINE, PETSC_DECIDE, dnnz, PETSC_DECIDE, onnz, A, ierr))
28 deallocate (onnz)
|
| /petsc/src/mat/graphops/order/ |
| H A D | spectral.c | 24 PetscInt *dnnz, *onnz; in MatCreateLaplacian() local
34 PetscCall(PetscMalloc2(m, &dnnz, m, &onnz)); in MatCreateLaplacian()
41 dnnz[r - rStart] = onnz[r - rStart] = 0; in MatCreateLaplacian()
50 else ++onnz[r - rStart]; in MatCreateLaplacian()
62 PetscCall(MatXAIJSetPreallocation(*L, 1, dnnz, onnz, NULL, NULL)); in MatCreateLaplacian()
75 newVals[newcols] = dnnz[r - rStart] + onnz[r - rStart] - 1; in MatCreateLaplacian()
87 newVals[newcols] = dnnz[r - rStart] + onnz[r - rStart] - 1; in MatCreateLaplacian()
93 PetscCall(PetscFree2(dnnz, onnz)); in MatCreateLaplacian()
|
| /petsc/src/mat/tests/ |
| H A D | ex261.c | 61 PetscInt *dnnz, *onnz; in InitializeMatrix() local
69 PetscCall(PetscMalloc2(rows, &dnnz, rows, &onnz)); in InitializeMatrix()
72 onnz[i] = offdiag_nnz; in InitializeMatrix()
74 PetscCall(MatXAIJSetPreallocation(A, PETSC_DECIDE, dnnz, onnz, NULL, NULL)); in InitializeMatrix()
75 PetscCall(PetscFree2(dnnz, onnz)); in InitializeMatrix()
|
| /petsc/src/mat/utils/ |
| H A D | gcreate.c | 353 …etPreallocation(Mat A, PetscInt bs, const PetscInt dnnz[], const PetscInt onnz[], const PetscInt d… in MatXAIJSetPreallocation() argument
367 PetscCall(MatMPIBAIJSetPreallocation(A, bs, 0, dnnz, 0, onnz)); in MatXAIJSetPreallocation()
381 PetscCall(MatMPIAIJSetPreallocation(A, 0, dnnz, 0, onnz)); in MatXAIJSetPreallocation()
382 PetscCall(MatISSetPreallocation(A, 0, dnnz, 0, onnz)); in MatXAIJSetPreallocation()
384 PetscCall(MatHYPRESetPreallocation(A, 0, dnnz, 0, onnz)); in MatXAIJSetPreallocation()
389 PetscCall(PetscMalloc2((!!dnnz) * m, &sdnnz, (!!onnz) * m, &sonnz)); in MatXAIJSetPreallocation()
392 if (onnz) sonnz[i] = onnz[i / bs] * cbs; in MatXAIJSetPreallocation()
395 PetscCall(MatMPIAIJSetPreallocation(A, 0, dnnz ? sdnnz : NULL, 0, onnz ? sonnz : NULL)); in MatXAIJSetPreallocation()
396 PetscCall(MatISSetPreallocation(A, 0, dnnz ? sdnnz : NULL, 0, onnz ? sonnz : NULL)); in MatXAIJSetPreallocation()
398 PetscCall(MatHYPRESetPreallocation(A, 0, dnnz ? sdnnz : NULL, 0, onnz ? sonnz : NULL)); in MatXAIJSetPreallocation()
|
| /petsc/src/mat/impls/shell/ |
| H A D | shellcnv.c | 9 PetscInt *dnnz, *onnz, *dnnzu, *onnzu; in MatConvert_Shell() local
30 PetscCall(PetscMalloc4(mbs, &dnnz, mbs, &onnz, mbs, &dnnzu, mbs, &onnzu)); in MatConvert_Shell()
33 onnz[i] = Nbs - nbs; in MatConvert_Shell()
37 PetscCall(MatXAIJSetPreallocation(mat, PETSC_DECIDE, dnnz, onnz, dnnzu, onnzu)); in MatConvert_Shell()
38 PetscCall(PetscFree4(dnnz, onnz, dnnzu, onnzu)); in MatConvert_Shell()
|
| /petsc/src/mat/impls/aij/mpi/crl/ |
| H A D | mcrl.c | 165 …cInt m, PetscInt n, PetscInt nz, const PetscInt nnz[], PetscInt onz, const PetscInt onnz[], Mat *A) in MatCreateMPIAIJCRL() argument
171 PetscCall(MatMPIAIJSetPreallocation_MPIAIJ(*A, nz, (PetscInt *)nnz, onz, (PetscInt *)onnz)); in MatCreateMPIAIJCRL()
|
| /petsc/src/mat/impls/hypre/ |
| H A D | mhypre.c | 801 PetscInt *garray, i, noffd, dnnz, onnz, *row_starts, *col_starts; local
857 onnz = offd->nz;
886 onnz = 0;
900 …gInt *)row_starts, (HYPRE_BigInt *)col_starts, (HYPRE_Int)noffd, (HYPRE_Int)dnnz, (HYPRE_Int)onnz);
923 PetscCall(PetscMalloc2(A->rmap->n + 1, &hoi, onnz, &hoj));
925 for (i = 0; i < onnz; i++) hoj[i] = (HYPRE_Int)poj[i];
1614 …reallocation_HYPRE(Mat A, PetscInt dnz, const PetscInt dnnz[], PetscInt onz, const PetscInt onnz[]) argument
1651 if (!onnz) {
1654 } else honnz = (HYPRE_Int *)onnz;
1681 if (!onnz && honnz) PetscCall(PetscFree(honnz));
[all …]
|
| /petsc/src/mat/impls/nest/ |
| H A D | matnest.c | 2020 PetscInt m, n, M, N, i, j, k, *dnnz, *onnz = NULL, rstart, cstart, cend; in MatConvert_Nest_AIJ() local
2104 onnz = dnnz + m; in MatConvert_Nest_AIJ()
2107 onnz[k] = 0; in MatConvert_Nest_AIJ()
2182 PetscCall(PetscSFReduceBegin(bmsf, MPIU_INT, sub_onnz, onnz, MPI_SUM)); in MatConvert_Nest_AIJ()
2183 PetscCall(PetscSFReduceEnd(bmsf, MPIU_INT, sub_onnz, onnz, MPI_SUM)); in MatConvert_Nest_AIJ()
2194 onnz[i] = PetscMin(onnz[i], A->cmap->N - A->cmap->n); in MatConvert_Nest_AIJ()
2197 PetscCall(MatMPIAIJSetPreallocation(C, 0, dnnz, 0, onnz)); in MatConvert_Nest_AIJ()
|
| /petsc/src/dm/impls/plex/ |
| H A D | plextree.c | 2488 PetscInt *dnnz, *onnz; in DMPlexComputeInterpolatorTree() local
2501 PetscCall(PetscCalloc2(nGlobal, &dnnz, nGlobal, &onnz)); in DMPlexComputeInterpolatorTree()
2590 onnz[gIndFine - rowStart] = 1; in DMPlexComputeInterpolatorTree()
2606 onnz[gIndFine - rowStart] = 1; in DMPlexComputeInterpolatorTree()
2622 onnz[gIndFine - rowStart] = numO[f]; in DMPlexComputeInterpolatorTree()
2633 onnz[gIndFine - rowStart] = numO[0]; in DMPlexComputeInterpolatorTree()
2648 onnz[gIndFine - rowStart] = numO[f]; in DMPlexComputeInterpolatorTree()
2659 onnz[gIndFine - rowStart] = numO[0]; in DMPlexComputeInterpolatorTree()
2665 PetscCall(MatXAIJSetPreallocation(mat, 1, dnnz, onnz, NULL, NULL)); in DMPlexComputeInterpolatorTree()
2666 PetscCall(PetscFree2(dnnz, onnz)); in DMPlexComputeInterpolatorTree()
|
| /petsc/src/dm/impls/network/ |
| H A D | network.c | 2475 PetscInt *dnnz, *onnz; in DMCreateMatrix_Network() local
2607 PetscCall(PetscMalloc2(localSize, &dnnz, localSize, &onnz)); in DMCreateMatrix_Network()
2616 onnz[j] = (PetscInt)PetscRealPart(vonz[j]); in DMCreateMatrix_Network()
2624 PetscCall(MatMPIAIJSetPreallocation(*J, 0, dnnz, 0, onnz)); in DMCreateMatrix_Network()
2627 PetscCall(PetscFree2(dnnz, onnz)); in DMCreateMatrix_Network()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 1501 …PetscInt i, j, m, n, ng, anz, bnz, *dnnz, *onnz, *tdnnz, *tonnz, *rdest, *cdest, *work, *gc… in MatPermute_MPIAIJ() local
1543 PetscCall(PetscCalloc4(m, &dnnz, m, &onnz, m, &tdnnz, m, &tonnz)); in MatPermute_MPIAIJ()
1555 else onnz[i]++; in MatPermute_MPIAIJ()
1562 else onnz[i]++; in MatPermute_MPIAIJ()
1567 PetscCall(PetscSFBcastBegin(rowsf, MPIU_INT, onnz, tonnz, MPI_REPLACE)); in MatPermute_MPIAIJ()
1568 PetscCall(PetscSFBcastEnd(rowsf, MPIU_INT, onnz, tonnz, MPI_REPLACE)); in MatPermute_MPIAIJ()
1575 PetscInt *acols = dnnz, *bcols = onnz; /* Repurpose now-unneeded arrays */ in MatPermute_MPIAIJ()
1594 PetscCall(PetscFree4(dnnz, onnz, tdnnz, tonnz)); in MatPermute_MPIAIJ()
|
| /petsc/src/binding/petsc4py/src/petsc4py/PETSc/ |
| H A D | Mat.pyx | 5020 onnz: Sequence[int]) -> Self:
5031 onnz:
5044 onnz = iarray_i(onnz, NULL, &connz)
|