| /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
582 PetscCall(PetscMalloc1(coo_n, &i)); in VecSetPreallocationCOO_Seq()
583 PetscCall(PetscArraycpy(i, coo_i, coo_n)); /* Make a copy since we'll modify it */ in VecSetPreallocationCOO_Seq()
584 PetscCall(PetscMalloc1(coo_n, &perm1)); in VecSetPreallocationCOO_Seq()
585 for (k = 0; k < coo_n; k++) perm1[k] = k; in VecSetPreallocationCOO_Seq()
586 PetscCall(PetscSortIntWithCountArray(coo_n, i, perm1)); in VecSetPreallocationCOO_Seq()
587 for (k = 0; k < coo_n; k++) { in VecSetPreallocationCOO_Seq()
594 …coo_n || i[coo_n - 1] < m, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Found index (%" PetscInt_FM… in VecSetPreallocationCOO_Seq()
597 for (; k < coo_n; k++) jmap1[i[k] + 1]++; /* Count repeats of each entry */ in VecSetPreallocationCOO_Seq()
603 PetscCall(PetscMalloc1(coo_n - nneg, &perm1_new)); in VecSetPreallocationCOO_Seq()
[all …]
|
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 44 static PetscErrorCode MatSetPreallocationCOO_MPIAIJHIPSPARSE(Mat mat, PetscCount coo_n, PetscInt co… in MatSetPreallocationCOO_MPIAIJHIPSPARSE() argument
67 PetscCall(PetscMalloc2(coo_n, &i, coo_n, &j)); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
68 PetscCallHIP(hipMemcpy(i, coo_i, coo_n * sizeof(PetscInt), hipMemcpyDeviceToHost)); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
69 PetscCallHIP(hipMemcpy(j, coo_j, coo_n * sizeof(PetscInt), hipMemcpyDeviceToHost)); 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 static PetscErrorCode MatSetPreallocationCOO_MPIAIJCUSPARSE(Mat mat, PetscCount coo_n, PetscInt coo… in MatSetPreallocationCOO_MPIAIJCUSPARSE() argument
67 PetscCall(PetscMalloc2(coo_n, &i, coo_n, &j)); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
68 PetscCallCUDA(cudaMemcpy(i, coo_i, coo_n * sizeof(PetscInt), cudaMemcpyDeviceToHost)); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
69 PetscCallCUDA(cudaMemcpy(j, coo_j, coo_n * sizeof(PetscInt), cudaMemcpyDeviceToHost)); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
75 PetscCall(MatSetPreallocationCOO_MPIAIJ(mat, coo_n, i, j)); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
|
| /petsc/src/vec/vec/impls/ |
| H A D | dvecimpl.h | 16 PetscCount coo_n; /* Number of entries in VecSetPreallocationCOO() */ member
|
| /petsc/src/mat/impls/hypre/ |
| H A D | mhypre.c | 648 PetscCount coo_n; local
668 PetscCall(MatSeqAIJGetCOO_Private(dA, &coo_n, &coo_i, &coo_j));
669 PetscCall(MatSetPreallocationCOO(dH, coo_n, coo_i, coo_j));
674 PetscCall(MatSeqAIJGetCOO_Private(oA, &coo_n, &coo_i, &coo_j));
675 PetscCall(MatSetPreallocationCOO(oH, coo_n, coo_i, coo_j));
748 PetscCount coo_n; local
760 PetscCall(hypreCSRMatrixGetCOO_Private(dH, &coo_n, &coo_i, &coo_j));
761 PetscCall(MatSetPreallocationCOO(dA, coo_n, coo_i, coo_j));
768 PetscCall(hypreCSRMatrixGetCOO_Private(oH, &coo_n, &coo_i, &coo_j));
769 PetscCall(MatSetPreallocationCOO(oA, coo_n, coo_i, coo_j));
[all …]
|
| /petsc/src/vec/vec/impls/mpi/kokkos/ |
| H A D | mpikok.kokkos.cxx | 171 …copy(DefaultMemorySpace(), PetscScalarKokkosViewHost(const_cast<PetscScalar *>(v), vecmpi->coo_n)); in VecSetValuesCOO_MPIKokkos()
173 vv = ConstPetscScalarKokkosView(v, vecmpi->coo_n); /* Directly use v[]'s memory */ in VecSetValuesCOO_MPIKokkos()
|
| /petsc/src/vec/vec/impls/mpi/ |
| H A D | pvecimpl.h | 46 PetscCount coo_n; member
|
| H A D | pdvec.c | 895 PetscErrorCode VecSetPreallocationCOO_MPI(Vec x, PetscCount coo_n, const PetscInt coo_i[]) in VecSetPreallocationCOO_MPI() argument
915 PetscCount n1 = coo_n, nneg, *perm; in VecSetPreallocationCOO_MPI()
1093 vmpi->coo_n = coo_n; in VecSetPreallocationCOO_MPI()
|
| /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
4574 PetscCall(PetscMalloc1(coo_n, &perm)); in MatSetPreallocationCOO_SeqAIJ()
4579 for (k = 0; k < coo_n; k++) { in MatSetPreallocationCOO_SeqAIJ()
4589 if (!isorted) PetscCall(PetscSortIntWithIntCountArrayPair(coo_n, i, j, perm)); in MatSetPreallocationCOO_SeqAIJ()
4590 …coo_n == 0 || i[coo_n - 1] < M, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "COO row index %" Petsc… in MatSetPreallocationCOO_SeqAIJ()
4593 for (k = 0; k < coo_n; k++) in MatSetPreallocationCOO_SeqAIJ()
4596 …PetscCall(PetscMalloc1(coo_n - nneg + 1, &jmap)); /* +1 to make a CSR-like data structure. jmap[i]… in MatSetPreallocationCOO_SeqAIJ()
4602 …PetscCall(PetscShmgetAllocateArray(coo_n - nneg, sizeof(PetscInt), (void **)&Aj)); /* We have at m… in MatSetPreallocationCOO_SeqAIJ()
4609 while (k < coo_n) { in MatSetPreallocationCOO_SeqAIJ()
4618 while (k < coo_n && i[k] == row) { in MatSetPreallocationCOO_SeqAIJ()
[all …]
|
| /petsc/src/vec/vec/impls/mpi/cupm/ |
| H A D | vecmpicupm_impl.hpp | 337 const auto size = vmpi->coo_n; in SetValuesCOO()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 3730 static PetscErrorCode MatSetPreallocationCOO_SeqAIJHIPSPARSE(Mat mat, PetscCount coo_n, PetscInt co… in MatSetPreallocationCOO_SeqAIJHIPSPARSE() argument
3742 PetscCall(PetscMalloc2(coo_n, &i, coo_n, &j)); in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3743 PetscCallHIP(hipMemcpy(i, coo_i, coo_n * sizeof(PetscInt), hipMemcpyDeviceToHost)); in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3744 PetscCallHIP(hipMemcpy(j, coo_j, coo_n * sizeof(PetscInt), hipMemcpyDeviceToHost)); in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3749 PetscCall(MatSetPreallocationCOO_SeqAIJ(mat, coo_n, i, j)); in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
|
| /petsc/src/vec/vec/impls/seq/kokkos/ |
| H A D | veckok.kokkos.cxx | 1589 …eate_mirror_view_and_copy(DefaultMemorySpace(), ConstPetscScalarKokkosViewHost(v, vecseq->coo_n))); in VecSetValuesCOO_SeqKokkos()
1591 PetscCallCXX(vv = ConstPetscScalarKokkosView(v, vecseq->coo_n)); /* Directly use v[]'s memory */ in VecSetValuesCOO_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/seqcusparse/ |
| H A D | aijcusparse.cu | 4372 static PetscErrorCode MatSetPreallocationCOO_SeqAIJCUSPARSE(Mat mat, PetscCount coo_n, PetscInt coo… in MatSetPreallocationCOO_SeqAIJCUSPARSE() argument
4384 PetscCall(PetscMalloc2(coo_n, &i, coo_n, &j)); in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4385 PetscCallCUDA(cudaMemcpy(i, coo_i, coo_n * sizeof(PetscInt), cudaMemcpyDeviceToHost)); in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4386 PetscCallCUDA(cudaMemcpy(j, coo_j, coo_n * sizeof(PetscInt), cudaMemcpyDeviceToHost)); in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4392 PetscCall(MatSetPreallocationCOO_SeqAIJ(mat, coo_n, i, j)); in MatSetPreallocationCOO_SeqAIJCUSPARSE()
|
| /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/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 2258 const auto size = VecIMPLCast(x)->coo_n; in SetValuesCOO()
|
| /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
6373 PetscCount n1 = coo_n, *perm1; in MatSetPreallocationCOO_MPIAIJ()
6672 coo->n = coo_n; in MatSetPreallocationCOO_MPIAIJ()
|