| /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()
137 coo_j[count] = Ii - 1 - n - n2; in FillCOO()
143 coo_j[count] = Ii - 1 + n - n2; in FillCOO()
149 coo_j[count] = Ii + 1 - n - n2; in FillCOO()
155 coo_j[count] = Ii + 1 + n - n2; in FillCOO()
161 coo_j[count] = Ii - 1 - n + n2; in FillCOO()
167 coo_j[count] = Ii - 1 + n + n2; in FillCOO()
173 coo_j[count] = Ii + 1 - n + n2; in FillCOO()
179 coo_j[count] = Ii + 1 + n + n2; in FillCOO()
[all …]
|
| /petsc/src/mat/utils/ |
| H A D | gcreate.c | 616 const PetscInt *coo_i, *coo_j; in MatSetValuesCOO_Basic() local
629 PetscCall(ISGetIndices(is_coo_j, &coo_j)); 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()
633 PetscCall(ISRestoreIndices(is_coo_j, &coo_j)); in MatSetValuesCOO_Basic()
637 …cErrorCode MatSetPreallocationCOO_Basic(Mat A, PetscCount ncoo, PetscInt coo_i[], PetscInt coo_j[]) 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()
659 PetscCall(ISCreateGeneral(PETSC_COMM_SELF, ncoo_i, coo_j, PETSC_COPY_VALUES, &is_coo_j)); in MatSetPreallocationCOO_Basic()
697 PetscErrorCode MatSetPreallocationCOO(Mat A, PetscCount ncoo, PetscInt coo_i[], PetscInt coo_j[]) in MatSetPreallocationCOO() argument
705 if (ncoo) PetscAssertPointer(coo_j, 4); in MatSetPreallocationCOO()
712 PetscCall((*f)(A, ncoo, coo_i, coo_j)); in MatSetPreallocationCOO()
[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()
195 coo_j[nnz] = jA[j] + cstart; in main()
202 coo_j[nnz] = garray[jB[j]]; 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()
207 jp = coo_j; in FormJacobianLocalVec()
249 PetscCall(MatSetPreallocationCOO(jacpre, ncoo, coo_i, coo_j)); in FormJacobianLocalVec()
250 PetscCall(PetscFree2(coo_i, coo_j)); in FormJacobianLocalVec()
|
| /petsc/src/tao/unconstrained/tutorials/ |
| H A D | rosenbrock4.h | 144 PetscInt *coo_j; in CreateHessian() local
175 PetscCall(PetscMalloc2(nnz_local, &coo_i, nnz_local, &coo_j)); in CreateHessian()
190 coo_j[k + 0] = i; in CreateHessian()
191 coo_j[k + 1] = i + 1; in CreateHessian()
192 coo_j[k + 2] = i; in CreateHessian()
193 coo_j[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()
|
| /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
515 *coo_j = cooj; 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
532 *coo_j = cooj; in CSRtoCOO_HYPRE_Int_Private()
537 …PetscErrorCode MatSeqAIJGetCOO_Private(Mat A, PetscCount *ncoo, PetscInt **coo_i, PetscInt **coo_j) 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/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 44 …tSetPreallocationCOO_MPIAIJHIPSPARSE(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt coo_j[]) in MatSetPreallocationCOO_MPIAIJHIPSPARSE() argument
69 PetscCallHIP(hipMemcpy(j, coo_j, coo_n * sizeof(PetscInt), hipMemcpyDeviceToHost)); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
72 j = coo_j; 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 …atSetPreallocationCOO_MPIAIJCUSPARSE(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt coo_j[]) in MatSetPreallocationCOO_MPIAIJCUSPARSE() argument
69 PetscCallCUDA(cudaMemcpy(j, coo_j, coo_n * sizeof(PetscInt), cudaMemcpyDeviceToHost)); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
72 j = coo_j; in MatSetPreallocationCOO_MPIAIJCUSPARSE()
|
| /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()
1969 coo_j[j * nr + i] = j; 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()
1989 PetscCall(PetscFree2(coo_i, coo_j)); in MatConvert_IS_XAIJ()
3414 …rrorCode MatSetPreallocationCOOLocal_IS(Mat A, PetscCount ncoo, PetscInt coo_i[], PetscInt coo_j[]) 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/binding/petsc4py/src/petsc4py/PETSc/ |
| H A D | Mat.pyx | 902 def setPreallocationCOO(self, coo_i: Sequence[int], coo_j: Sequence[int]) -> Self:
911 coo_j
925 coo_j = iarray_i(coo_j, &ncoo_j, &ccoo_j)
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:
943 coo_j
957 coo_j = iarray_i(coo_j, &ncoo_j, &ccoo_j)
960 raise ValueError("coo_i and coo_j must have the same length")
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 6339 …orCode MatSetPreallocationCOO_MPIAIJ(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt coo_j[]) 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()
7455 PetscInt *coj = coo_j + ncoo_o; in MatProductSymbolic_MPIAIJBACKEND()
7496 PetscCall(PetscSFGatherBegin(mmdata->sf, MPIU_INT, coo_j, coo_j2 + ncoo_d + ncoo_oown)); in MatProductSymbolic_MPIAIJBACKEND()
7497 PetscCall(PetscSFGatherEnd(mmdata->sf, MPIU_INT, coo_j, coo_j2 + ncoo_d + ncoo_oown)); in MatProductSymbolic_MPIAIJBACKEND()
7498 PetscCall(PetscFree2(coo_i, coo_j)); in MatProductSymbolic_MPIAIJBACKEND()
7502 coo_j = coo_j2; in MatProductSymbolic_MPIAIJBACKEND()
7505 PetscCall(PetscMalloc2(ncoo, &coo_i, ncoo, &coo_j)); in MatProductSymbolic_MPIAIJBACKEND()
[all …]
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 1550 … MatSetPreallocationCOO_MPIAIJKokkos(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt coo_j[]) 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 … MatSetPreallocationCOO_SeqAIJKokkos(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt coo_j[]) 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 …tSetPreallocationCOO_SeqAIJHIPSPARSE(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt coo_j[]) in MatSetPreallocationCOO_SeqAIJHIPSPARSE() argument
3744 PetscCallHIP(hipMemcpy(j, coo_j, coo_n * sizeof(PetscInt), hipMemcpyDeviceToHost)); in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3747 j = coo_j; in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 4372 …atSetPreallocationCOO_SeqAIJCUSPARSE(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt coo_j[]) in MatSetPreallocationCOO_SeqAIJCUSPARSE() argument
4386 PetscCallCUDA(cudaMemcpy(j, coo_j, coo_n * sizeof(PetscInt), cudaMemcpyDeviceToHost)); in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4389 j = coo_j; in MatSetPreallocationCOO_SeqAIJCUSPARSE()
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aij.c | 4553 …orCode MatSetPreallocationCOO_SeqAIJ(Mat mat, PetscCount coo_n, PetscInt coo_i[], PetscInt coo_j[]) in MatSetPreallocationCOO_SeqAIJ() argument
4573 j = coo_j; in MatSetPreallocationCOO_SeqAIJ()
|