Home
last modified time | relevance | path

Searched refs:o_nnz (Results 1 – 25 of 30) sorted by relevance

12

/petsc/src/mat/impls/sbaij/mpi/
H A Dmpiaijsbaij.c14 PetscInt *d_nnz, *o_nnz; in MatConvert_MPIAIJ_MPISBAIJ() local
25 PetscCall(MatConvert_SeqAIJ_SeqBAIJ_Preallocate(mpimat->B, &o_nnz)); in MatConvert_MPIAIJ_MPISBAIJ()
30 PetscCall(MatMPISBAIJSetPreallocation(M, bs, 0, d_nnz, 0, o_nnz)); in MatConvert_MPIAIJ_MPISBAIJ()
32 PetscCall(PetscFree(o_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()
74 o_nnz[i] = Ba->i[i + 1] - Ba->i[i]; in MatConvert_MPIBAIJ_MPISBAIJ()
81 PetscCall(MatMPISBAIJSetPreallocation(M, bs, 0, d_nnz, 0, o_nnz)); in MatConvert_MPIBAIJ_MPISBAIJ()
83 PetscCall(PetscFree2(d_nnz, o_nnz)); in MatConvert_MPIBAIJ_MPISBAIJ()
H A Dmpisbaij.c1898 …AIJ(Mat B, PetscInt bs, PetscInt d_nz, const PetscInt *d_nnz, PetscInt o_nz, const PetscInt *o_nnz) in MatMPISBAIJSetPreallocation_MPISBAIJ() argument
1967 PetscCall(MatSeqBAIJSetPreallocation(b->B, bs, o_nz, o_nnz)); in MatMPISBAIJSetPreallocation_MPISBAIJ()
1978 PetscInt i, j, d, nz, bd, nz_max = 0, *d_nnz = NULL, *o_nnz = NULL; in MatMPISBAIJSetPreallocationCSR_MPISBAIJ() local
1996 PetscCall(PetscMalloc2(m, &d_nnz, m, &o_nnz)); in MatMPISBAIJSetPreallocationCSR_MPISBAIJ()
2014 o_nnz[i] = nz - d - bd; in MatMPISBAIJSetPreallocationCSR_MPISBAIJ()
2018 PetscCall(MatMPISBAIJSetPreallocation(B, bs, 0, d_nnz, 0, o_nnz)); in MatMPISBAIJSetPreallocationCSR_MPISBAIJ()
2020 PetscCall(PetscFree2(d_nnz, o_nnz)); in MatMPISBAIJSetPreallocationCSR_MPISBAIJ()
2266 …n(Mat B, PetscInt bs, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[]) in MatMPISBAIJSetPreallocation() argument
2272 …scInt, PetscInt, const PetscInt[], PetscInt, const PetscInt[]), (B, bs, d_nz, d_nnz, o_nz, o_nnz)); in MatMPISBAIJSetPreallocation()
2379 …, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[], Mat *A) in MatCreateSBAIJ() argument
[all …]
/petsc/src/mat/impls/aij/mpi/aijsell/
H A Dmpiaijsell.c82 …, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[], Mat *A) in MatCreateMPIAIJSELL() argument
92 PetscCall(MatMPIAIJSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateMPIAIJSELL()
102 …ion_MPIAIJSELL(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[]) 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 …, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[], Mat *A) in MatCreateMPIAIJPERM() argument
97 PetscCall(MatMPIAIJSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateMPIAIJPERM()
105 …ion_MPIAIJPERM(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[]) 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 …, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[], Mat *A) in MatCreateMPIAIJMKL() argument
96 PetscCall(MatMPIAIJSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateMPIAIJMKL()
106 …tion_MPIAIJMKL(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[]) 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
23 PetscCall(MatConvert_SeqAIJ_SeqBAIJ_Preallocate(mpimat->B, &o_nnz)); in MatConvert_MPIAIJ_MPIBAIJ()
28 PetscCall(MatMPIBAIJSetPreallocation(M, bs, 0, d_nnz, 0, o_nnz)); in MatConvert_MPIAIJ_MPIBAIJ()
30 PetscCall(PetscFree(o_nnz)); in MatConvert_MPIAIJ_MPIBAIJ()
H A Dmpibaij.h52 …IJ(Mat B, PetscInt bs, PetscInt d_nz, const PetscInt *d_nnz, PetscInt o_nz, const PetscInt *o_nnz);
H A Dmpibaij.c2548 PetscInt i, j, dlen, olen, nz, nz_max = 0, *d_nnz = NULL, *o_nnz = NULL; in MatMPIBAIJSetPreallocationCSR_MPIBAIJ() local
2566 PetscCall(PetscMalloc2(m, &d_nnz, m, &o_nnz)); in MatMPIBAIJSetPreallocationCSR_MPIBAIJ()
2580 o_nnz[i] = olen; in MatMPIBAIJSetPreallocationCSR_MPIBAIJ()
2582 PetscCall(MatMPIBAIJSetPreallocation(B, bs, 0, d_nnz, 0, o_nnz)); in MatMPIBAIJSetPreallocationCSR_MPIBAIJ()
2583 PetscCall(PetscFree2(d_nnz, o_nnz)); in MatMPIBAIJSetPreallocationCSR_MPIBAIJ()
2653 …AIJ(Mat B, PetscInt bs, PetscInt d_nz, const PetscInt *d_nnz, PetscInt o_nz, const PetscInt *o_nnz) in MatMPIBAIJSetPreallocation_MPIBAIJ() argument
2673 if (o_nnz) { in MatMPIBAIJSetPreallocation_MPIBAIJ()
2674 …k(o_nnz[i] >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "o_nnz cannot be less than -1: local r… in MatMPIBAIJSetPreallocation_MPIBAIJ()
2715 PetscCall(MatSeqBAIJSetPreallocation(b->B, bs, o_nz, o_nnz)); in MatMPIBAIJSetPreallocation_MPIBAIJ()
3019 …n(Mat B, PetscInt bs, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[]) in MatMPIBAIJSetPreallocation() argument
[all …]
/petsc/src/mat/impls/baij/mpi/baijmkl/
H A Dmpibaijmkl.c5 …MKL(Mat B, PetscInt bs, PetscInt d_nz, const PetscInt *d_nnz, PetscInt o_nz, const PetscInt *o_nnz) in MatMPIBAIJSetPreallocation_MPIBAIJMKL() argument
10 PetscCall(MatMPIBAIJSetPreallocation_MPIBAIJ(B, bs, d_nz, d_nnz, o_nz, o_nnz)); in MatMPIBAIJSetPreallocation_MPIBAIJMKL()
131 …, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[], Mat *A) in MatCreateBAIJMKL() argument
141 PetscCall(MatMPIBAIJSetPreallocation(*A, bs, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateBAIJMKL()
/petsc/src/dm/impls/sliced/
H A Dsliced.c12 PetscInt d_nz, o_nz, *d_nnz, *o_nnz; member
30 …PetscCall(MatMPIBAIJSetPreallocation(*J, bs, slice->d_nz, slice->d_nnz, slice->o_nz, slice->o_nnz)… in DMCreateMatrix_Sliced()
38 … PetscCall(MatMPIAIJSetPreallocation(*J, slice->d_nz, slice->d_nnz, slice->o_nz, slice->o_nnz)); in DMCreateMatrix_Sliced()
44 PetscCall(PetscMalloc2(slice->n * bs, &sd_nnz, (!!slice->o_nnz) * slice->n * bs, &so_nnz)); in DMCreateMatrix_Sliced()
47 …if (so_nnz) so_nnz[i] = slice->o_nnz[i / bs] * (slice->ofill ? slice->ofill->i[i % bs + 1] - slice… in DMCreateMatrix_Sliced()
127 …tPreallocation(DM dm, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[]) in DMSlicedSetPreallocation() argument
136 slice->o_nnz = (PetscInt *)o_nnz; in DMSlicedSetPreallocation()
301 … PetscInt Nghosts, const PetscInt ghosts[], const PetscInt d_nnz[], const PetscInt o_nnz[], DM *dm) in DMSlicedCreate() argument
308 if (d_nnz) PetscCall(DMSlicedSetPreallocation(*dm, 0, d_nnz, 0, o_nnz)); in DMSlicedCreate()
/petsc/src/mat/impls/aij/mpi/mpiviennacl/
H A Dmpiaijviennacl.cxx7 …MPIAIJViennaCL(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[]) in MatMPIAIJSetPreallocation_MPIAIJViennaCL() argument
24 PetscCall(MatSeqAIJSetPreallocation(b->B, o_nz, o_nnz)); in MatMPIAIJSetPreallocation_MPIAIJViennaCL()
106 …, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[], Mat *A) in MatCreateAIJViennaCL() argument
116 PetscCall(MatMPIAIJSetPreallocation(*A, d_nz, d_nnz, o_nz, o_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()
54 o_nnz[i] = 1; in main()
57 o_nnz[i] = 0; 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/tests/
H A Dex300.c19 PetscInt o_nnz[3] = {0, 0, 0}; in main() local
51 PetscCall(MatXAIJSetPreallocation(C, 1, d_nnz, o_nnz, NULL, NULL)); in main()
/petsc/src/ksp/pc/impls/spai/
H A Dispai.c656 int *d_nnz, *o_nnz; in ConvertMatrixToMat() local
669 PetscCall(PetscMalloc1(m, &o_nnz)); in ConvertMatrixToMat()
670 for (i = 0; i < m; i++) d_nnz[i] = o_nnz[i] = 0; in ConvertMatrixToMat()
677 else o_nnz[i]++; in ConvertMatrixToMat()
687 PetscCall(MatMPIAIJSetPreallocation(*PB, d_nz, d_nnz, o_nz, o_nnz)); in ConvertMatrixToMat()
699 PetscCall(PetscFree(o_nnz)); in ConvertMatrixToMat()
/petsc/src/mat/impls/sell/mpi/mpihip/
H A Dmpisellhip.hip.cxx136 …, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[], Mat *A) in MatCreateSELLHIP() argument
146 PetscCall(MatMPISELLSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateSELLHIP()
/petsc/src/mat/impls/sell/mpi/mpicuda/
H A Dmpisellcuda.cu136 …, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[], Mat *A) in MatCreateSELLCUDA() argument
146 PetscCall(MatMPISELLSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateSELLCUDA()
/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()
62 o_nnz[(subrow - subrstart) * blocksize + i] = onz; in PCHMGExpandInterpolation_Private()
66 …atCreateAIJ(comm, rowsize, colsize, PETSC_DETERMINE, PETSC_DETERMINE, 0, d_nnz, 0, o_nnz, interp)); in PCHMGExpandInterpolation_Private()
73 PetscCall(PetscFree2(d_nnz, o_nnz)); in PCHMGExpandInterpolation_Private()
/petsc/src/mat/impls/aij/mpi/mpihipsparse/
H A Dmpiaijhipsparse.hip.cxx262 …PIAIJHIPSPARSE(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[]) in MatMPIAIJSetPreallocation_MPIAIJHIPSPARSE() argument
278 if (PetscDefined(USE_DEBUG) && o_nnz) { in MatMPIAIJSetPreallocation_MPIAIJHIPSPARSE()
279 …ck(o_nnz[i] >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "o_nnz cannot be less than 0: local r… in MatMPIAIJSetPreallocation_MPIAIJHIPSPARSE()
306 PetscCall(MatSeqAIJSetPreallocation(b->B, o_nz, o_nnz)); in MatMPIAIJSetPreallocation_MPIAIJHIPSPARSE()
532 …, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[], Mat *A) in MatCreateAIJHIPSPARSE() argument
542 PetscCall(MatMPIAIJSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateAIJHIPSPARSE()
/petsc/src/ksp/ksp/tutorials/
H A Dex56.c94 PetscInt *d_nnz, *o_nnz, osz[4] = {0, 9, 15, 19}, nbc; in main() local
101 PetscCall(PetscMalloc1(m + 1, &o_nnz)); in main()
111 o_nnz[ic] = 3 * osz[nbc]; in main()
127 PetscCall(MatMPIAIJSetPreallocation(Amat, 0, d_nnz, 0, o_nnz)); in main()
130 PetscCall(PetscFree(o_nnz)); in main()
/petsc/src/mat/impls/aij/mpi/mpicusparse/
H A Dmpiaijcusparse.cu262 …MPIAIJCUSPARSE(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[]) in MatMPIAIJSetPreallocation_MPIAIJCUSPARSE() argument
278 if (PetscDefined(USE_DEBUG) && o_nnz) { in MatMPIAIJSetPreallocation_MPIAIJCUSPARSE()
279 …ck(o_nnz[i] >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "o_nnz cannot be less than 0: local r… in MatMPIAIJSetPreallocation_MPIAIJCUSPARSE()
306 PetscCall(MatSeqAIJSetPreallocation(b->B, o_nz, o_nnz)); in MatMPIAIJSetPreallocation_MPIAIJCUSPARSE()
559 …, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[], Mat *A) in MatCreateAIJCUSPARSE() argument
569 PetscCall(MatMPIAIJSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateAIJCUSPARSE()
/petsc/src/mat/impls/sell/mpi/
H A Dmpisell.c1322 …tPreallocation(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[]) 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()
1686 o_nnz[i] = Ba->i[i + 1] - Ba->i[i]; in MatConvert_MPIAIJ_MPISELL()
1688 if (o_nnz[i] > o_nz) o_nz = o_nnz[i]; 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/src/binding/petsc4py/src/petsc4py/PETSc/
H A Dpetscmat.pxi803 cdef PetscInt o_nz=PETSC_DECIDE, o_n=0, *o_nnz=NULL
805 oo_nnz = iarray_i(oo_nnz, &o_n, &o_nnz)
806 if o_n == 0: o_nnz = NULL # just in case
807 elif o_n == 1: o_nz = o_nnz[0]; o_n=0; o_nnz = NULL
816 "size(o_nnz) is %d, expected %d" %
821 CHKERR(MatMPIAIJSetPreallocation(A, d_nz, d_nnz, o_nz, o_nnz))
824 CHKERR(MatMPIBAIJSetPreallocation(A, bs, d_nz, d_nnz, o_nz, o_nnz))
827 CHKERR(MatMPISBAIJSetPreallocation(A, bs, d_nz, d_nnz, o_nz, o_nnz))
829 CHKERR(MatISSetPreallocation(A, d_nz, d_nnz, o_nz, o_nnz))
/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
/petsc/src/ksp/pc/impls/gamg/
H A Dgamg.c215 PetscInt *d_nnz, *o_nnz, M, N, maxnnz = 0, *j_buf = NULL; in PCGAMGCreateLevel_GAMG() local
220 PetscCall(PetscMalloc2(ncrs, &d_nnz, ncrs, &o_nnz)); in PCGAMGCreateLevel_GAMG()
226 o_nnz[jj] = ncols / cr_bs; in PCGAMGCreateLevel_GAMG()
230 if (o_nnz[jj] > (M / cr_bs - ncrs)) o_nnz[jj] = M / cr_bs - ncrs; in PCGAMGCreateLevel_GAMG()
238 PetscCall(MatMPIAIJSetPreallocation(tMat, 0, d_nnz, 0, o_nnz)); in PCGAMGCreateLevel_GAMG()
239 PetscCall(PetscFree2(d_nnz, o_nnz)); in PCGAMGCreateLevel_GAMG()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.c1904 PetscInt *d_nnz, *g_nnz, *o_nnz; in MatTranspose_MPIAIJ() local
1908 PetscCall(PetscMalloc4(na, &d_nnz, na, &o_nnz, nb, &g_nnz, nb, &oloc)); in MatTranspose_MPIAIJ()
1919 PetscCall(PetscArrayzero(o_nnz, na)); in MatTranspose_MPIAIJ()
1920 PetscCall(PetscSFReduceBegin(sf, MPIU_INT, g_nnz, o_nnz, MPI_SUM)); in MatTranspose_MPIAIJ()
1921 PetscCall(PetscSFReduceEnd(sf, MPIU_INT, g_nnz, o_nnz, MPI_SUM)); in MatTranspose_MPIAIJ()
1928 PetscCall(MatMPIAIJSetPreallocation(B, 0, d_nnz, 0, o_nnz)); in MatTranspose_MPIAIJ()
1929 PetscCall(PetscFree4(d_nnz, o_nnz, g_nnz, oloc)); in MatTranspose_MPIAIJ()
2873 …ocation_MPIAIJ(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[]) in MatMPIAIJSetPreallocation_MPIAIJ() argument
2914 PetscCall(MatSeqAIJSetPreallocation(b->B, o_nz, o_nnz)); in MatMPIAIJSetPreallocation_MPIAIJ()
3869 PetscInt *d_nnz, *o_nnz, nnz_max = 0, rstart, ii, irstart; in MatMPIAIJSetPreallocationCSR_MPIAIJ() local
[all …]

12