| /petsc/src/dm/impls/da/kokkos/ |
| H A D | dagetov.kokkos.cxx | 30 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView_Private() local
48 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView_Private() local
64 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView() local
80 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView() local
96 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView_Private() local
113 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView_Private() local
130 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView() local
146 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView() local
162 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView_Private() local
179 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView_Private() local
[all …]
|
| /petsc/src/vec/vec/tests/ |
| H A D | ex16k.kokkos.cxx | 22 auto kv = Kokkos::View<PetscScalar *>("kv", n); in main() local
|
| /petsc/src/vec/vec/impls/seq/kokkos/ |
| H A D | veckok.kokkos.cxx | 18 static PetscErrorCode VecGetKokkosView_Private(Vec v, PetscScalarKokkosViewType<MemorySpace> *kv, P… in VecGetKokkosView_Private()
33 …tic PetscErrorCode VecRestoreKokkosView_Private(Vec v, PetscScalarKokkosViewType<MemorySpace> *kv,… in VecRestoreKokkosView_Private()
47 PetscErrorCode VecGetKokkosView(Vec v, ConstPetscScalarKokkosViewType<MemorySpace> *kv) in VecGetKokkosView()
62 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecGetKokkosView(Vec v, PetscScalarKokkosView *kv) in VecGetKokkosView()
67 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecRestoreKokkosView(Vec v, PetscScalarKokkosView *kv) in VecRestoreKokkosView()
72 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecGetKokkosViewWrite(Vec v, PetscScalarKokkosView *kv) in VecGetKokkosViewWrite()
77 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecRestoreKokkosViewWrite(Vec v, PetscScalarKokkosView *kv) in VecRestoreKokkosViewWrite()
85 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecGetKokkosView(Vec v, PetscScalarKokkosViewHost *kv) in VecGetKokkosView()
90 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecRestoreKokkosView(Vec v, PetscScalarKokkosViewHost *kv) in VecRestoreKokkosView()
95 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecGetKokkosViewWrite(Vec v, PetscScalarKokkosViewHost *kv) in VecGetKokkosViewWrite()
[all …]
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 120 __global__ static void MatPackCOOValues(const PetscScalar kv[], PetscCount nnz, const PetscCount pe… in MatPackCOOValues()
127 __global__ static void MatAddLocalCOOValues(const PetscScalar kv[], InsertMode imode, PetscCount An… in MatAddLocalCOOValues()
144 __global__ static void MatAddRemoteCOOValues(const PetscScalar kv[], PetscCount Annz2, const PetscC… in MatAddRemoteCOOValues()
|
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 120 __global__ static void MatPackCOOValues(const PetscScalar kv[], PetscCount nnz, const PetscCount pe… in MatPackCOOValues()
127 __global__ static void MatAddLocalCOOValues(const PetscScalar kv[], InsertMode imode, PetscCount An… in MatAddLocalCOOValues()
144 __global__ static void MatAddRemoteCOOValues(const PetscScalar kv[], PetscCount Annz2, const PetscC… in MatAddRemoteCOOValues()
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 1177 PetscErrorCode MatSeqAIJGetKokkosView(Mat A, ConstMatScalarKokkosView *kv) in MatSeqAIJGetKokkosView()
1191 PetscErrorCode MatSeqAIJRestoreKokkosView(Mat A, ConstMatScalarKokkosView *kv) in MatSeqAIJRestoreKokkosView()
1200 PetscErrorCode MatSeqAIJGetKokkosView(Mat A, MatScalarKokkosView *kv) in MatSeqAIJGetKokkosView()
1214 PetscErrorCode MatSeqAIJRestoreKokkosView(Mat A, MatScalarKokkosView *kv) in MatSeqAIJRestoreKokkosView()
1224 PetscErrorCode MatSeqAIJGetKokkosViewWrite(Mat A, MatScalarKokkosView *kv) in MatSeqAIJGetKokkosViewWrite()
1237 PetscErrorCode MatSeqAIJRestoreKokkosViewWrite(Mat A, MatScalarKokkosView *kv) in MatSeqAIJRestoreKokkosViewWrite()
1396 ConstMatScalarKokkosView kv; in MatSetValuesCOO_SeqAIJKokkos() local
|
| /petsc/src/sys/tests/ |
| H A D | ex64.cxx | 271 value_type kv{key, value}; in test_insert() local
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 3770 __global__ static void MatAddCOOValues(const PetscScalar kv[], PetscCount nnz, const PetscCount jma… in MatAddCOOValues()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 4413 __global__ static void MatAddCOOValues(const PetscScalar kv[], PetscCount nnz, const PetscCount jma… in MatAddCOOValues()
|