seq/kokkos/aijkok.kokkos.cxx

324     *csrmatT = akok->csrmat;
362     *csrmatH = akok->csrmat;
398   PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), "N", 1.0 /*alpha*/, aijkok->csrmat, xv, 0.0 /*beta*/, yv)); /* y = alpha A x + beta y */
402   PetscCall(PetscLogGpuFlops(2.0 * aijkok->csrmat.nnz()));
414   KokkosCsrMatrix            csrmat;
422     PetscCall(MatSeqAIJKokkosGenerateTranspose_Private(A, &csrmat));
426     csrmat = aijkok->csrmat;
429   PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 0.0 /*beta*/, yv)); /* y = alpha A^T x + beta y */
432   PetscCall(PetscLogGpuFlops(2.0 * csrmat.nnz()));
444   KokkosCsrMatrix            csrmat;
452     PetscCall(MatSeqAIJKokkosGenerateHermitian_Private(A, &csrmat));
456     csrmat = aijkok->csrmat;
459   PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 0.0 /*beta*/, yv)); /* y = alpha A^H x + beta y */
462   PetscCall(PetscLogGpuFlops(2.0 * csrmat.nnz()));
481   PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), "N", 1.0 /*alpha*/, aijkok->csrmat, xv, 1.0 /*beta*/, zv)); /* z = alpha A x + beta z */
484   PetscCall(PetscLogGpuFlops(2.0 * aijkok->csrmat.nnz()));
496   KokkosCsrMatrix            csrmat;
505     PetscCall(MatSeqAIJKokkosGenerateTranspose_Private(A, &csrmat));
509     csrmat = aijkok->csrmat;
512   PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 1.0 /*beta*/, zv)); /* z = alpha A^T x + beta z */
515   PetscCall(PetscLogGpuFlops(2.0 * csrmat.nnz()));
527   KokkosCsrMatrix            csrmat;
536     PetscCall(MatSeqAIJKokkosGenerateHermitian_Private(A, &csrmat));
540     csrmat = aijkok->csrmat;
543   PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 1.0 /*beta*/, zv)); /* z = alpha A^H x + beta z */
546   PetscCall(PetscLogGpuFlops(2.0 * csrmat.nnz()));
647     PetscCallCXX(atkok = new Mat_SeqAIJKokkos(KokkosCsrMatrix("csrmat", internT)));
851   csrmatA = akok->csrmat;
852   csrmatB = bkok->csrmat;
865   PetscCallCXX(KokkosSparse::spgemm_numeric(pdata->kh, csrmatA, transA, csrmatB, transB, ckok->csrmat));
868   if (spgemmHandle->get_sort_option() != 1) PetscCallCXX(sort_crs_matrix(ckok->csrmat)); /* without sort, mat_tests-ex62_14_seqaijkokkos fails */
907   csrmatA = akok->csrmat;
908   csrmatB = bkok->csrmat;
1331     KokkosSparse::spadd_symbolic(&kh, xkok->csrmat, ykok->csrmat, zcsr);
1332     KokkosSparse::spadd_numeric(&kh, alpha, xkok->csrmat, (PetscScalar)1.0, ykok->csrmat, zcsr);
1599   *csr = static_cast<Mat_SeqAIJKokkos *>(A->spptr)->csrmat;
2358 PETSC_INTERN PetscErrorCode PrintCsrMatrix(const KokkosCsrMatrix &csrmat)
2360   const auto        &iv = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), csrmat.graph.row_map);
2361   const auto        &jv = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), csrmat.graph.entries);
2362   const auto        &av = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), csrmat.values);
2366   PetscInt           m = csrmat.numRows(), n = csrmat.numCols(), nnz = csrmat.nnz();