Home
last modified time | relevance | path

Searched refs:diag_d (Results 1 – 4 of 4) sorted by relevance

/petsc/src/ksp/pc/impls/vpbjacobi/cuda/ !
H A Dvpbjacobi_cuda.cu19 MatScalar *diag_d; /* [nsize], store inverse of the point blocks on device */ member
28 PetscCallCUDAVoid(cudaMalloc(&diag_d, sizeof(MatScalar) * nsize)); in PC_VPBJacobi_CUDA()
39 PetscCallCUDA(cudaMemcpy(diag_d, diag_h, sizeof(MatScalar) * nsize, cudaMemcpyHostToDevice)); in UpdateOffsetsOnDevice()
50 PetscCallCUDAVoid(cudaFree(diag_d)); in ~PC_VPBJacobi_CUDA()
126 …MatMultBatched<<<gridSize, 256>>>(n, pcuda->bs_d, pcuda->bs2_d, pcuda->matIdx_d, pcuda->diag_d, xx… in PCApplyOrTranspose_VPBJacobi_CUDA()
/petsc/src/mat/impls/aij/seq/kokkos/ !
H A Daijkok.hpp135 auto diag_d = Kokkos::create_mirror_view_and_copy(exec, diag_h); in Mat_SeqAIJKokkos() local
139 diag_dual = MatColIdxKokkosDualView(diag_d, diag_h); in Mat_SeqAIJKokkos()
189 …auto diag_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), diag_h); in SetDiagonal() local
190 diag_dual = MatRowMapKokkosDualView(diag_d, diag_h); in SetDiagonal()
H A Daijkok.kokkos.cxx103 …auto diag_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), diag_h); in MatAssemblyEnd_SeqAIJKokkos() local
104 aijkok->diag_dual = MatRowMapKokkosDualView(diag_d, diag_h); in MatAssemblyEnd_SeqAIJKokkos()
/petsc/src/ksp/pc/impls/pbjacobi/kokkos/ !
H A Dpbjacobi_kok.kokkos.cxx13 …auto diag_d = Kokkos::create_mirror_view_and_copy(PetscGetKokkosExecutionSpac… in PC_PBJacobi_Kokkos() local
14 diag_dual = PetscScalarKokkosDualView(diag_d, diag_h); in PC_PBJacobi_Kokkos()