| /petsc/src/mat/utils/ |
| H A D | gcreate.c | 637 PetscErrorCode MatSetPreallocationCOO_Basic(Mat A, PetscCount ncoo, PetscInt coo_i[], PetscInt coo_… in MatSetPreallocationCOO_Basic() argument
645 PetscCall(PetscIntCast(ncoo, &ncoo_i)); in MatSetPreallocationCOO_Basic()
653 …for (PetscCount n = 0; n < ncoo; n++) PetscCall(MatSetValue(preallocator, coo_i[n], coo_j[n], zero… 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()
705 if (ncoo) PetscAssertPointer(coo_j, 4); in MatSetPreallocationCOO()
712 PetscCall((*f)(A, ncoo, coo_i, coo_j)); in MatSetPreallocationCOO()
714 PetscCall(MatSetPreallocationCOO_Basic(A, ncoo, coo_i, coo_j)); in MatSetPreallocationCOO()
751 PetscErrorCode MatSetPreallocationCOOLocal(Mat A, PetscCount ncoo, PetscInt coo_i[], PetscInt coo_j… in MatSetPreallocationCOOLocal() argument
758 if (ncoo) PetscAssertPointer(coo_i, 3); in MatSetPreallocationCOOLocal()
[all …]
|
| /petsc/src/snes/tutorials/ |
| H A D | ex55k.kokkos.cxx | 201 if (!user->ncoo) { in FormJacobianLocalVec()
202 PetscCount ncoo = ((PetscCount)xm) * ((PetscCount)ym) * 5; in FormJacobianLocalVec() local
204 …PetscCall(PetscMalloc2(ncoo, &coo_i, ncoo, &coo_j)); /* 5-point stencil such that each row has at … in FormJacobianLocalVec()
249 PetscCall(MatSetPreallocationCOO(jacpre, ncoo, coo_i, coo_j)); in FormJacobianLocalVec()
251 user->ncoo = ncoo; in FormJacobianLocalVec()
257 PetscScalarKokkosView coo_v("coo_v", user->ncoo); in FormJacobianLocalVec()
|
| H A D | ex55.h | 13 PetscCount ncoo; /* support for coo setvalues only */ member
|
| H A D | ex55.c | 586 user.ncoo = 0; in main()
|
| /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()
202 PetscCall(PetscIntCast(ncoo, &ncoo_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()
248 PetscCall(PetscIntCast(ncoo, &ncoo_i)); in VecSetPreallocationCOOLocal()
252 PetscCall(VecSetPreallocationCOO(x, ncoo, coo_i)); in VecSetPreallocationCOOLocal()
293 PetscInt ncoo; in VecSetValuesCOO() local
300 PetscCall(ISGetLocalSize(is_coo_i, &ncoo)); in VecSetValuesCOO()
[all …]
|
| /petsc/src/vec/vec/impls/mpi/cupm/ |
| H A D | vecmpicupm_impl.hpp | 236 inline PetscErrorCode VecMPI_CUPM<T>::SetPreallocationCOO(Vec x, PetscCount ncoo, const PetscInt co… 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/hypre/ |
| H A D | mhypre.c | 503 …Private(PetscInt n, const PetscInt ii[], const PetscInt jj[], PetscCount *ncoo, PetscInt **coo_i, … in CSRtoCOO_Private() argument
508 *ncoo = ii[n]; in CSRtoCOO_Private()
509 PetscCall(PetscMalloc2(*ncoo, &cooi, *ncoo, &cooj)); in CSRtoCOO_Private()
513 PetscCall(PetscArraycpy(cooj, jj, *ncoo)); in CSRtoCOO_Private()
520 …ivate(PetscInt n, const HYPRE_Int ii[], const HYPRE_Int jj[], PetscCount *ncoo, PetscInt **coo_i, … in CSRtoCOO_HYPRE_Int_Private() argument
525 *ncoo = ii[n]; in CSRtoCOO_HYPRE_Int_Private()
526 PetscCall(PetscMalloc2(*ncoo, &cooi, *ncoo, &cooj)); in CSRtoCOO_HYPRE_Int_Private()
530 for (PetscCount i = 0; i < *ncoo; i++) cooj[i] = jj[i]; 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()
[all …]
|
| /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 | 1946 PetscCount ncoo; in MatConvert_IS_XAIJ() local
1964 ncoo = nr * nc; in MatConvert_IS_XAIJ()
1965 PetscCall(PetscMalloc2(ncoo, &coo_i, ncoo, &coo_j)); in MatConvert_IS_XAIJ()
1979 ncoo = ii[nr]; in MatConvert_IS_XAIJ()
1980 PetscCall(PetscMalloc2(ncoo, &coo_i, ncoo, &coo_j)); in MatConvert_IS_XAIJ()
1981 PetscCall(PetscArraycpy(coo_j, jj, ncoo)); in MatConvert_IS_XAIJ()
1988 PetscCall(MatSetPreallocationCOOLocal(MT, ncoo, 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/cupm/ |
| H A D | vecseqcupm_impl.hpp | 2234 inline PetscErrorCode VecSeq_CUPM<T>::SetPreallocationCOO(Vec v, PetscCount ncoo, const PetscInt co… 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/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 7090 PetscCount ncoo, ncoo_d, ncoo_o, ncoo_oown; in MatProductSymbolic_MPIAIJBACKEND() local
7492 ncoo = ncoo_d + ncoo_oown + ncoo2; in MatProductSymbolic_MPIAIJBACKEND()
7493 PetscCall(PetscMalloc2(ncoo, &coo_i2, ncoo, &coo_j2)); in MatProductSymbolic_MPIAIJBACKEND()
7504 ncoo = ncoo_d; in MatProductSymbolic_MPIAIJBACKEND()
7505 PetscCall(PetscMalloc2(ncoo, &coo_i, ncoo, &coo_j)); in MatProductSymbolic_MPIAIJBACKEND()
7567 …PetscCall(PetscSFMalloc(mmdata->sf, mmdata->mtype, ncoo * sizeof(PetscScalar), (void **)&mmdata->c… in MatProductSymbolic_MPIAIJBACKEND()
7596 PetscCall(MatSetPreallocationCOO(C, ncoo, coo_i, coo_j)); in MatProductSymbolic_MPIAIJBACKEND()
|
| /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/binding/petsc4py/src/petsc4py/PETSc/ |
| H A D | Mat.pyx | 922 cdef PetscCount ncoo = 0
930 ncoo = <PetscCount> ncoo_i
931 CHKERR(MatSetPreallocationCOO(self.mat, ncoo, ccoo_i, ccoo_j))
954 cdef PetscCount ncoo = 0
962 ncoo = <PetscCount> ncoo_i
963 CHKERR(MatSetPreallocationCOOLocal(self.mat, ncoo, ccoo_i, ccoo_j))
|