| /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()
20 dnnz[i] = 1; in main()
23 …PetscCall(MatCreateAIJ(comm, m, n, PETSC_DETERMINE, PETSC_DETERMINE, PETSC_DECIDE, dnnz, PETSC_DEC… in main()
26 PetscCall(PetscFree2(dnnz, onnz)); in main()
|
| H A D | ex4f.F90 | 10 PetscInt, pointer, dimension(:) :: dnnz, onnz
16 allocate (dnnz(0:m - 1))
20 dnnz(i) = 1
24 …J(PETSC_COMM_WORLD, m, n, PETSC_DETERMINE, PETSC_DETERMINE, PETSC_DECIDE, dnnz, PETSC_DECIDE, onnz…
27 deallocate (dnnz)
|
| /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()
47 ++dnnz[r - rStart]; in MatCreateLaplacian()
49 if ((cols[c] >= rStart) && (cols[c] < rEnd)) ++dnnz[r - rStart]; in MatCreateLaplacian()
56 ++dnnz[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 | ex184.c | 11 PetscInt *dnnz; in main() local
34 PetscCall(PetscMalloc1(m / bs, &dnnz)); in main()
35 for (j = 0; j < m / bs; j++) dnnz[j] = 1; in main()
36 PetscCall(MatXAIJSetPreallocation(A, bs, dnnz, NULL, NULL, NULL)); in main()
37 PetscCall(PetscFree(dnnz)); in main()
|
| H A D | ex261.c | 61 PetscInt *dnnz, *onnz; in InitializeMatrix() local
69 PetscCall(PetscMalloc2(rows, &dnnz, rows, &onnz)); in InitializeMatrix()
71 dnnz[i] = diag_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 PetscErrorCode MatXAIJSetPreallocation(Mat A, PetscInt bs, const PetscInt dnnz[], const PetscInt on… in MatXAIJSetPreallocation() argument
366 PetscCall(MatSeqBAIJSetPreallocation(A, bs, 0, dnnz)); in MatXAIJSetPreallocation()
367 PetscCall(MatMPIBAIJSetPreallocation(A, bs, 0, dnnz, 0, onnz)); in MatXAIJSetPreallocation()
380 PetscCall(MatSeqAIJSetPreallocation(A, 0, dnnz)); 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()
391 if (dnnz) sdnnz[i] = dnnz[i / bs] * cbs; in MatXAIJSetPreallocation()
394 PetscCall(MatSeqAIJSetPreallocation(A, 0, dnnz ? sdnnz : NULL)); in MatXAIJSetPreallocation()
[all …]
|
| /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()
32 dnnz[i] = 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/hypre/ |
| H A D | mhypre.c | 801 PetscInt *garray, i, noffd, dnnz, onnz, *row_starts, *col_starts; local
856 dnnz = diag->nz;
885 dnnz = diag->nz;
900 …gInt *)row_starts, (HYPRE_BigInt *)col_starts, (HYPRE_Int)noffd, (HYPRE_Int)dnnz, (HYPRE_Int)onnz);
909 PetscCall(PetscMalloc2(A->rmap->n + 1, &hdi, dnnz, &hdj));
911 for (i = 0; i < dnnz; i++) hdj[i] = (HYPRE_Int)pdj[i];
1614 static PetscErrorCode MatHYPRESetPreallocation_HYPRE(Mat A, PetscInt dnz, const PetscInt dnnz[], Pe… argument
1642 if (!dnnz) {
1646 hdnnz = (HYPRE_Int *)dnnz;
1680 if (!dnnz) PetscCall(PetscFree(hdnnz));
[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
2102 PetscCall(PetscMalloc1(2 * m, &dnnz)); in MatConvert_Nest_AIJ()
2104 onnz = dnnz + m; in MatConvert_Nest_AIJ()
2106 dnnz[k] = 0; in MatConvert_Nest_AIJ()
2180 PetscCall(PetscSFReduceBegin(bmsf, MPIU_INT, sub_dnnz, dnnz, MPI_SUM)); in MatConvert_Nest_AIJ()
2181 PetscCall(PetscSFReduceEnd(bmsf, MPIU_INT, sub_dnnz, dnnz, MPI_SUM)); in MatConvert_Nest_AIJ()
2193 dnnz[i] = PetscMin(dnnz[i], A->cmap->n); in MatConvert_Nest_AIJ()
2196 PetscCall(MatSeqAIJSetPreallocation(C, 0, dnnz)); in MatConvert_Nest_AIJ()
2197 PetscCall(MatMPIAIJSetPreallocation(C, 0, dnnz, 0, onnz)); in MatConvert_Nest_AIJ()
2198 PetscCall(PetscFree(dnnz)); in MatConvert_Nest_AIJ()
|
| /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()
2615 dnnz[j] = (PetscInt)PetscRealPart(vdnz[j]); in DMCreateMatrix_Network()
2623 PetscCall(MatSeqAIJSetPreallocation(*J, 0, dnnz)); in DMCreateMatrix_Network()
2624 PetscCall(MatMPIAIJSetPreallocation(*J, 0, dnnz, 0, onnz)); in DMCreateMatrix_Network()
2627 PetscCall(PetscFree2(dnnz, onnz)); in DMCreateMatrix_Network()
|
| /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()
2587 dnnz[gIndFine - rowStart] = 1; in DMPlexComputeInterpolatorTree()
2603 dnnz[gIndFine - rowStart] = 1; in DMPlexComputeInterpolatorTree()
2621 dnnz[gIndFine - rowStart] = numD[f]; in DMPlexComputeInterpolatorTree()
2632 dnnz[gIndFine - rowStart] = numD[0]; in DMPlexComputeInterpolatorTree()
2647 dnnz[gIndFine - rowStart] = numD[f]; in DMPlexComputeInterpolatorTree()
2658 dnnz[gIndFine - rowStart] = numD[0]; in DMPlexComputeInterpolatorTree()
2665 PetscCall(MatXAIJSetPreallocation(mat, 1, dnnz, onnz, NULL, NULL)); in DMPlexComputeInterpolatorTree()
2666 PetscCall(PetscFree2(dnnz, onnz)); in DMPlexComputeInterpolatorTree()
|
| /petsc/src/ksp/pc/impls/patch/ |
| H A D | pcpatch.c | 1729 PetscInt *dnnz = NULL; in PCPatchCreateMatrix_Private() local
1751 PetscCall(PetscCalloc1(rsize, &dnnz)); in PCPatchCreateMatrix_Private()
1762 if (!PetscBTLookupSet(bt, key)) ++dnnz[row]; in PCPatchCreateMatrix_Private()
1788 if (!PetscBTLookupSet(bt, key)) ++dnnz[row]; in PCPatchCreateMatrix_Private()
1799 if (!PetscBTLookupSet(bt, key)) ++dnnz[row]; in PCPatchCreateMatrix_Private()
1805 PetscCall(MatXAIJSetPreallocation(*mat, 1, dnnz, NULL, NULL, NULL)); in PCPatchCreateMatrix_Private()
1806 PetscCall(PetscFree(dnnz)); in PCPatchCreateMatrix_Private()
|
| /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()
1554 if (rowner == cowner) dnnz[i]++; in MatPermute_MPIAIJ()
1561 if (rowner == cowner) dnnz[i]++; in MatPermute_MPIAIJ()
1565 PetscCall(PetscSFBcastBegin(rowsf, MPIU_INT, dnnz, tdnnz, MPI_REPLACE)); in MatPermute_MPIAIJ()
1566 PetscCall(PetscSFBcastEnd(rowsf, MPIU_INT, dnnz, tdnnz, 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/mat/interface/ |
| H A D | matrix.c | 10865 PetscInt *dnnz; in MatInvertBlockDiagonalMat() local
10874 PetscCall(PetscMalloc1(m / bs, &dnnz)); in MatInvertBlockDiagonalMat()
10875 for (j = 0; j < m / bs; j++) dnnz[j] = 1; in MatInvertBlockDiagonalMat()
10876 PetscCall(MatXAIJSetPreallocation(C, bs, dnnz, NULL, NULL, NULL)); in MatInvertBlockDiagonalMat()
10877 PetscCall(PetscFree(dnnz)); in MatInvertBlockDiagonalMat()
|