| /petsc/src/dm/impls/da/kokkos/ |
| H A D | dagetov.kokkos.cxx | 30 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView_Private() local
38 if (overwrite) PetscCall(VecGetKokkosViewWrite(vec, &kv)); in DMDAVecGetKokkosOffsetView_Private()
39 else PetscCall(VecGetKokkosView(vec, &kv)); in DMDAVecGetKokkosOffsetView_Private()
41 *ov = PetscScalarKokkosOffsetView1DType<MemorySpace>(kv.data(), {gxs * dof}, {(gxs + gxm) * dof}); in DMDAVecGetKokkosOffsetView_Private()
48 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView_Private() local
54 kv = ov->view(); /* OffsetView to View */ in DMDAVecRestoreKokkosOffsetView_Private()
55 if (overwrite) PetscCall(VecRestoreKokkosViewWrite(vec, &kv)); in DMDAVecRestoreKokkosOffsetView_Private()
56 else PetscCall(VecRestoreKokkosView(vec, &kv)); in DMDAVecRestoreKokkosOffsetView_Private()
64 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView() local
72 PetscCall(VecGetKokkosView(vec, &kv)); in DMDAVecGetKokkosOffsetView()
[all …]
|
| /petsc/src/vec/vec/tests/ |
| H A D | ex16k.kokkos.cxx | 22 auto kv = Kokkos::View<PetscScalar *>("kv", n); in main() local
23 PetscCallCXX(Kokkos::deep_copy(kv, 2.0)); in main()
26 PetscCall(VecKokkosPlaceArray(x, kv.data())); // x = {2.0, 2.0, ...} in main()
31 PetscCall(VecCreateMPIKokkosWithArray(PETSC_COMM_WORLD, 1, n, PETSC_DECIDE, kv.data(), &y)); in main()
|
| /petsc/src/sys/tests/ |
| H A D | ex64.cxx | 271 value_type kv{key, value}; in test_insert() local
273 PetscCall(check_all_reinsert(kv)); in test_insert()
378 for (auto &&kv : sample_values) { in test_find()
379 auto &&key = kv.first; in test_find()
380 auto &&value = kv.second; in test_find()
434 …for (auto &&kv : map) MapCheck(map, kv.first != begin_key, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Erasi… in test_erase()
481 for (auto &&kv : sample_values) PetscCallCXX(map.emplace(kv.first, kv.second)); in test_erase()
504 for (auto &&kv : sample_values) PetscCallCXX(map.erase(kv.first)); in test_erase()
|
| /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() argument
124 for (; i < nnz; i += grid_size) buf[i] = kv[perm[i]]; in MatPackCOOValues()
127 __global__ static void MatAddLocalCOOValues(const PetscScalar kv[], InsertMode imode, PetscCount An… in MatAddLocalCOOValues() argument
134 for (PetscCount k = Ajmap1[i]; k < Ajmap1[i + 1]; k++) sum += kv[Aperm1[k]]; in MatAddLocalCOOValues()
138 for (PetscCount k = Bjmap1[i]; k < Bjmap1[i + 1]; k++) sum += kv[Bperm1[k]]; in MatAddLocalCOOValues()
144 __global__ static void MatAddRemoteCOOValues(const PetscScalar kv[], PetscCount Annz2, const PetscC… in MatAddRemoteCOOValues() argument
150 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += kv[Aperm2[k]]; in MatAddRemoteCOOValues()
153 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += kv[Bperm2[k]]; in MatAddRemoteCOOValues()
|
| /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() argument
124 for (; i < nnz; i += grid_size) buf[i] = kv[perm[i]]; in MatPackCOOValues()
127 __global__ static void MatAddLocalCOOValues(const PetscScalar kv[], InsertMode imode, PetscCount An… in MatAddLocalCOOValues() argument
134 for (PetscCount k = Ajmap1[i]; k < Ajmap1[i + 1]; k++) sum += kv[Aperm1[k]]; in MatAddLocalCOOValues()
138 for (PetscCount k = Bjmap1[i]; k < Bjmap1[i + 1]; k++) sum += kv[Bperm1[k]]; in MatAddLocalCOOValues()
144 __global__ static void MatAddRemoteCOOValues(const PetscScalar kv[], PetscCount Annz2, const PetscC… in MatAddRemoteCOOValues() argument
150 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += kv[Aperm2[k]]; in MatAddRemoteCOOValues()
153 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += kv[Bperm2[k]]; in MatAddRemoteCOOValues()
|
| /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() argument
28 PetscCallCXX(*kv = veckok->v_dual.view<ExecutionSpace>()); in VecGetKokkosView_Private()
33 …ecRestoreKokkosView_Private(Vec v, PetscScalarKokkosViewType<MemorySpace> *kv, PetscBool overwrite) in VecRestoreKokkosView_Private() argument
47 PetscErrorCode VecGetKokkosView(Vec v, ConstPetscScalarKokkosViewType<MemorySpace> *kv) in VecGetKokkosView() argument
55 PetscCallCXX(*kv = veckok->v_dual.view<ExecutionSpace>()); in VecGetKokkosView()
62 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecGetKokkosView(Vec v, PetscScalarKokkosView *kv) in VecGetKokkosView() argument
64 return VecGetKokkosView_Private(v, kv, PETSC_FALSE); in VecGetKokkosView()
67 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecRestoreKokkosView(Vec v, PetscScalarKokkosView *kv) in VecRestoreKokkosView() argument
69 return VecRestoreKokkosView_Private(v, kv, PETSC_FALSE); in VecRestoreKokkosView()
72 PETSC_VISIBILITY_PUBLIC PetscErrorCode VecGetKokkosViewWrite(Vec v, PetscScalarKokkosView *kv) in VecGetKokkosViewWrite() argument
[all …]
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 1177 PetscErrorCode MatSeqAIJGetKokkosView(Mat A, ConstMatScalarKokkosView *kv) in MatSeqAIJGetKokkosView() argument
1183 PetscAssertPointer(kv, 2); in MatSeqAIJGetKokkosView()
1187 *kv = aijkok->a_dual.view_device(); in MatSeqAIJGetKokkosView()
1191 PetscErrorCode MatSeqAIJRestoreKokkosView(Mat A, ConstMatScalarKokkosView *kv) in MatSeqAIJRestoreKokkosView() argument
1195 PetscAssertPointer(kv, 2); in MatSeqAIJRestoreKokkosView()
1200 PetscErrorCode MatSeqAIJGetKokkosView(Mat A, MatScalarKokkosView *kv) in MatSeqAIJGetKokkosView() argument
1206 PetscAssertPointer(kv, 2); in MatSeqAIJGetKokkosView()
1210 *kv = aijkok->a_dual.view_device(); in MatSeqAIJGetKokkosView()
1214 PetscErrorCode MatSeqAIJRestoreKokkosView(Mat A, MatScalarKokkosView *kv) in MatSeqAIJRestoreKokkosView() argument
1218 PetscAssertPointer(kv, 2); in MatSeqAIJRestoreKokkosView()
[all …]
|
| /petsc/config/ |
| H A D | testparse.py | 253 for kv in sep_dicts:
254 kvardict=kv.copy()
|
| /petsc/include/petsc/private/cpp/ |
| H A D | unordered_map.hpp | 1148 PETSC_NODISCARD std::size_t operator()(const std::pair<key_type, T> &kv) const noexcept in operator ()()
1150 return static_cast<const nested_value_type &>(*this)(kv.first); in operator ()()
1158 PETSC_NODISCARD std::size_t operator()(const std::pair<key_type, T> &kv) noexcept in operator ()()
1160 return static_cast<nested_value_type &>(*this)(kv.first); in operator ()()
|
| /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() argument
3776 for (PetscCount k = jmap[i]; k < jmap[i + 1]; k++) sum += kv[perm[k]]; 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() argument
4419 for (PetscCount k = jmap[i]; k < jmap[i + 1]; k++) sum += kv[perm[k]]; in MatAddCOOValues()
|
| /petsc/doc/manual/ |
| H A D | vec.md | 443 VecGetKokkosView(Vec v, Kokkos::View<const PetscScalar*, MemorySpace> *kv)
|
| /petsc/share/petsc/datafiles/meshes/ |
| H A D | testcase3D.cas | 3826 (pb-dqmom/kv 0.523598776)
3839 (pb-smm/kv 0.523598776)
3856 (pb-qmom/kv 0.523598776)
3901 (pb-disc/kv 0.523598776)
|