seq/kokkos/aijkok.kokkos.cxx

389   ConstPetscScalarKokkosView xv;
395   PetscCall(VecGetKokkosView(xx, &xv));
398   PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), "N", 1.0 /*alpha*/, aijkok->csrmat, xv, 0.0 /*beta*/, yv)); /* y = alpha A x + beta y */
399   PetscCall(VecRestoreKokkosView(xx, &xv));
412   ConstPetscScalarKokkosView xv;
419   PetscCall(VecGetKokkosView(xx, &xv));
429   PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 0.0 /*beta*/, yv)); /* y = alpha A^T x + beta y */
430   PetscCall(VecRestoreKokkosView(xx, &xv));
442   ConstPetscScalarKokkosView xv;
449   PetscCall(VecGetKokkosView(xx, &xv));
459   PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 0.0 /*beta*/, yv)); /* y = alpha A^H x + beta y */
460   PetscCall(VecRestoreKokkosView(xx, &xv));
471   ConstPetscScalarKokkosView xv;
478   PetscCall(VecGetKokkosView(xx, &xv));
481   PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), "N", 1.0 /*alpha*/, aijkok->csrmat, xv, 1.0 /*beta*/, zv)); /* z = alpha A x + beta z */
482   PetscCall(VecRestoreKokkosView(xx, &xv));
494   ConstPetscScalarKokkosView xv;
502   PetscCall(VecGetKokkosView(xx, &xv));
512   PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 1.0 /*beta*/, zv)); /* z = alpha A^T x + beta z */
513   PetscCall(VecRestoreKokkosView(xx, &xv));
525   ConstPetscScalarKokkosView xv;
533   PetscCall(VecGetKokkosView(xx, &xv));
543   PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 1.0 /*beta*/, zv)); /* z = alpha A^H x + beta z */
544   PetscCall(VecRestoreKokkosView(xx, &xv));
1152   PetscScalarKokkosView xv;
1166   PetscCall(VecGetKokkosViewWrite(x, &xv));
1169       if (Adiag(i) < Ai(i + 1)) xv(i) = Aa(Adiag(i));
1170       else xv(i) = 0;
1172   PetscCall(VecRestoreKokkosViewWrite(x, &xv));