Home
last modified time | relevance | path

Searched refs:diag_dual (Results 1 – 3 of 3) sorted by relevance

/petsc/src/ksp/pc/impls/pbjacobi/kokkos/
H A Dpbjacobi_kok.kokkos.cxx8 PetscScalarKokkosDualView diag_dual; member
14 diag_dual = PetscScalarKokkosDualView(diag_d, diag_h); in PC_PBJacobi_Kokkos()
20 …PetscCheck(diag_dual.view_host().data() == diag_ptr_h, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Host poin… in Update()
21 PetscCallCXX(diag_dual.modify_host()); /* mark the host has newer data */ in Update()
22 PetscCall(KokkosDualViewSyncDevice(diag_dual, PetscGetKokkosExecutionSpace())); in Update()
37 PetscScalarKokkosView Av = pckok->diag_dual.view_device(); in PCApplyOrTranspose_PBJacobi_Kokkos()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp109 MatRowMapKokkosDualView diag_dual; /* Diagonal pointer, built on demand */ member
139 diag_dual = MatColIdxKokkosDualView(diag_d, diag_h); in Mat_SeqAIJKokkos()
190 diag_dual = MatRowMapKokkosDualView(diag_d, diag_h); in SetDiagonal()
H A Daijkok.kokkos.cxx98 …} else if (A->rmap->n && aijkok->diag_dual.extent(0) == 0) { // MatProduct might directly produce … in MatAssemblyEnd_SeqAIJKokkos()
104 aijkok->diag_dual = MatRowMapKokkosDualView(diag_d, diag_h); in MatAssemblyEnd_SeqAIJKokkos()
1047 const auto &Adiag = aijkok->diag_dual.view_device(); in MatShift_SeqAIJKokkos()
1076 const auto &Adiag = aijkok->diag_dual.view_device(); in MatDiagonalSet_SeqAIJKokkos()
1164 const auto &Adiag = aijkok->diag_dual.view_device(); in MatGetDiagonal_SeqAIJKokkos()
1504 auto Adiag = akok->diag_dual.view_device(); in MatInvertVariableBlockDiagonal_SeqAIJKokkos()