Home
last modified time | relevance | path

Searched refs:Kokkos (Results 1 – 25 of 52) sorted by relevance

123

/petsc/include/
H A Dpetscdmda_kokkos.hpp121 PetscErrorCode DMDAVecGetKokkosOffsetView(DM, Vec, Kokkos::Experimental::OffsetView<const PetscScal…
123 PetscErrorCode DMDAVecGetKokkosOffsetView(DM, Vec, Kokkos::Experimental::OffsetView<PetscScalar *, …
125 PetscErrorCode DMDAVecGetKokkosOffsetViewWrite(DM, Vec, Kokkos::Experimental::OffsetView<PetscScala…
128 …ErrorCode DMDAVecGetKokkosOffsetView(DM, Vec, Kokkos::Experimental::OffsetView<const PetscScalar *…
130 …tscErrorCode DMDAVecGetKokkosOffsetView(DM, Vec, Kokkos::Experimental::OffsetView<PetscScalar **, …
132 …rorCode DMDAVecGetKokkosOffsetViewWrite(DM, Vec, Kokkos::Experimental::OffsetView<PetscScalar **, …
135 …rrorCode DMDAVecGetKokkosOffsetView(DM, Vec, Kokkos::Experimental::OffsetView<const PetscScalar **…
137 …scErrorCode DMDAVecGetKokkosOffsetView(DM, Vec, Kokkos::Experimental::OffsetView<PetscScalar ***, …
139 …orCode DMDAVecGetKokkosOffsetViewWrite(DM, Vec, Kokkos::Experimental::OffsetView<PetscScalar ***, …
173 PetscErrorCode DMDAVecRestoreKokkosOffsetView(DM, Vec, Kokkos::Experimental::OffsetView<const Petsc…
[all …]
H A Dpetscvec_kokkos.hpp53 PetscErrorCode VecGetKokkosView(Vec, Kokkos::View<const PetscScalar *, MemorySpace> *);
55 PetscErrorCode VecGetKokkosView(Vec, Kokkos::View<PetscScalar *, MemorySpace> *);
81 PetscErrorCode VecRestoreKokkosView(Vec, Kokkos::View<const PetscScalar *, MemorySpace> *) in VecRestoreKokkosView()
86 PetscErrorCode VecRestoreKokkosView(Vec, Kokkos::View<PetscScalar *, MemorySpace> *);
118 PetscErrorCode VecGetKokkosViewWrite(Vec, Kokkos::View<PetscScalar *, MemorySpace> *);
144 PetscErrorCode VecRestoreKokkosViewWrite(Vec, Kokkos::View<PetscScalar *, MemorySpace> *);
147 static_assert(std::alignment_of<Kokkos::complex<PetscReal>>::value == std::alignment_of<std::comple…
H A Dpetscmat_kokkos.hpp42 …ithKokkosViews(MPI_Comm, PetscInt, PetscInt, Kokkos::View<PetscInt *> &, Kokkos::View<PetscInt *> …
H A Dpetsc_kokkos.hpp7 extern Kokkos::DefaultExecutionSpace *PetscKokkosExecutionSpacePtr;
15 inline Kokkos::DefaultExecutionSpace PetscGetKokkosExecutionSpace(void) in PetscGetKokkosExecutionSpace()
/petsc/src/ts/utils/dmplexlandau/kokkos/
H A Dlandau.kokkos.cxx37 typedef Kokkos::TeamPolicy<>::member_type team_member;
84 namespace Kokkos namespace
96 …const Kokkos::View<pointInterpolationP4est *[LANDAU_MAX_Q_FACE], Kokkos::LayoutRight, Kokkos::Host… in LandauKokkosCreateMatMaps()
97 …const Kokkos::View<LandauIdx *[LANDAU_MAX_SPECIES][LANDAU_MAX_NQND], Kokkos::LayoutRight, Kokkos::… in LandauKokkosCreateMatMaps()
98 …Kokkos::View<pointInterpolationP4est *[LANDAU_MAX_Q_FACE], Kokkos::LayoutRight> *d_points = ne… in LandauKokkosCreateMatMaps()
99 …Kokkos::View<LandauIdx *[LANDAU_MAX_SPECIES][LANDAU_MAX_NQND], Kokkos::LayoutRight> *d_gidx = ne… in LandauKokkosCreateMatMaps()
102 Kokkos::deep_copy(*d_gidx, h_gidx); in LandauKokkosCreateMatMaps()
103 Kokkos::deep_copy(*d_points, h_points); in LandauKokkosCreateMatMaps()
115 Kokkos::View<P4estVertexMaps, Kokkos::HostSpace> h_maps_k(&h_maps); in LandauKokkosCreateMatMaps()
116 …Kokkos::View<P4estVertexMaps> *d_maps_k = new Kokkos::View<P4estVertexMaps>(Kokk… in LandauKokkosCreateMatMaps()
[all …]
/petsc/include/petsc/private/
H A Dkokkosimpl.hpp14 using DefaultExecutionSpace = Kokkos::DefaultExecutionSpace;
15 using DefaultMemorySpace = Kokkos::DefaultExecutionSpace::memory_space;
16 using HostMirrorMemorySpace = Kokkos::DualView<PetscScalar *>::host_mirror_space::memory_space;
25 using PetscScalarKokkosViewType = Kokkos::View<PetscScalar *, MemorySpace>;
27 using PetscScalarKokkosView1DType = Kokkos::View<PetscScalar *, MemorySpace>;
29 using PetscScalarKokkosView2DType = Kokkos::View<PetscScalar **, Kokkos::LayoutRight, MemorySpace>;
31 using PetscScalarKokkosView3DType = Kokkos::View<PetscScalar ***, Kokkos::LayoutRight, MemorySpace>;
33 using PetscScalarKokkosView4DType = Kokkos::View<PetscScalar ****, Kokkos::LayoutRight, MemorySpace…
36 using ConstPetscScalarKokkosViewType = Kokkos::View<const PetscScalar *, MemorySpace>;
38 using ConstPetscScalarKokkosView1DType = Kokkos::View<const PetscScalar *, MemorySpace>;
[all …]
H A Dpcbjkokkosimpl.h32 typedef Kokkos::DefaultExecutionSpace exec_space;
33 using layout = Kokkos::LayoutRight;
34 using IntView = Kokkos::View<PetscInt **, layout, exec_space>;
35 using AMatrixValueView = const Kokkos::View<PetscScalar **, layout, exec_space>;
36 using XYType = const Kokkos::View<PetscScalar **, layout, exec_space>;
43 Kokkos::View<PetscInt *, Kokkos::LayoutRight> *d_bid_eqOffset_k;
44 Kokkos::View<PetscScalar *, Kokkos::LayoutRight> *d_idiag_k;
45 Kokkos::View<PetscInt *> *d_isrow_k;
46 Kokkos::View<PetscInt *> *d_isicol_k;
67 typedef Kokkos::TeamPolicy<>::member_type team_member;
/petsc/src/dm/impls/plex/kokkos/
H A Dplexlocalizationletkf.kokkos.cxx124 using ExecSpace = Kokkos::DefaultExecutionSpace; in DMPlexGetLETKFLocalizationMatrix()
129 …Kokkos::View<PetscScalar **, Kokkos::LayoutLeft, MemSpace> vertex_coords_dev("vertex_coords", n_ve… in DMPlexGetLETKFLocalizationMatrix()
132 …Kokkos::View<PetscScalar **, Kokkos::LayoutLeft, Kokkos::HostSpace> vertex_coords_host("vertex_coo… in DMPlexGetLETKFLocalizationMatrix()
140 Kokkos::deep_copy(vertex_coords_dev, vertex_coords_host); in DMPlexGetLETKFLocalizationMatrix()
144 …Kokkos::View<PetscReal **, Kokkos::LayoutRight, MemSpace> obs_coords_dev("obs_coords", n_obs_globa… in DMPlexGetLETKFLocalizationMatrix()
146 …Kokkos::View<PetscReal **, Kokkos::LayoutRight, Kokkos::HostSpace> obs_coords_host("obs_coords_hos… in DMPlexGetLETKFLocalizationMatrix()
162 Kokkos::deep_copy(obs_coords_dev, obs_coords_host); in DMPlexGetLETKFLocalizationMatrix()
170 …Kokkos::View<PetscInt **, Kokkos::LayoutLeft, MemSpace> indices_dev("indices", n_vert_local, n_… in DMPlexGetLETKFLocalizationMatrix()
171 …Kokkos::View<PetscScalar **, Kokkos::LayoutLeft, MemSpace> values_dev("values", n_vert_local, n_ob… in DMPlexGetLETKFLocalizationMatrix()
178 …Kokkos::View<PetscReal **, Kokkos::LayoutLeft, MemSpace> best_dists_dev("best_dists", n_vert_local… in DMPlexGetLETKFLocalizationMatrix()
[all …]
/petsc/src/mat/tests/
H A Dex302k.kokkos.cxx10 using HostMirrorMemorySpace = Kokkos::DualView<PetscScalar *>::host_mirror_space::memory_space;
73 Kokkos::View<PetscScalar *> a_local_d; in main()
74 Kokkos::View<PetscInt *> i_local_d; in main()
75 Kokkos::View<PetscInt *> j_local_d; in main()
78 Kokkos::View<PetscScalar *> a_nonlocal_d; in main()
79 Kokkos::View<PetscInt *> i_nonlocal_d; in main()
80 Kokkos::View<PetscInt *> j_nonlocal_d; in main()
83 PetscCallCXX(a_local_d = Kokkos::View<PetscScalar *>("a_local_d", di[5])); in main()
84 PetscCallCXX(i_local_d = Kokkos::View<PetscInt *>("i_local_d", m + 1)); in main()
85 PetscCallCXX(j_local_d = Kokkos::View<PetscInt *>("j_local_d", di[5])); in main()
[all …]
/petsc/src/ksp/pc/impls/bjacobi/bjkokkos/
H A Dbjkokkoskernels.kokkos.cxx81 …auto d = Kokkos::subview(_D, Kokkos::make_pair(first_matrix, last_matrix),… in operator ()()
82 …auto x = Kokkos::subview(_X, Kokkos::make_pair(first_matrix, last_matrix),… in operator ()()
83 …auto b = Kokkos::subview(_B, Kokkos::make_pair(first_matrix, last_matrix),… in operator ()()
84 …using ScratchPadIntViewType = Kokkos::View<typename IntView::non_const_value_type *, typename I… in operator ()()
85 …using ScratchPadValuesViewType = Kokkos::View<typename ValuesViewType::non_const_value_type **, ty… in operator ()()
91 TeamVectorCopy1D::invoke(member, Kokkos::subview(_r, graphID, Kokkos::ALL), r); in operator ()()
92 TeamVectorCopy1D::invoke(member, Kokkos::subview(_c, graphID, Kokkos::ALL), c); in operator ()()
98 …VectorCopy<MemberType>::invoke(member, Kokkos::subview(_diag, Kokkos::make_pair(first_matrix, last… in operator ()()
108 Kokkos::Timer timer; in run()
109 Kokkos::Profiling::pushRegion(name.c_str()); in run()
[all …]
H A Dbjkokkos.kokkos.cxx48 Kokkos::parallel_for(Kokkos::TeamThreadRange(team, start, end), [=](const int rowb) { in MatMult()
54 …Kokkos::parallel_reduce(Kokkos::ThreadVectorRange(team, n), [=](const int i, PetscScalar &lsum) { … in MatMult()
55 Kokkos::single(Kokkos::PerThread(team), [=]() { y_loc[rowb - start] = sum; }); in MatMult()
64 Kokkos::parallel_for(Kokkos::TeamVectorRange(team, end - start), [=](int i) { y_loc[i] = 0; }); in MatMultTranspose()
66 Kokkos::parallel_for(Kokkos::TeamThreadRange(team, start, end), [=](const int rowb) { in MatMultTranspose()
72 Kokkos::parallel_for(Kokkos::ThreadVectorRange(team, n), [=](const int &i) { in MatMultTranspose()
74 Kokkos::atomic_fetch_add(&y_loc[r[aj[i]] - start], val); in MatMultTranspose()
90 using Kokkos::parallel_for; in BJSolve_TFQMR()
91 using Kokkos::parallel_reduce; in BJSolve_TFQMR()
158 parallel_for(Kokkos::TeamVectorRange(team, start, end), [=](int rowb) { in BJSolve_TFQMR()
[all …]
/petsc/src/sys/objects/kokkos/
H A Dkinit.kokkos.cxx11 Kokkos::DefaultExecutionSpace *PetscKokkosExecutionSpacePtr = nullptr;
18 PetscCallCXX(Kokkos::kokkos_free(PetscScalarPool)); in PetscKokkosFinalize_Private()
21 PetscCallCXX(Kokkos::finalize()); in PetscKokkosFinalize_Private()
30 *isInitialized = Kokkos::is_initialized() ? PETSC_TRUE : PETSC_FALSE; in PetscKokkosIsInitialized_Private()
38 if (!Kokkos::is_initialized()) { in PetscKokkosInitializeCheck()
40 auto args = Kokkos::InitializationSettings(); in PetscKokkosInitializeCheck()
42 auto args = Kokkos::InitArguments{}; /* use default constructor */ in PetscKokkosInitializeCheck()
72 PetscCallCXX(Kokkos::initialize(args)); in PetscKokkosInitializeCheck()
85 …if (dtype == PETSC_DEVICE_CUDA) PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExe… in PetscKokkosInitializeCheck()
87 …if (dtype == PETSC_DEVICE_HIP) PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExec… in PetscKokkosInitializeCheck()
[all …]
/petsc/src/vec/is/sf/tests/
H A Dex4k.kokkos.cxx50 Kokkos::initialize(argc, argv); // Test initializing kokkos before petsc in main()
76 PetscCallCXX(rootdata = (PetscScalar *)Kokkos::kokkos_malloc(sizeof(PetscScalar) * maxn)); in main()
77 PetscCallCXX(leafdata = (PetscScalar *)Kokkos::kokkos_malloc(sizeof(PetscScalar) * maxn)); in main()
117 Kokkos::View<PetscScalar *> dst1((PetscScalar *)rootdata, n); in main()
118 Kokkos::View<PetscScalar *> dst2((PetscScalar *)leafdata, n); in main()
119 Kokkos::View<const PetscScalar *, Kokkos::HostSpace> src1((const PetscScalar *)pbuf, n); in main()
120 Kokkos::View<const PetscScalar *, Kokkos::HostSpace> src2((const PetscScalar *)ebuf, n); in main()
121 PetscCallCXX(Kokkos::deep_copy(dst1, src1)); in main()
122 PetscCallCXX(Kokkos::deep_copy(dst2, src2)); in main()
133 PetscCallCXX(Kokkos::fence()); in main()
[all …]
/petsc/src/dm/tests/
H A Dex10k.kokkos.cxx7 using Kokkos::Iterate;
8 using Kokkos::MDRangePolicy;
9 using Kokkos::Rank;
10 … PetscScalarKokkosOffsetView2D = Kokkos::Experimental::OffsetView<PetscScalar **, Kokkos::Lay…
11 …nstPetscScalarKokkosOffsetView2D = Kokkos::Experimental::OffsetView<const PetscScalar **, Kokkos::…
13 … PetscScalarKokkosOffsetView3D = Kokkos::Experimental::OffsetView<PetscScalar ***, Kokkos::La…
14 …nstPetscScalarKokkosOffsetView3D = Kokkos::Experimental::OffsetView<const PetscScalar ***, Kokkos:…
24 using NodeKokkosOffsetView2D = Kokkos::Experimental::OffsetView<Node **, Kokkos::LayoutRight, …
25 using ConstNodeKokkosOffsetView2D = Kokkos::Experimental::OffsetView<const Node **, Kokkos::LayoutR…
69 Kokkos::parallel_for( in main()
[all …]
H A Dex2k.kokkos.cxx24 using Kokkos::Iterate;
25 using Kokkos::MDRangePolicy;
26 using Kokkos::Rank;
27 using HostMirrorMemorySpace = Kokkos::DualView<PetscScalar *>::host_mirror_space::memo…
28 using PetscScalarKokkosOffsetView3D = Kokkos::Experimental::OffsetView<PetscScalar ***, Kokkos…
29 using ConstPetscScalarKokkosOffsetView3D = Kokkos::Experimental::OffsetView<const PetscScalar ***, …
145 Kokkos::parallel_for( in main()
146 …"stencil", MDRangePolicy<Kokkos::DefaultHostExecutionSpace, Rank<3, Iterate::Right, Iterate::Right… in main()
/petsc/src/mat/impls/hypre/kokkos/
H A Dhypre3.kokkos.cxx10 Kokkos::parallel_for( in MatZeroRows_Kokkos()
11 …Kokkos::TeamPolicy<>(n, Kokkos::AUTO()), KOKKOS_LAMBDA(const Kokkos::TeamPolicy<>::member_type &t)… in MatZeroRows_Kokkos()
13 Kokkos::parallel_for(Kokkos::TeamThreadRange(t, i[r + 1] - i[r]), [&](PetscInt c) { in MatZeroRows_Kokkos()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp14 return Kokkos::view_alloc(Kokkos::WithoutInitializing, std::move(label)); in NoInit()
49 using MatRowMapKokkosDualView = Kokkos::DualView<MatRowMapType *>;
50 using MatColIdxKokkosDualView = Kokkos::DualView<MatColIdxType *>;
51 using MatScalarKokkosDualView = Kokkos::DualView<MatScalarType *>;
55 using KokkosTeamMemberType = Kokkos::TeamPolicy<DefaultExecutionSpace>::member_type;
132 auto a_d = Kokkos::create_mirror_view(Kokkos::WithoutInitializing, exec, a_h); in Mat_SeqAIJKokkos()
133 auto i_d = Kokkos::create_mirror_view_and_copy(exec, i_h); in Mat_SeqAIJKokkos()
134 auto j_d = Kokkos::create_mirror_view_and_copy(exec, j_h); in Mat_SeqAIJKokkos()
135 auto diag_d = Kokkos::create_mirror_view_and_copy(exec, diag_h); in Mat_SeqAIJKokkos()
155 …auto a_h = Kokkos::create_mirror_view(Kokkos::WithoutInitializing, HostMirrorMemory… in Mat_SeqAIJKokkos()
[all …]
H A Daijkok.kokkos.cxx103 …auto diag_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), diag_h); in MatAssemblyEnd_SeqAIJKokkos()
277 PetscCallCXX(Kokkos::deep_copy(Ti_h, 0)); in MatSeqAIJKokkosGenerateTransposeStructure()
301 auto Ti_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Ti_h); in MatSeqAIJKokkosGenerateTransposeStructure()
302 auto Tj_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Tj_h); in MatSeqAIJKokkosGenerateTransposeStructure()
304 PetscCallCXX(perm_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), perm_h)); in MatSeqAIJKokkosGenerateTransposeStructure()
332 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KO… in MatSeqAIJKokkosGenerateTranspose_Private()
339 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KO… in MatSeqAIJKokkosGenerateTranspose_Private()
370 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KO… in MatSeqAIJKokkosGenerateHermitian_Private()
377 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KO… in MatSeqAIJKokkosGenerateHermitian_Private()
616 Kokkos::deep_copy(bkok->a_dual.view_host(), akok->a_dual.view_host()); in MatDuplicate_SeqAIJKokkos()
[all …]
/petsc/src/sys/objects/device/tests/
H A Dex1k.kokkos.cxx38 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in main()
39 Kokkos::RangePolicy<Kokkos::DefaultExecutionSpace> policy(exec, 0, N); in main()
44 …for (i = 0; i < n; i++) PetscCallCXX(Kokkos::parallel_for(policy, KOKKOS_LAMBDA(const PetscInt &i)… in main()
53 PetscCallCXX(Kokkos::parallel_for(policy, KOKKOS_LAMBDA(const PetscInt &i){})); in main()
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckok.kokkos.cxx141 PetscCallCXX(Kokkos::parallel_for( in VecReciprocal_SeqKokkos()
142 …"VecReciprocal", Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, xin->map->n), KOKKOS_LAM… in VecReciprocal_SeqKokkos()
153 Kokkos::MinFirstLoc<PetscReal, PetscInt>::value_type result; in VecMin_SeqKokkos()
158 PetscCallCXX(Kokkos::parallel_reduce( in VecMin_SeqKokkos()
159 "VecMin", Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, xin->map->n), in VecMin_SeqKokkos()
160 … KOKKOS_LAMBDA(const PetscInt &i, Kokkos::MinFirstLoc<PetscReal, PetscInt>::value_type &lupdate) { in VecMin_SeqKokkos()
166 Kokkos::MinFirstLoc<PetscReal, PetscInt>(result))); in VecMin_SeqKokkos()
177 Kokkos::MaxFirstLoc<PetscReal, PetscInt>::value_type result; in VecMax_SeqKokkos()
182 PetscCallCXX(Kokkos::parallel_reduce( in VecMax_SeqKokkos()
183 "VecMax", Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, xin->map->n), in VecMax_SeqKokkos()
[all …]
H A Dveckokkosimpl.hpp54 …v_d = Kokkos::create_mirror_view(Kokkos::WithoutInitializing, DefaultMemorySpace(), v_h); /* Creat… in Vec_Kokkos()
103 …PetscCallCXX(jmap1_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
104 …PetscCallCXX(perm1_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
111 …PetscCallCXX(jmap1_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
112 …PetscCallCXX(perm1_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
113 …PetscCallCXX(imap2_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
114 …PetscCallCXX(jmap2_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
115 …PetscCallCXX(perm2_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
116 …PetscCallCXX(Cperm_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
117 …PetscCallCXX(sendbuf_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscScalarKokk… in SetUpCOO()
[all …]
/petsc/src/vec/is/sf/impls/basic/kokkos/
H A Dsfkok.kokkos.cxx6 using DeviceExecutionSpace = Kokkos::DefaultExecutionSpace;
8 typedef Kokkos::View<char *, DefaultMemorySpace> deviceBuffer_t;
9 typedef Kokkos::View<char *, HostMirrorMemorySpace> HostBuffer_t;
11 typedef Kokkos::View<const char *, DefaultMemorySpace> deviceConstBuffer_t;
12 typedef Kokkos::View<const char *, HostMirrorMemorySpace> HostConstBuffer_t;
142 KOKKOS_INLINE_FUNCTION void operator()(Type &x, Type y) const { Kokkos::atomic_store(&x, y); } in operator ()()
146 KOKKOS_INLINE_FUNCTION void operator()(Type &x, Type y) const { Kokkos::atomic_add(&x, y); } in operator ()()
150 KOKKOS_INLINE_FUNCTION void operator()(Type &x, Type y) const { Kokkos::atomic_and(&x, y); } in operator ()()
154 KOKKOS_INLINE_FUNCTION void operator()(Type &x, Type y) const { Kokkos::atomic_or(&x, y); } in operator ()()
158 KOKKOS_INLINE_FUNCTION void operator()(Type &x, Type y) const { Kokkos::atomic_fetch_xor(&x, y); } in operator ()()
[all …]
/petsc/src/snes/tutorials/
H A Dex55k.kokkos.cxx13 using DefaultMemorySpace = Kokkos::DefaultExecutionSpace::memory_space;
14 using ConstPetscScalarKokkosOffsetView2D = Kokkos::Experimental::OffsetView<const PetscScalar **, K…
15 using PetscScalarKokkosOffsetView2D = Kokkos::Experimental::OffsetView<PetscScalar **, Kokkos:…
17 using PetscCountKokkosView = Kokkos::View<PetscCount *, DefaultMemorySpace>;
18 using PetscIntKokkosView = Kokkos::View<PetscInt *, DefaultMemorySpace>;
19 using PetscScalarKokkosView = Kokkos::View<PetscScalar *, DefaultMemorySpace>;
20 using Kokkos::Iterate;
21 using Kokkos::MDRangePolicy;
22 using Kokkos::Rank;
46 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FormFunctionLocalVec()
[all …]
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx191 Kokkos::TeamPolicy<ExecutionSpace> teamPolicy(128, Kokkos::AUTO); in MatMergeGetLaunchParameters()
352 …auto Ai_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), A.graph.r… in MatMPIAIJKokkosReduceBegin()
353 …auto Aj_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), A.graph.e… in MatMPIAIJKokkosReduceBegin()
354 …auto Bi_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), B.graph.r… in MatMPIAIJKokkosReduceBegin()
355 …auto Bj_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), B.graph.e… in MatMPIAIJKokkosReduceBegin()
480 Kokkos::deep_copy(Fdi_h, 0); // zero, as we will do 'val++' on them in MatMPIAIJKokkosReduceBegin()
481 Kokkos::deep_copy(Foi_h, 0); in MatMPIAIJKokkosReduceBegin()
603 mm->E_NzLeft = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), E_NzLeft_h); in MatMPIAIJKokkosReduceBegin()
604 mm->Fdjmap = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Fdjmap_h); in MatMPIAIJKokkosReduceBegin()
605 mm->Fdjperm = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Fdjperm_h); in MatMPIAIJKokkosReduceBegin()
[all …]
/petsc/src/mat/tutorials/
H A Dex18k.kokkos.cxx5 using DefaultMemorySpace = Kokkos::DefaultExecutionSpace::memory_space;
9 Kokkos::View<PetscScalar *, DefaultMemorySpace> v("v", 3 * 3 * fe->Ne); in FillMatrixKokkosCOO()
13 Kokkos::parallel_for( in FillMatrixKokkosCOO()

123