Home
last modified time | relevance | path

Searched refs:d_nnz (Results 1 – 25 of 35) sorted by relevance

12

/petsc/src/mat/tests/
H A Dex300.c18 PetscInt d_nnz[3] = {0, 0, 0}; in main() local
32 d_nnz[0] = 1; in main()
33 d_nnz[1] = 1; in main()
34 d_nnz[2] = 1; in main()
41 d_nnz[0] = 1; in main()
42 d_nnz[1] = 1; in main()
45 d_nnz[0] = 1; in main()
51 PetscCall(MatXAIJSetPreallocation(C, 1, d_nnz, o_nnz, NULL, NULL)); in main()
/petsc/src/mat/impls/sbaij/mpi/
H A Dmpiaijsbaij.c14 PetscInt *d_nnz, *o_nnz; in MatConvert_MPIAIJ_MPISBAIJ() local
24 PetscCall(MatConvert_SeqAIJ_SeqSBAIJ_Preallocate(mpimat->A, &d_nnz)); in MatConvert_MPIAIJ_MPISBAIJ()
29 PetscCall(MatSeqSBAIJSetPreallocation(M, bs, 0, d_nnz)); in MatConvert_MPIAIJ_MPISBAIJ()
30 PetscCall(MatMPISBAIJSetPreallocation(M, bs, 0, d_nnz, 0, o_nnz)); in MatConvert_MPIAIJ_MPISBAIJ()
31 PetscCall(PetscFree(d_nnz)); in MatConvert_MPIAIJ_MPISBAIJ()
56 PetscInt *d_nnz, *o_nnz; in MatConvert_MPIBAIJ_MPISBAIJ() local
69 PetscCall(PetscMalloc2(lm / bs, &d_nnz, lm / bs, &o_nnz)); in MatConvert_MPIBAIJ_MPISBAIJ()
73 d_nnz[i] = Aa->i[i + 1] - adiag[i]; in MatConvert_MPIBAIJ_MPISBAIJ()
80 PetscCall(MatSeqSBAIJSetPreallocation(M, bs, 0, d_nnz)); in MatConvert_MPIBAIJ_MPISBAIJ()
81 PetscCall(MatMPISBAIJSetPreallocation(M, bs, 0, d_nnz, 0, o_nnz)); in MatConvert_MPIBAIJ_MPISBAIJ()
[all …]
/petsc/src/mat/impls/aij/mpi/aijsell/
H A Dmpiaijsell.c82 …cInt m, PetscInt n, PetscInt M, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatCreateMPIAIJSELL() argument
92 PetscCall(MatMPIAIJSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateMPIAIJSELL()
95 PetscCall(MatSeqAIJSetPreallocation(*A, d_nz, d_nnz)); in MatCreateMPIAIJSELL()
102 …MatMPIAIJSetPreallocation_MPIAIJSELL(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatMPIAIJSetPreallocation_MPIAIJSELL() argument
107 PetscCall(MatMPIAIJSetPreallocation_MPIAIJ(B, d_nz, d_nnz, o_nz, o_nnz)); in MatMPIAIJSetPreallocation_MPIAIJSELL()
/petsc/src/mat/impls/aij/mpi/aijperm/
H A Dmpiaijperm.c87 …cInt m, PetscInt n, PetscInt M, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatCreateMPIAIJPERM() argument
97 PetscCall(MatMPIAIJSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateMPIAIJPERM()
100 PetscCall(MatSeqAIJSetPreallocation(*A, d_nz, d_nnz)); in MatCreateMPIAIJPERM()
105 …MatMPIAIJSetPreallocation_MPIAIJPERM(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatMPIAIJSetPreallocation_MPIAIJPERM() argument
110 PetscCall(MatMPIAIJSetPreallocation_MPIAIJ(B, d_nz, d_nnz, o_nz, o_nnz)); in MatMPIAIJSetPreallocation_MPIAIJPERM()
/petsc/src/mat/impls/aij/mpi/aijmkl/
H A Dmpiaijmkl.c86 …cInt m, PetscInt n, PetscInt M, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatCreateMPIAIJMKL() argument
96 PetscCall(MatMPIAIJSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateMPIAIJMKL()
99 PetscCall(MatSeqAIJSetPreallocation(*A, d_nz, d_nnz)); in MatCreateMPIAIJMKL()
106 … MatMPIAIJSetPreallocation_MPIAIJMKL(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatMPIAIJSetPreallocation_MPIAIJMKL() argument
111 PetscCall(MatMPIAIJSetPreallocation_MPIAIJ(B, d_nz, d_nnz, o_nz, o_nnz)); in MatMPIAIJSetPreallocation_MPIAIJMKL()
/petsc/src/mat/impls/baij/mpi/
H A Dmpiaijbaij.c12 PetscInt *d_nnz, *o_nnz; in MatConvert_MPIAIJ_MPIBAIJ() local
22 PetscCall(MatConvert_SeqAIJ_SeqBAIJ_Preallocate(mpimat->A, &d_nnz)); in MatConvert_MPIAIJ_MPIBAIJ()
27 PetscCall(MatSeqBAIJSetPreallocation(M, bs, 0, d_nnz)); in MatConvert_MPIAIJ_MPIBAIJ()
28 PetscCall(MatMPIBAIJSetPreallocation(M, bs, 0, d_nnz, 0, o_nnz)); in MatConvert_MPIAIJ_MPIBAIJ()
29 PetscCall(PetscFree(d_nnz)); in MatConvert_MPIAIJ_MPIBAIJ()
H A Dmpibaij.h52 …tPreallocation_MPIBAIJ(Mat B, PetscInt bs, PetscInt d_nz, const PetscInt *d_nnz, PetscInt o_nz, co…
/petsc/src/mat/impls/baij/mpi/baijmkl/
H A Dmpibaijmkl.c5 …eallocation_MPIBAIJMKL(Mat B, PetscInt bs, PetscInt d_nz, const PetscInt *d_nnz, PetscInt o_nz, co… in MatMPIBAIJSetPreallocation_MPIBAIJMKL() argument
10 PetscCall(MatMPIBAIJSetPreallocation_MPIBAIJ(B, bs, d_nz, d_nnz, o_nz, o_nnz)); in MatMPIBAIJSetPreallocation_MPIBAIJMKL()
131 …cInt m, PetscInt n, PetscInt M, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatCreateBAIJMKL() argument
141 PetscCall(MatMPIBAIJSetPreallocation(*A, bs, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateBAIJMKL()
144 PetscCall(MatSeqBAIJSetPreallocation(*A, bs, d_nz, d_nnz)); in MatCreateBAIJMKL()
/petsc/src/dm/impls/sliced/
H A Dsliced.c12 PetscInt d_nz, o_nz, *d_nnz, *o_nnz; member
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()
37 PetscCall(MatSeqAIJSetPreallocation(*J, slice->d_nz, slice->d_nnz)); in DMCreateMatrix_Sliced()
38 … PetscCall(MatMPIAIJSetPreallocation(*J, slice->d_nz, slice->d_nnz, slice->o_nz, slice->o_nnz)); in DMCreateMatrix_Sliced()
39 } else if (!slice->d_nnz) { in DMCreateMatrix_Sliced()
46 …sd_nnz[i] = (slice->d_nnz[i / bs] - 1) * (slice->ofill ? slice->ofill->i[i % bs + 1] - slice->ofil… in DMCreateMatrix_Sliced()
127 PetscErrorCode DMSlicedSetPreallocation(DM dm, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz… in DMSlicedSetPreallocation() argument
134 slice->d_nnz = (PetscInt *)d_nnz; in DMSlicedSetPreallocation()
301 …etscInt nlocal, PetscInt Nghosts, const PetscInt ghosts[], const PetscInt d_nnz[], const PetscInt … in DMSlicedCreate() argument
[all …]
/petsc/src/mat/impls/aij/mpi/mpiviennacl/
H A Dmpiaijviennacl.cxx7 …PIAIJSetPreallocation_MPIAIJViennaCL(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatMPIAIJSetPreallocation_MPIAIJViennaCL() argument
23 PetscCall(MatSeqAIJSetPreallocation(b->A, d_nz, d_nnz)); in MatMPIAIJSetPreallocation_MPIAIJViennaCL()
106 …cInt m, PetscInt n, PetscInt M, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatCreateAIJViennaCL() argument
116 PetscCall(MatMPIAIJSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateAIJViennaCL()
119 PetscCall(MatSeqAIJSetPreallocation(*A, d_nz, d_nnz)); in MatCreateAIJViennaCL()
/petsc/src/dm/tests/
H A Dex30.c11 …PetscInt i, bs = 1, N = 5, n, m, rstart, ghosts[2], *d_nnz, *o_nnz, dfill[4] = {1, 0, 0, 1}, o… in main() local
50 PetscCall(PetscMalloc2(n, &d_nnz, n, &o_nnz)); in main()
53 d_nnz[i] = 2; in main()
56 d_nnz[i] = 3; in main()
60 …PetscCall(DMSlicedCreate(comm, bs, n, 2, ghosts, d_nnz, o_nnz, &slice)); /* Currently does not cop… in main()
65 PetscCall(PetscFree2(d_nnz, o_nnz)); in main()
/petsc/src/mat/impls/sell/mpi/mpihip/
H A Dmpisellhip.hip.cxx136 …cInt m, PetscInt n, PetscInt M, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatCreateSELLHIP() argument
146 PetscCall(MatMPISELLSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateSELLHIP()
149 PetscCall(MatSeqSELLSetPreallocation(*A, d_nz, d_nnz)); in MatCreateSELLHIP()
/petsc/src/mat/impls/sell/mpi/mpicuda/
H A Dmpisellcuda.cu136 …cInt m, PetscInt n, PetscInt M, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatCreateSELLCUDA() argument
146 PetscCall(MatMPISELLSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateSELLCUDA()
149 PetscCall(MatSeqSELLSetPreallocation(*A, d_nz, d_nnz)); in MatCreateSELLCUDA()
/petsc/src/ts/tutorials/power_grid/stability_9bus/
H A Dex9busadj.c383 PetscInt *d_nnz; in PreallocateJacobian() local
388 PetscCall(PetscMalloc1(user->neqs_pgrid, &d_nnz)); in PreallocateJacobian()
389 for (i = 0; i < user->neqs_pgrid; i++) d_nnz[i] = 0; in PreallocateJacobian()
392 d_nnz[idx] += 3; in PreallocateJacobian()
393 d_nnz[idx + 1] += 2; in PreallocateJacobian()
394 d_nnz[idx + 2] += 2; in PreallocateJacobian()
395 d_nnz[idx + 3] += 5; in PreallocateJacobian()
396 d_nnz[idx + 4] += 6; in PreallocateJacobian()
397 d_nnz[idx + 5] += 6; in PreallocateJacobian()
399 d_nnz[user->neqs_gen + 2 * gbus[i]] += 3; in PreallocateJacobian()
[all …]
H A Dex9busoptfd.c384 PetscInt *d_nnz; in PreallocateJacobian() local
389 PetscCall(PetscMalloc1(user->neqs_pgrid, &d_nnz)); in PreallocateJacobian()
390 for (i = 0; i < user->neqs_pgrid; i++) d_nnz[i] = 0; in PreallocateJacobian()
393 d_nnz[idx] += 3; in PreallocateJacobian()
394 d_nnz[idx + 1] += 2; in PreallocateJacobian()
395 d_nnz[idx + 2] += 2; in PreallocateJacobian()
396 d_nnz[idx + 3] += 5; in PreallocateJacobian()
397 d_nnz[idx + 4] += 6; in PreallocateJacobian()
398 d_nnz[idx + 5] += 6; in PreallocateJacobian()
400 d_nnz[user->neqs_gen + 2 * gbus[i]] += 3; in PreallocateJacobian()
[all …]
H A Dex9bus.c644 PetscInt *d_nnz; in PreallocateJacobian() local
649 PetscCall(PetscMalloc1(user->neqs_pgrid, &d_nnz)); in PreallocateJacobian()
650 for (i = 0; i < user->neqs_pgrid; i++) d_nnz[i] = 0; in PreallocateJacobian()
653 d_nnz[idx] += 3; in PreallocateJacobian()
654 d_nnz[idx + 1] += 2; in PreallocateJacobian()
655 d_nnz[idx + 2] += 2; in PreallocateJacobian()
656 d_nnz[idx + 3] += 5; in PreallocateJacobian()
657 d_nnz[idx + 4] += 6; in PreallocateJacobian()
658 d_nnz[idx + 5] += 6; in PreallocateJacobian()
660 d_nnz[user->neqs_gen + 2 * gbus[i]] += 3; in PreallocateJacobian()
[all …]
H A Dex9busopt.c529 PetscInt *d_nnz; in PreallocateJacobian() local
534 PetscCall(PetscMalloc1(user->neqs_pgrid, &d_nnz)); in PreallocateJacobian()
535 for (i = 0; i < user->neqs_pgrid; i++) d_nnz[i] = 0; in PreallocateJacobian()
538 d_nnz[idx] += 3; in PreallocateJacobian()
539 d_nnz[idx + 1] += 2; in PreallocateJacobian()
540 d_nnz[idx + 2] += 2; in PreallocateJacobian()
541 d_nnz[idx + 3] += 5; in PreallocateJacobian()
542 d_nnz[idx + 4] += 6; in PreallocateJacobian()
543 d_nnz[idx + 5] += 6; in PreallocateJacobian()
545 d_nnz[user->neqs_gen + 2 * gbus[i]] += 3; in PreallocateJacobian()
[all …]
/petsc/src/ksp/pc/impls/spai/
H A Dispai.c656 int *d_nnz, *o_nnz; in ConvertMatrixToMat() local
668 PetscCall(PetscMalloc1(m, &d_nnz)); in ConvertMatrixToMat()
670 for (i = 0; i < m; i++) d_nnz[i] = o_nnz[i] = 0; in ConvertMatrixToMat()
676 if ((global_col >= first_diag_col) && (global_col < last_diag_col)) d_nnz[i]++; in ConvertMatrixToMat()
686 PetscCall(MatSeqAIJSetPreallocation(*PB, d_nz, d_nnz)); in ConvertMatrixToMat()
687 PetscCall(MatMPIAIJSetPreallocation(*PB, d_nz, d_nnz, o_nz, o_nnz)); in ConvertMatrixToMat()
698 PetscCall(PetscFree(d_nnz)); in ConvertMatrixToMat()
/petsc/src/ksp/ksp/tutorials/
H A Dex56.c94 PetscInt *d_nnz, *o_nnz, osz[4] = {0, 9, 15, 19}, nbc; in main() local
100 PetscCall(PetscMalloc1(m + 1, &d_nnz)); in main()
110 d_nnz[ic] = 3 * (27 - osz[nbc]); in main()
126 PetscCall(MatSeqAIJSetPreallocation(Amat, 0, d_nnz)); in main()
127 PetscCall(MatMPIAIJSetPreallocation(Amat, 0, d_nnz, 0, o_nnz)); in main()
129 PetscCall(PetscFree(d_nnz)); in main()
/petsc/src/mat/impls/aij/mpi/mpihipsparse/
H A Dmpiaijhipsparse.hip.cxx262 …IAIJSetPreallocation_MPIAIJHIPSPARSE(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatMPIAIJSetPreallocation_MPIAIJHIPSPARSE() argument
275 if (PetscDefined(USE_DEBUG) && d_nnz) { in MatMPIAIJSetPreallocation_MPIAIJHIPSPARSE()
276 …ck(d_nnz[i] >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "d_nnz cannot be less than 0: local r… in MatMPIAIJSetPreallocation_MPIAIJHIPSPARSE()
305 PetscCall(MatSeqAIJSetPreallocation(b->A, d_nz, d_nnz)); in MatMPIAIJSetPreallocation_MPIAIJHIPSPARSE()
532 …cInt m, PetscInt n, PetscInt M, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatCreateAIJHIPSPARSE() argument
542 PetscCall(MatMPIAIJSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateAIJHIPSPARSE()
545 PetscCall(MatSeqAIJSetPreallocation(*A, d_nz, d_nnz)); in MatCreateAIJHIPSPARSE()
/petsc/src/ksp/pc/impls/hmg/
H A Dhmg.c36 PetscInt subrow, row, nz, *d_nnz, *o_nnz, i, j, dnz, onz, max_nz, *indices; in PCHMGExpandInterpolation_Private() local
46 PetscCall(PetscCalloc2(rowsize, &d_nnz, rowsize, &o_nnz)); in PCHMGExpandInterpolation_Private()
61 d_nnz[(subrow - subrstart) * blocksize + i] = dnz; in PCHMGExpandInterpolation_Private()
66 …PetscCall(MatCreateAIJ(comm, rowsize, colsize, PETSC_DETERMINE, PETSC_DETERMINE, 0, d_nnz, 0, o_nn… in PCHMGExpandInterpolation_Private()
73 PetscCall(PetscFree2(d_nnz, o_nnz)); in PCHMGExpandInterpolation_Private()
/petsc/src/mat/impls/aij/mpi/mpicusparse/
H A Dmpiaijcusparse.cu262 …PIAIJSetPreallocation_MPIAIJCUSPARSE(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatMPIAIJSetPreallocation_MPIAIJCUSPARSE() argument
275 if (PetscDefined(USE_DEBUG) && d_nnz) { in MatMPIAIJSetPreallocation_MPIAIJCUSPARSE()
276 …ck(d_nnz[i] >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "d_nnz cannot be less than 0: local r… in MatMPIAIJSetPreallocation_MPIAIJCUSPARSE()
305 PetscCall(MatSeqAIJSetPreallocation(b->A, d_nz, d_nnz)); in MatMPIAIJSetPreallocation_MPIAIJCUSPARSE()
559 …cInt m, PetscInt n, PetscInt M, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, … in MatCreateAIJCUSPARSE() argument
569 PetscCall(MatMPIAIJSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateAIJCUSPARSE()
572 PetscCall(MatSeqAIJSetPreallocation(*A, d_nz, d_nnz)); in MatCreateAIJCUSPARSE()
/petsc/src/binding/petsc4py/src/petsc4py/PETSc/
H A Dpetscmat.pxi798 cdef PetscInt d_nz=PETSC_DECIDE, d_n=0, *d_nnz=NULL
800 od_nnz = iarray_i(od_nnz, &d_n, &d_nnz)
801 if d_n == 0: d_nnz = NULL # just in case
802 elif d_n == 1: d_nz = d_nnz[0]; d_n=0; d_nnz = NULL
813 "size(d_nnz) is %d, expected %d" %
820 CHKERR(MatSeqAIJSetPreallocation(A, d_nz, d_nnz))
821 CHKERR(MatMPIAIJSetPreallocation(A, d_nz, d_nnz, o_nz, o_nnz))
823 CHKERR(MatSeqBAIJSetPreallocation(A, bs, d_nz, d_nnz))
824 CHKERR(MatMPIBAIJSetPreallocation(A, bs, d_nz, d_nnz, o_nz, o_nnz))
826 CHKERR(MatSeqSBAIJSetPreallocation(A, bs, d_nz, d_nnz))
[all …]
/petsc/src/mat/impls/sell/mpi/
H A Dmpisell.c1322 PetscErrorCode MatMPISELLSetPreallocation(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_… in MatMPISELLSetPreallocation() argument
1327 …_C", (Mat, PetscInt, const PetscInt[], PetscInt, const PetscInt[]), (B, d_nz, d_nnz, o_nz, o_nnz)); in MatMPISELLSetPreallocation()
1682 PetscInt *d_nnz, *o_nnz; in MatConvert_MPIAIJ_MPISELL() local
1683 PetscCall(PetscMalloc2(lm, &d_nnz, lm, &o_nnz)); in MatConvert_MPIAIJ_MPISELL()
1685 d_nnz[i] = Aa->i[i + 1] - Aa->i[i]; in MatConvert_MPIAIJ_MPISELL()
1687 if (d_nnz[i] > d_nz) d_nz = d_nnz[i]; in MatConvert_MPIAIJ_MPISELL()
1694 PetscCall(MatSeqSELLSetPreallocation(B, d_nz, d_nnz)); in MatConvert_MPIAIJ_MPISELL()
1695 PetscCall(MatMPISELLSetPreallocation(B, d_nz, d_nnz, o_nz, o_nnz)); in MatConvert_MPIAIJ_MPISELL()
1696 PetscCall(PetscFree2(d_nnz, o_nnz)); in MatConvert_MPIAIJ_MPISELL()
/petsc/doc/developers/
H A Dmatrices.md149 MatCreateBAIJ(MPI_Comm comm,int nb,int m,int n,int M,int N,int d_nz,int *d_nnz,int o_nz,int *o_nnz,…
167 You must set `d_nz=0`, `o_nz=0`, `d_nnz=NULL`, and `o_nnz=NULL` for
171 (`d_nz` and `d_nnz`) and off-diagonal (`o_nz` and `o_nnz`) parts of

12