| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.hpp | 149 …Mat_SeqAIJKokkos(const KokkosCsrMatrix &csr) : csrmat(csr) /* Shallow-copy csr's views to csrmat */ in Mat_SeqAIJKokkos()
151 auto a_d = csr.values; in Mat_SeqAIJKokkos()
153 …MatRowMapKokkosView i_d(const_cast<MatRowMapType *>(csr.graph.row_map.data()), csr.graph.row_map.e… in Mat_SeqAIJKokkos()
154 auto j_d = csr.graph.entries; in Mat_SeqAIJKokkos()
|
| H A D | aijkok.kokkos.cxx | 1595 PETSC_INTERN PetscErrorCode MatSeqAIJKokkosGetKokkosCsrMatrix(Mat A, KokkosCsrMatrix *csr) in MatSeqAIJKokkosGetKokkosCsrMatrix() argument
1599 *csr = static_cast<Mat_SeqAIJKokkos *>(A->spptr)->csrmat; in MatSeqAIJKokkosGetKokkosCsrMatrix()
1603 …PetscErrorCode MatCreateSeqAIJKokkosWithKokkosCsrMatrix(MPI_Comm comm, KokkosCsrMatrix csr, Mat *A) in MatCreateSeqAIJKokkosWithKokkosCsrMatrix() argument
1608 PetscCallCXX(akok = new Mat_SeqAIJKokkos(csr)); in MatCreateSeqAIJKokkosWithKokkosCsrMatrix()
|
| /petsc/src/binding/petsc4py/src/petsc4py/PETSc/ |
| H A D | Mat.pyx | 701 csr: CSRIndicesSpec | None = None,
707 To preallocate the matrix the user can either pass ``nnz`` or ``csr``
720 csr
738 Mat_AllocAIJ(self.mat, nnz, csr)
746 csr: CSRIndicesSpec | None = None,
752 To preallocate the matrix the user can either pass ``nnz`` or ``csr``
765 csr
782 Mat_AllocAIJ(self.mat, nnz, csr)
790 csr: CSRIndicesSpec | None = None,
796 To preallocate the matrix the user can either pass ``nnz`` or ``csr``
[all …]
|
| H A D | PC.pyx | 2113 def setBDDCLocalAdjacency(self, csr: CSRIndicesSpec) -> None:
2120 csr
2131 oi, oj = csr
|
| /petsc/src/mat/impls/hypre/ |
| H A D | mhypre.c | 380 hypre_CSRMatrix *csr; in MatHYPRE_DestroyCOOMat() local
385 csr = hypre_ParCSRMatrixDiag(parcsr); in MatHYPRE_DestroyCOOMat()
386 if (csr) { in MatHYPRE_DestroyCOOMat()
387 mem = hypre_CSRMatrixMemoryLocation(csr); in MatHYPRE_DestroyCOOMat()
388 PetscStackCallExternalVoid("hypre_TFree", hypre_TFree(hypre_CSRMatrixJ(csr), mem)); in MatHYPRE_DestroyCOOMat()
389 PetscStackCallExternalVoid("hypre_TFree", hypre_TFree(hypre_CSRMatrixBigJ(csr), mem)); in MatHYPRE_DestroyCOOMat()
391 csr = hypre_ParCSRMatrixOffd(parcsr); in MatHYPRE_DestroyCOOMat()
392 if (csr) { in MatHYPRE_DestroyCOOMat()
393 mem = hypre_CSRMatrixMemoryLocation(csr); in MatHYPRE_DestroyCOOMat()
394 PetscStackCallExternalVoid("hypre_TFree", hypre_TFree(hypre_CSRMatrixJ(csr), mem)); in MatHYPRE_DestroyCOOMat()
[all …]
|
| /petsc/src/binding/petsc4py/test/ |
| H A D | test_mat_aij.py | 112 csr = [ai, aj]
113 self.A.setPreallocationCSR(csr)
129 csr = [ai, aj, av]
130 self.A.setPreallocationCSR(csr)
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 3845 CsrMatrix *csr; in MatSeqAIJHIPSPARSEGetIJ() local
3854 csr = (CsrMatrix *)cusp->mat->mat; in MatSeqAIJHIPSPARSEGetIJ()
3863 } else *i = csr->row_offsets->data().get(); in MatSeqAIJHIPSPARSEGetIJ()
3865 if (j) *j = csr->column_indices->data().get(); in MatSeqAIJHIPSPARSEGetIJ()
3915 CsrMatrix *csr; in MatSeqAIJHIPSPARSEGetArrayRead() local
3924 csr = (CsrMatrix *)cusp->mat->mat; in MatSeqAIJHIPSPARSEGetArrayRead()
3925 PetscCheck(csr->values, PETSC_COMM_SELF, PETSC_ERR_COR, "Missing HIP memory"); in MatSeqAIJHIPSPARSEGetArrayRead()
3926 *a = csr->values->data().get(); in MatSeqAIJHIPSPARSEGetArrayRead()
3974 CsrMatrix *csr; in MatSeqAIJHIPSPARSEGetArray() local
3983 csr = (CsrMatrix *)cusp->mat->mat; in MatSeqAIJHIPSPARSEGetArray()
[all …]
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 4487 CsrMatrix *csr; in MatSeqAIJCUSPARSEGetIJ() local
4497 csr = (CsrMatrix *)cusp->mat->mat; in MatSeqAIJCUSPARSEGetIJ()
4506 } else *i = csr->row_offsets->data().get(); in MatSeqAIJCUSPARSEGetIJ()
4508 if (j) *j = csr->column_indices->data().get(); in MatSeqAIJCUSPARSEGetIJ()
4559 CsrMatrix *csr; in MatSeqAIJCUSPARSEGetArrayRead() local
4568 csr = (CsrMatrix *)cusp->mat->mat; in MatSeqAIJCUSPARSEGetArrayRead()
4569 PetscCheck(csr->values, PETSC_COMM_SELF, PETSC_ERR_COR, "Missing CUDA memory"); in MatSeqAIJCUSPARSEGetArrayRead()
4570 *a = csr->values->data().get(); in MatSeqAIJCUSPARSEGetArrayRead()
4618 CsrMatrix *csr; in MatSeqAIJCUSPARSEGetArray() local
4627 csr = (CsrMatrix *)cusp->mat->mat; in MatSeqAIJCUSPARSEGetArray()
[all …]
|
| /petsc/src/ksp/pc/impls/bddc/ |
| H A D | bddc.c | 2607 PetscInt *csr, n; in PCBDDCCreateFETIDPOperators_BDDC() local
2611 PetscCall(PetscContainerGetPointer(c, &csr)); in PCBDDCCreateFETIDPOperators_BDDC()
2612 …PetscCall(PCBDDCSetLocalAdjacencyGraph(bddcipc_ctx->bddc, n, csr, csr + (n + 1), PETSC_COPY_VALUES… in PCBDDCCreateFETIDPOperators_BDDC()
|