Home
last modified time | relevance | path

Searched refs:csrmat (Results 1 – 4 of 4) sorted by relevance

/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp111 KokkosCsrMatrix csrmat; /* The CSR matrix, used to call KK functions */ member
144 csrmat = KokkosCsrMatrix("csrmat", ncols, a_d, KokkosCsrGraph(j_d, i_d)); in Mat_SeqAIJKokkos()
149 …Mat_SeqAIJKokkos(const KokkosCsrMatrix &csr) : csrmat(csr) /* Shallow-copy csr's views to csrmat */ in Mat_SeqAIJKokkos()
179 MatColIdxType nrows() { return csrmat.numRows(); } in nrows()
180 MatColIdxType ncols() { return csrmat.numCols(); } in ncols()
181 MatRowMapType nnz() { return csrmat.nnz(); } in nnz()
184 …void SetColSize(MatColIdxType n) { csrmat = KokkosCsrMatrix("csrmat", n, a_dual.view_device(), csr… in SetColSize()
222 PETSC_INTERN PetscErrorCode PrintCsrMatrix(const KokkosCsrMatrix &csrmat);
H A Daijkok.kokkos.cxx324 *csrmatT = akok->csrmat; in MatSeqAIJKokkosGenerateTranspose_Private()
362 *csrmatH = akok->csrmat; in MatSeqAIJKokkosGenerateHermitian_Private()
398 …Sparse::spmv(PetscGetKokkosExecutionSpace(), "N", 1.0 /*alpha*/, aijkok->csrmat, xv, 0.0 /*beta*/,… in MatMult_SeqAIJKokkos()
402 PetscCall(PetscLogGpuFlops(2.0 * aijkok->csrmat.nnz())); in MatMult_SeqAIJKokkos()
414 KokkosCsrMatrix csrmat; in MatMultTranspose_SeqAIJKokkos() local
422 PetscCall(MatSeqAIJKokkosGenerateTranspose_Private(A, &csrmat)); in MatMultTranspose_SeqAIJKokkos()
426 csrmat = aijkok->csrmat; in MatMultTranspose_SeqAIJKokkos()
429 …PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 0… in MatMultTranspose_SeqAIJKokkos()
432 PetscCall(PetscLogGpuFlops(2.0 * csrmat.nnz())); in MatMultTranspose_SeqAIJKokkos()
444 KokkosCsrMatrix csrmat; in MatMultHermitianTranspose_SeqAIJKokkos() local
[all …]
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu2660 CsrMatrix *csrmat; in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA() local
2706 csrmat = (CsrMatrix *)mat->mat; in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2759 …ADescr, csrmat->num_rows, csrmat->num_cols, csrmat->num_entries, csrmat->row_offsets->data().get()… in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2779 PetscCallCUSPARSE(cusparseSpMatSetValues(matADescr, csrmat->values->data().get())); in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2803 …, opA, m, n, k, csrmat->num_entries, mat->alpha_one, mat->descr, csrmat->values->data().get(), csr… in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2807 PetscCall(PetscLogGpuFlops(n * 2.0 * csrmat->num_entries)); in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
/petsc/src/mat/impls/aij/seq/seqhipsparse/
H A Daijhipsparse.hip.cxx2226 CsrMatrix *csrmat; in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP() local
2272 csrmat = (CsrMatrix *)mat->mat; in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2308 …tDescr, csrmat->num_rows, csrmat->num_cols, csrmat->num_entries, csrmat->row_offsets->data().get()… in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2320 PetscCallHIPSPARSE(hipsparseSpMatSetValues(mat->matDescr, csrmat->values->data().get())); in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2329 PetscCall(PetscLogGpuFlops(n * 2.0 * csrmat->num_entries)); in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()