| /petsc/src/ksp/ksp/tutorials/ |
| H A D | bench_kspsolve.c | 120 PetscInt *coo_i, *coo_j; in FillCOO() local
129 PetscCall(PetscCalloc3(user->nnz, &coo_i, user->nnz, &coo_j, user->nnz, &coo_v)); in FillCOO()
136 coo_i[count] = Ii; in FillCOO()
142 coo_i[count] = Ii; in FillCOO()
148 coo_i[count] = Ii; in FillCOO()
154 coo_i[count] = Ii; in FillCOO()
160 coo_i[count] = Ii; in FillCOO()
166 coo_i[count] = Ii; in FillCOO()
172 coo_i[count] = Ii; in FillCOO()
178 coo_i[count] = Ii; in FillCOO()
[all …]
|
| /petsc/src/mat/utils/ |
| H A D | gcreate.c | 616 const PetscInt *coo_i, *coo_j; in MatSetValuesCOO_Basic() local
628 PetscCall(ISGetIndices(is_coo_i, &coo_i)); in MatSetValuesCOO_Basic()
631 …for (n = 0; n < n_i; n++) PetscCall(MatSetValue(A, coo_i[n], coo_j[n], coo_v ? coo_v[n] : zero, AD… in MatSetValuesCOO_Basic()
632 PetscCall(ISRestoreIndices(is_coo_i, &coo_i)); in MatSetValuesCOO_Basic()
637 PetscErrorCode MatSetPreallocationCOO_Basic(Mat A, PetscCount ncoo, PetscInt coo_i[], PetscInt coo_… in MatSetPreallocationCOO_Basic() argument
653 …for (PetscCount n = 0; n < ncoo; n++) PetscCall(MatSetValue(preallocator, coo_i[n], coo_j[n], zero… in MatSetPreallocationCOO_Basic()
658 PetscCall(ISCreateGeneral(PETSC_COMM_SELF, ncoo_i, coo_i, PETSC_COPY_VALUES, &is_coo_i)); in MatSetPreallocationCOO_Basic()
697 PetscErrorCode MatSetPreallocationCOO(Mat A, PetscCount ncoo, PetscInt coo_i[], PetscInt coo_j[]) in MatSetPreallocationCOO() argument
704 if (ncoo) PetscAssertPointer(coo_i, 3); in MatSetPreallocationCOO()
712 PetscCall((*f)(A, ncoo, coo_i, coo_j)); in MatSetPreallocationCOO()
[all …]
|
| /petsc/src/tao/unconstrained/tutorials/ |
| H A D | rosenbrock4.h | 143 PetscInt *coo_i; in CreateHessian() local
175 PetscCall(PetscMalloc2(nnz_local, &coo_i, nnz_local, &coo_j)); in CreateHessian()
185 coo_i[k + 0] = i; in CreateHessian()
186 coo_i[k + 1] = i; in CreateHessian()
187 coo_i[k + 2] = i + 1; in CreateHessian()
188 coo_i[k + 3] = i + 1; in CreateHessian()
195 PetscCall(MatSetPreallocationCOO(H, nnz_local, coo_i, coo_j)); in CreateHessian()
196 PetscCall(PetscFree2(coo_i, coo_j)); in CreateHessian()
254 PetscInt n_coo, *coo_i, i_start, i_end; in CreateVectors() local
287 PetscCall(PetscMalloc1(n_coo, &coo_i)); in CreateVectors()
[all …]
|
| /petsc/src/mat/tests/ |
| H A D | ex123.c | 178 PetscInt *coo_i, *coo_j; in main() local
190 PetscCall(PetscMalloc3(nnz, &coo_i, nnz, &coo_j, nnz, &coo_v)); in main()
194 coo_i[nnz] = i + rstart; in main()
201 coo_i[nnz] = i + rstart; in main()
212 PetscCall(MatSetPreallocationCOO(A, nnz, coo_i, coo_j)); in main()
234 PetscCall(PetscFree3(coo_i, coo_j, coo_v)); in main()
|
| /petsc/src/snes/tutorials/ |
| H A D | ex55k.kokkos.cxx | 203 PetscInt *coo_i, *coo_j, *ip, *jp; in FormJacobianLocalVec() local
204 …PetscCall(PetscMalloc2(ncoo, &coo_i, ncoo, &coo_j)); /* 5-point stencil such that each row has at … in FormJacobianLocalVec()
206 ip = coo_i; in FormJacobianLocalVec()
249 PetscCall(MatSetPreallocationCOO(jacpre, ncoo, coo_i, coo_j)); in FormJacobianLocalVec()
250 PetscCall(PetscFree2(coo_i, coo_j)); in FormJacobianLocalVec()
|
| /petsc/src/vec/vec/interface/ |
| H A D | vector.c | 188 PetscErrorCode VecSetPreallocationCOO(Vec x, PetscCount ncoo, const PetscInt coo_i[]) in VecSetPreallocationCOO() argument
193 if (ncoo) PetscAssertPointer(coo_i, 3); in VecSetPreallocationCOO()
197 PetscUseTypeMethod(x, setpreallocationcoo, ncoo, coo_i); in VecSetPreallocationCOO()
203 PetscCall(ISCreateGeneral(PETSC_COMM_SELF, ncoo_i, coo_i, PETSC_COPY_VALUES, &is_coo_i)); in VecSetPreallocationCOO()
239 PetscErrorCode VecSetPreallocationCOOLocal(Vec x, PetscCount ncoo, PetscInt coo_i[]) in VecSetPreallocationCOOLocal() argument
247 if (ncoo) PetscAssertPointer(coo_i, 3); in VecSetPreallocationCOOLocal()
251 if (ltog) PetscCall(ISLocalToGlobalMappingApply(ltog, ncoo_i, coo_i, coo_i)); in VecSetPreallocationCOOLocal()
252 PetscCall(VecSetPreallocationCOO(x, ncoo, coo_i)); in VecSetPreallocationCOOLocal()
292 const PetscInt *coo_i; in VecSetValuesCOO() local
301 PetscCall(ISGetIndices(is_coo_i, &coo_i)); in VecSetValuesCOO()
[all …]
|
| /petsc/src/mat/impls/hypre/ |
| H A D | mhypre.c | 503 … n, const PetscInt ii[], const PetscInt jj[], PetscCount *ncoo, PetscInt **coo_i, PetscInt **coo_j) in CSRtoCOO_Private() argument
514 *coo_i = cooi; in CSRtoCOO_Private()
520 …, const HYPRE_Int ii[], const HYPRE_Int jj[], PetscCount *ncoo, PetscInt **coo_i, PetscInt **coo_j) in CSRtoCOO_HYPRE_Int_Private() argument
531 *coo_i = cooi; in CSRtoCOO_HYPRE_Int_Private()
537 static PetscErrorCode MatSeqAIJGetCOO_Private(Mat A, PetscCount *ncoo, PetscInt **coo_i, PetscInt *… in MatSeqAIJGetCOO_Private() argument
546 PetscCall(CSRtoCOO_Private(n, ii, jj, ncoo, coo_i, coo_j)); in MatSeqAIJGetCOO_Private()
553 …reCSRMatrixGetCOO_Private(hypre_CSRMatrix *A, PetscCount *ncoo, PetscInt **coo_i, PetscInt **coo_j) argument
574 PetscCall(CSRtoCOO_HYPRE_Int_Private(n, ii, jj, ncoo, coo_i, coo_j));
649 PetscInt *coo_i, *coo_j; local
668 PetscCall(MatSeqAIJGetCOO_Private(dA, &coo_n, &coo_i, &coo_j));
[all …]
|
| /petsc/src/vec/vec/impls/mpi/cupm/ |
| H A D | vecmpicupm_impl.hpp | 236 …orCode VecMPI_CUPM<T>::SetPreallocationCOO(Vec x, PetscCount ncoo, const PetscInt coo_i[]) noexcept in SetPreallocationCOO() argument
242 PetscCall(VecSetPreallocationCOO_MPI(x, ncoo, coo_i)); in SetPreallocationCOO()
252 x, ncoo, coo_i, dctx, in SetPreallocationCOO()
|
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 44 …atSetPreallocationCOO_MPIAIJHIPSPARSE(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt coo_j[… in MatSetPreallocationCOO_MPIAIJHIPSPARSE() argument
64 PetscCall(PetscGetMemType(coo_i, &mtype)); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
68 PetscCallHIP(hipMemcpy(i, coo_i, coo_n * sizeof(PetscInt), hipMemcpyDeviceToHost)); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
71 i = coo_i; in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
75 PetscCall(MatSetPreallocationCOO_MPIAIJ(mat, coo_n, coo_i, coo_j)); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 44 …MatSetPreallocationCOO_MPIAIJCUSPARSE(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt coo_j[… in MatSetPreallocationCOO_MPIAIJCUSPARSE() argument
64 PetscCall(PetscGetMemType(coo_i, &mtype)); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
68 PetscCallCUDA(cudaMemcpy(i, coo_i, coo_n * sizeof(PetscInt), cudaMemcpyDeviceToHost)); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
71 i = coo_i; in MatSetPreallocationCOO_MPIAIJCUSPARSE()
|
| /petsc/src/vec/vec/impls/mpi/kokkos/ |
| H A D | mpikok.kokkos.cxx | 137 static PetscErrorCode VecSetPreallocationCOO_MPIKokkos(Vec x, PetscCount ncoo, const PetscInt coo_i… in VecSetPreallocationCOO_MPIKokkos() argument
145 PetscCall(VecSetPreallocationCOO_MPI(x, ncoo, coo_i)); in VecSetPreallocationCOO_MPIKokkos()
|
| /petsc/src/mat/impls/is/ |
| H A D | matis.c | 1947 PetscInt *coo_i, *coo_j; in MatConvert_IS_XAIJ() local
1965 PetscCall(PetscMalloc2(ncoo, &coo_i, ncoo, &coo_j)); in MatConvert_IS_XAIJ()
1968 coo_i[j * nr + i] = i; in MatConvert_IS_XAIJ()
1980 PetscCall(PetscMalloc2(ncoo, &coo_i, ncoo, &coo_j)); in MatConvert_IS_XAIJ()
1983 for (PetscInt j = ii[i]; j < ii[i + 1]; j++) coo_i[j] = i; in MatConvert_IS_XAIJ()
1988 PetscCall(MatSetPreallocationCOOLocal(MT, ncoo, coo_i, coo_j)); in MatConvert_IS_XAIJ()
1989 PetscCall(PetscFree2(coo_i, coo_j)); in MatConvert_IS_XAIJ()
3414 static PetscErrorCode MatSetPreallocationCOOLocal_IS(Mat A, PetscCount ncoo, PetscInt coo_i[], Pets… in MatSetPreallocationCOOLocal_IS() argument
3421 PetscCall(MatSetPreallocationCOOLocal(a->A, ncoo, coo_i, coo_j)); in MatSetPreallocationCOOLocal_IS()
3423 PetscCall(MatSetPreallocationCOO(a->A, ncoo, coo_i, coo_j)); in MatSetPreallocationCOOLocal_IS()
[all …]
|
| /petsc/src/vec/vec/impls/seq/ |
| H A D | bvec2.c | 573 PetscErrorCode VecSetPreallocationCOO_Seq(Vec x, PetscCount coo_n, const PetscInt coo_i[]) in VecSetPreallocationCOO_Seq() argument
583 PetscCall(PetscArraycpy(i, coo_i, coo_n)); /* Make a copy since we'll modify it */ in VecSetPreallocationCOO_Seq()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 6339 PetscErrorCode MatSetPreallocationCOO_MPIAIJ(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt … in MatSetPreallocationCOO_MPIAIJ() argument
6374 PetscInt *i1 = coo_i, *j1 = coo_j; in MatSetPreallocationCOO_MPIAIJ()
7089 PetscInt i, j, cp, m, n, M, N, *coo_i, *coo_j; in MatProductSymbolic_MPIAIJBACKEND() local
7444 …PetscCall(PetscMalloc2(ncoo_o, &coo_i, ncoo_o, &coo_j)); /* to collect (i,j) of entries to be sent… in MatProductSymbolic_MPIAIJBACKEND()
7454 PetscInt *coi = coo_i + ncoo_o; in MatProductSymbolic_MPIAIJBACKEND()
7489 …tGraphLayout(mmdata->sf, C->rmap, incoo_o /*nleaves*/, NULL /*ilocal*/, PETSC_OWN_POINTER, coo_i)); in MatProductSymbolic_MPIAIJBACKEND()
7494 …PetscCall(PetscSFGatherBegin(mmdata->sf, MPIU_INT, coo_i, coo_i2 + ncoo_d + ncoo_oown)); /* put (i… in MatProductSymbolic_MPIAIJBACKEND()
7495 PetscCall(PetscSFGatherEnd(mmdata->sf, MPIU_INT, coo_i, coo_i2 + ncoo_d + ncoo_oown)); in MatProductSymbolic_MPIAIJBACKEND()
7498 PetscCall(PetscFree2(coo_i, coo_j)); in MatProductSymbolic_MPIAIJBACKEND()
7501 coo_i = coo_i2; in MatProductSymbolic_MPIAIJBACKEND()
[all …]
|
| /petsc/src/binding/petsc4py/src/petsc4py/PETSc/ |
| H A D | Mat.pyx | 902 def setPreallocationCOO(self, coo_i: Sequence[int], coo_j: Sequence[int]) -> Self:
909 coo_i
924 coo_i = iarray_i(coo_i, &ncoo_i, &ccoo_i)
928 raise ValueError("coo_i and coo_j must have the same length")
934 def setPreallocationCOOLocal(self, coo_i: Sequence[int], coo_j: Sequence[int]) -> Self:
941 coo_i
956 coo_i = iarray_i(coo_i, &ncoo_i, &ccoo_i)
960 raise ValueError("coo_i and coo_j must have the same length")
|
| /petsc/src/vec/vec/impls/mpi/ |
| H A D | pdvec.c | 895 PetscErrorCode VecSetPreallocationCOO_MPI(Vec x, PetscCount coo_n, const PetscInt coo_i[]) in VecSetPreallocationCOO_MPI() argument
919 PetscCall(PetscArraycpy(i1, coo_i, n1)); /* Make a copy since we'll modify it */ in VecSetPreallocationCOO_MPI()
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 2234 …orCode VecSeq_CUPM<T>::SetPreallocationCOO(Vec v, PetscCount ncoo, const PetscInt coo_i[]) noexcept in SetPreallocationCOO() argument
2240 PetscCall(VecSetPreallocationCOO_Seq(v, ncoo, coo_i)); in SetPreallocationCOO()
2241 PetscCall(SetPreallocationCOO_CUPMBase(v, ncoo, coo_i, dctx)); in SetPreallocationCOO()
|
| /petsc/src/vec/vec/impls/seq/kokkos/ |
| H A D | veckok.kokkos.cxx | 1561 static PetscErrorCode VecSetPreallocationCOO_SeqKokkos(Vec x, PetscCount ncoo, const PetscInt coo_i… in VecSetPreallocationCOO_SeqKokkos() argument
1568 PetscCall(VecSetPreallocationCOO_Seq(x, ncoo, coo_i)); in VecSetPreallocationCOO_SeqKokkos()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 1550 static PetscErrorCode MatSetPreallocationCOO_MPIAIJKokkos(Mat mat, PetscCount coo_n, PetscInt coo_i… in MatSetPreallocationCOO_MPIAIJKokkos() argument
1557 …PetscCall(MatSetPreallocationCOO_MPIAIJ(mat, coo_n, coo_i, coo_j)); /* mpiaij->A,B's type is set t… in MatSetPreallocationCOO_MPIAIJKokkos()
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 1367 static PetscErrorCode MatSetPreallocationCOO_SeqAIJKokkos(Mat mat, PetscCount coo_n, PetscInt coo_i… in MatSetPreallocationCOO_SeqAIJKokkos() argument
1376 PetscCall(MatSetPreallocationCOO_SeqAIJ(mat, coo_n, coo_i, coo_j)); in MatSetPreallocationCOO_SeqAIJKokkos()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 3730 …atSetPreallocationCOO_SeqAIJHIPSPARSE(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt coo_j[… in MatSetPreallocationCOO_SeqAIJHIPSPARSE() argument
3739 PetscCall(PetscGetMemType(coo_i, &mtype)); in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3743 PetscCallHIP(hipMemcpy(i, coo_i, coo_n * sizeof(PetscInt), hipMemcpyDeviceToHost)); in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3746 i = coo_i; in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 4372 …MatSetPreallocationCOO_SeqAIJCUSPARSE(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt coo_j[… in MatSetPreallocationCOO_SeqAIJCUSPARSE() argument
4381 PetscCall(PetscGetMemType(coo_i, &mtype)); in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4385 PetscCallCUDA(cudaMemcpy(i, coo_i, coo_n * sizeof(PetscInt), cudaMemcpyDeviceToHost)); in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4388 i = coo_i; in MatSetPreallocationCOO_SeqAIJCUSPARSE()
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aij.c | 4553 PetscErrorCode MatSetPreallocationCOO_SeqAIJ(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt … in MatSetPreallocationCOO_SeqAIJ() argument
4572 i = coo_i; in MatSetPreallocationCOO_SeqAIJ()
|