Home
last modified time | relevance | path

Searched refs:diag_h (Results 1 – 6 of 6) sorted by relevance

/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp130 MatRowMapKokkosViewHost diag_h(aseq->diag, nrows); in Mat_SeqAIJKokkos() local
135 auto diag_d = Kokkos::create_mirror_view_and_copy(exec, diag_h); in Mat_SeqAIJKokkos()
139 diag_dual = MatColIdxKokkosDualView(diag_d, diag_h); in Mat_SeqAIJKokkos()
188 MatRowMapKokkosViewHost diag_h(const_cast<MatRowMapType *>(diag), nrows()); in SetDiagonal() local
189 …auto diag_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), diag_h); in SetDiagonal()
190 diag_dual = MatRowMapKokkosDualView(diag_d, diag_h); in SetDiagonal()
H A Daijkok.kokkos.cxx102 MatRowMapKokkosViewHost diag_h(aijseq->diag, A->rmap->n); in MatAssemblyEnd_SeqAIJKokkos() local
103 …auto diag_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), diag_h); in MatAssemblyEnd_SeqAIJKokkos()
104 aijkok->diag_dual = MatRowMapKokkosDualView(diag_d, diag_h); in MatAssemblyEnd_SeqAIJKokkos()
/petsc/src/ksp/pc/impls/pbjacobi/kokkos/
H A Dpbjacobi_kok.kokkos.cxx12 PetscScalarKokkosViewHost diag_h(diag_ptr_h, len); in PC_PBJacobi_Kokkos() local
13 … diag_d = Kokkos::create_mirror_view_and_copy(PetscGetKokkosExecutionSpace(), diag_h); in PC_PBJacobi_Kokkos()
14 diag_dual = PetscScalarKokkosDualView(diag_d, diag_h); in PC_PBJacobi_Kokkos()
/petsc/src/ksp/pc/impls/vpbjacobi/cuda/
H A Dvpbjacobi_cuda.cu21 … n, PetscInt nblocks, PetscInt nsize, const PetscInt *bsizes, MatScalar *diag_h) : n(n), nblocks(n… in PC_VPBJacobi_CUDA()
29 PetscCallVoid(UpdateOffsetsOnDevice(bsizes, diag_h)); in PC_VPBJacobi_CUDA()
32 PetscErrorCode UpdateOffsetsOnDevice(const PetscInt *bsizes, MatScalar *diag_h) in UpdateOffsetsOnDevice()
39 PetscCallCUDA(cudaMemcpy(diag_d, diag_h, sizeof(MatScalar) * nsize, cudaMemcpyHostToDevice)); in UpdateOffsetsOnDevice()
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Dcusparsematimpl.h236 …PetscScalar *csrVal_h, *diag_h; // Since LU is done on host, we prepare a factored matrix in r… member
H A Daijcusparse.cu682 fs->diag_h = D; in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
687 D = fs->diag_h; in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
4317 PetscCall(PetscFree(fs->diag_h)); in MatSeqAIJCUSPARSETriFactors_Reset()