Searched refs:ykok (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/vec/vec/impls/seq/kokkos/ |
| H A D | veckok.kokkos.cxx | 680 Vec_Kokkos *ykok = static_cast<Vec_Kokkos *>(yin->spptr); in VecCopy_SeqKokkos() local
685 ykok->v_dual.clear_sync_state(); in VecCopy_SeqKokkos()
686 …PetscCallCXX(Kokkos::deep_copy(exec, ykok->v_dual, xkok->v_dual)); // either cpu2cpu or gpu2cpu, s… in VecCopy_SeqKokkos()
1460 Vec_Kokkos *ykok = static_cast<Vec_Kokkos *>(yout->spptr); in VecCopySyncState_Kokkos_Private() local
1463 PetscCallCXX(ykok->v_dual.clear_sync_state()); in VecCopySyncState_Kokkos_Private()
1465 PetscCallCXX(ykok->v_dual.modify_device()); in VecCopySyncState_Kokkos_Private()
1467 PetscCallCXX(ykok->v_dual.modify_host()); in VecCopySyncState_Kokkos_Private()
1540 Vec_Kokkos *ykok = static_cast<Vec_Kokkos *>((*y)->spptr); in VecRestoreSubVector_SeqKokkos() local
1550 PetscCall(KokkosDualViewSyncHost(ykok->v_dual, exec)); in VecRestoreSubVector_SeqKokkos()
1552 PetscCall(KokkosDualViewSyncDevice(ykok->v_dual, exec)); /* Move y's latest data to device */ in VecRestoreSubVector_SeqKokkos()
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 1262 Mat_SeqAIJKokkos *xkok, *ykok, *zkok; in MatAXPY_SeqAIJKokkos() local
1288 ykok = static_cast<Mat_SeqAIJKokkos *>(Y->spptr); in MatAXPY_SeqAIJKokkos()
1291 Ya = ykok->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()
|