Searched refs:xkok (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/vec/vec/impls/seq/kokkos/ |
| H A D | veckok.kokkos.cxx | 677 Vec_Kokkos *xkok = static_cast<Vec_Kokkos *>(xin->spptr); in VecCopy_SeqKokkos() local
686 …PetscCallCXX(Kokkos::deep_copy(exec, ykok->v_dual, xkok->v_dual)); // either cpu2cpu or gpu2cpu, s… in VecCopy_SeqKokkos()
691 …if (xkok->v_dual.need_sync_host()) { // x's device has newer d… in VecCopy_SeqKokkos()
692 PetscCallCXX(Kokkos::deep_copy(exec, yv, xkok->v_dual.view_device())); // gpu2cpu in VecCopy_SeqKokkos()
694 PetscCall(PetscLogGpuToCpu(xkok->v_dual.extent(0) * sizeof(PetscScalar))); in VecCopy_SeqKokkos()
697 …PetscCallCXX(Kokkos::deep_copy(exec, yv, xkok->v_dual.view_host())); // Host view to host view dee… in VecCopy_SeqKokkos()
1459 Vec_Kokkos *xkok = static_cast<Vec_Kokkos *>(xin->spptr); in VecCopySyncState_Kokkos_Private() local
1464 if (xkok->v_dual.need_sync_host()) { in VecCopySyncState_Kokkos_Private()
1466 } else if (xkok->v_dual.need_sync_device()) { in VecCopySyncState_Kokkos_Private()
1489 Vec_Kokkos *xkok = static_cast<Vec_Kokkos *>(x->spptr); in VecGetSubVector_Kokkos_Private() local
[all …]
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 1262 Mat_SeqAIJKokkos *xkok, *ykok, *zkok; in MatAXPY_SeqAIJKokkos() local
1289 xkok = static_cast<Mat_SeqAIJKokkos *>(X->spptr); in MatAXPY_SeqAIJKokkos()
1290 Xa = xkok->a_dual.view_device(); in MatAXPY_SeqAIJKokkos()
1297 MatRowMapKokkosView Xi = xkok->i_dual.view_device(), Yi = ykok->i_dual.view_device(); in MatAXPY_SeqAIJKokkos()
1298 MatColIdxKokkosView Xj = xkok->j_dual.view_device(), Yj = ykok->j_dual.view_device(); in MatAXPY_SeqAIJKokkos()
1331 KokkosSparse::spadd_symbolic(&kh, xkok->csrmat, ykok->csrmat, zcsr); in MatAXPY_SeqAIJKokkos()
1332 KokkosSparse::spadd_numeric(&kh, alpha, xkok->csrmat, (PetscScalar)1.0, ykok->csrmat, zcsr); in MatAXPY_SeqAIJKokkos()
1339 …PetscCall(PetscLogGpuFlops(xkok->a_dual.extent(0) * 2)); // Because we scaled X and then added it … in MatAXPY_SeqAIJKokkos()
|