Lines Matching refs:MemorySpace

26 template <class MemorySpace>
27 …osOffsetView_Private(DM da, Vec vec, PetscScalarKokkosOffsetView1DType<MemorySpace> *ov, PetscBool… in DMDAVecGetKokkosOffsetView_Private()
30 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView_Private()
41 *ov = PetscScalarKokkosOffsetView1DType<MemorySpace>(kv.data(), {gxs * dof}, {(gxs + gxm) * dof}); in DMDAVecGetKokkosOffsetView_Private()
45 template <class MemorySpace>
46 …osOffsetView_Private(DM da, Vec vec, PetscScalarKokkosOffsetView1DType<MemorySpace> *ov, PetscBool… in DMDAVecRestoreKokkosOffsetView_Private()
48 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView_Private()
60 template <class MemorySpace>
61 …DMDAVecGetKokkosOffsetView(DM da, Vec vec, ConstPetscScalarKokkosOffsetView1DType<MemorySpace> *ov) in DMDAVecGetKokkosOffsetView()
64 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView()
73 …*ov = ConstPetscScalarKokkosOffsetView1DType<MemorySpace>(kv.data(), {gxs * dof}, {(gxs + gxm) * d… in DMDAVecGetKokkosOffsetView()
77 template <class MemorySpace>
78 …VecRestoreKokkosOffsetView(DM da, Vec vec, ConstPetscScalarKokkosOffsetView1DType<MemorySpace> *ov) in DMDAVecRestoreKokkosOffsetView()
80 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView()
92 template <class MemorySpace>
93 …osOffsetView_Private(DM da, Vec vec, PetscScalarKokkosOffsetView2DType<MemorySpace> *ov, PetscBool… in DMDAVecGetKokkosOffsetView_Private()
96 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView_Private()
106 …*ov = PetscScalarKokkosOffsetView2DType<MemorySpace>(kv.data(), {gys * dof, gxs * dof}, {(gys + gy… in DMDAVecGetKokkosOffsetView_Private()
110 template <class MemorySpace>
111 …osOffsetView_Private(DM da, Vec vec, PetscScalarKokkosOffsetView2DType<MemorySpace> *ov, PetscBool… in DMDAVecRestoreKokkosOffsetView_Private()
113 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView_Private()
120 kv = PetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1)); in DMDAVecRestoreKokkosOffsetView_Private()
126 template <class MemorySpace>
127 …DMDAVecGetKokkosOffsetView(DM da, Vec vec, ConstPetscScalarKokkosOffsetView2DType<MemorySpace> *ov) in DMDAVecGetKokkosOffsetView()
130 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView()
139 …*ov = ConstPetscScalarKokkosOffsetView2DType<MemorySpace>(kv.data(), {gys * dof, gxs * dof}, {(gys… in DMDAVecGetKokkosOffsetView()
143 template <class MemorySpace>
144 …VecRestoreKokkosOffsetView(DM da, Vec vec, ConstPetscScalarKokkosOffsetView2DType<MemorySpace> *ov) in DMDAVecRestoreKokkosOffsetView()
146 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView()
152 kv = ConstPetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1)); in DMDAVecRestoreKokkosOffsetView()
158 template <class MemorySpace>
159 …osOffsetView_Private(DM da, Vec vec, PetscScalarKokkosOffsetView3DType<MemorySpace> *ov, PetscBool… in DMDAVecGetKokkosOffsetView_Private()
162 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView_Private()
172 …*ov = PetscScalarKokkosOffsetView3DType<MemorySpace>(kv.data(), {gzs * dof, gys * dof, gxs * dof},… in DMDAVecGetKokkosOffsetView_Private()
176 template <class MemorySpace>
177 …osOffsetView_Private(DM da, Vec vec, PetscScalarKokkosOffsetView3DType<MemorySpace> *ov, PetscBool… in DMDAVecRestoreKokkosOffsetView_Private()
179 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView_Private()
185 …kv = PetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1) * ov->extent… in DMDAVecRestoreKokkosOffsetView_Private()
191 template <class MemorySpace>
192 …DMDAVecGetKokkosOffsetView(DM da, Vec vec, ConstPetscScalarKokkosOffsetView3DType<MemorySpace> *ov) in DMDAVecGetKokkosOffsetView()
195 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetView()
204 …*ov = ConstPetscScalarKokkosOffsetView3DType<MemorySpace>(kv.data(), {gzs * dof, gys * dof, gxs * … in DMDAVecGetKokkosOffsetView()
208 template <class MemorySpace>
209 …VecRestoreKokkosOffsetView(DM da, Vec vec, ConstPetscScalarKokkosOffsetView3DType<MemorySpace> *ov) in DMDAVecRestoreKokkosOffsetView()
211 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetView()
217 …kv = ConstPetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1) * ov->e… in DMDAVecRestoreKokkosOffsetView()
364 template <class MemorySpace>
365 …ffsetViewDOF_Private(DM da, Vec vec, PetscScalarKokkosOffsetView2DType<MemorySpace> *ov, PetscBool… in DMDAVecGetKokkosOffsetViewDOF_Private()
368 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetViewDOF_Private()
378 *ov = PetscScalarKokkosOffsetView2DType<MemorySpace>(kv.data(), {gxs, 0}, {gxs + gxm, dof}); in DMDAVecGetKokkosOffsetViewDOF_Private()
382 template <class MemorySpace>
383 …ffsetViewDOF_Private(DM da, Vec vec, PetscScalarKokkosOffsetView2DType<MemorySpace> *ov, PetscBool… in DMDAVecRestoreKokkosOffsetViewDOF_Private()
385 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetViewDOF_Private()
391 kv = PetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1)); in DMDAVecRestoreKokkosOffsetViewDOF_Private()
397 template <class MemorySpace>
398 …AVecGetKokkosOffsetViewDOF(DM da, Vec vec, ConstPetscScalarKokkosOffsetView2DType<MemorySpace> *ov) in DMDAVecGetKokkosOffsetViewDOF()
401 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetViewDOF()
410 *ov = ConstPetscScalarKokkosOffsetView2DType<MemorySpace>(kv.data(), {gxs, 0}, {gxs + gxm, dof}); in DMDAVecGetKokkosOffsetViewDOF()
414 template <class MemorySpace>
415 …RestoreKokkosOffsetViewDOF(DM da, Vec vec, ConstPetscScalarKokkosOffsetView2DType<MemorySpace> *ov) in DMDAVecRestoreKokkosOffsetViewDOF()
417 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetViewDOF()
423 kv = ConstPetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1)); in DMDAVecRestoreKokkosOffsetViewDOF()
429 template <class MemorySpace>
430 …ffsetViewDOF_Private(DM da, Vec vec, PetscScalarKokkosOffsetView3DType<MemorySpace> *ov, PetscBool… in DMDAVecGetKokkosOffsetViewDOF_Private()
433 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetViewDOF_Private()
443 …*ov = PetscScalarKokkosOffsetView3DType<MemorySpace>(kv.data(), {gys, gxs, 0}, {gys + gym, gxs + g… in DMDAVecGetKokkosOffsetViewDOF_Private()
447 template <class MemorySpace>
448 …ffsetViewDOF_Private(DM da, Vec vec, PetscScalarKokkosOffsetView3DType<MemorySpace> *ov, PetscBool… in DMDAVecRestoreKokkosOffsetViewDOF_Private()
450 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetViewDOF_Private()
456 …kv = PetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1) * ov->extent… in DMDAVecRestoreKokkosOffsetViewDOF_Private()
462 template <class MemorySpace>
463 …AVecGetKokkosOffsetViewDOF(DM da, Vec vec, ConstPetscScalarKokkosOffsetView3DType<MemorySpace> *ov) in DMDAVecGetKokkosOffsetViewDOF()
466 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetViewDOF()
475 …*ov = ConstPetscScalarKokkosOffsetView3DType<MemorySpace>(kv.data(), {gys, gxs, 0}, {gys + gym, gx… in DMDAVecGetKokkosOffsetViewDOF()
479 template <class MemorySpace>
480 …RestoreKokkosOffsetViewDOF(DM da, Vec vec, ConstPetscScalarKokkosOffsetView3DType<MemorySpace> *ov) in DMDAVecRestoreKokkosOffsetViewDOF()
482 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetViewDOF()
488 …kv = ConstPetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1) * ov->e… in DMDAVecRestoreKokkosOffsetViewDOF()
494 template <class MemorySpace>
495 …ffsetViewDOF_Private(DM da, Vec vec, PetscScalarKokkosOffsetView4DType<MemorySpace> *ov, PetscBool… in DMDAVecGetKokkosOffsetViewDOF_Private()
498 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetViewDOF_Private()
508 …*ov = PetscScalarKokkosOffsetView4DType<MemorySpace>(kv.data(), {gzs, gys, gxs, 0}, {gzs + gzm, gy… in DMDAVecGetKokkosOffsetViewDOF_Private()
512 template <class MemorySpace>
513 …ffsetViewDOF_Private(DM da, Vec vec, PetscScalarKokkosOffsetView4DType<MemorySpace> *ov, PetscBool… in DMDAVecRestoreKokkosOffsetViewDOF_Private()
515 PetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetViewDOF_Private()
521 …kv = PetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1) * ov->extent… in DMDAVecRestoreKokkosOffsetViewDOF_Private()
527 template <class MemorySpace>
528 …AVecGetKokkosOffsetViewDOF(DM da, Vec vec, ConstPetscScalarKokkosOffsetView4DType<MemorySpace> *ov) in DMDAVecGetKokkosOffsetViewDOF()
531 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecGetKokkosOffsetViewDOF()
540 …*ov = ConstPetscScalarKokkosOffsetView4DType<MemorySpace>(kv.data(), {gzs, gys, gxs, 0}, {gzs + gz… in DMDAVecGetKokkosOffsetViewDOF()
544 template <class MemorySpace>
545 …RestoreKokkosOffsetViewDOF(DM da, Vec vec, ConstPetscScalarKokkosOffsetView4DType<MemorySpace> *ov) in DMDAVecRestoreKokkosOffsetViewDOF()
547 ConstPetscScalarKokkosViewType<MemorySpace> kv; in DMDAVecRestoreKokkosOffsetViewDOF()
553 …kv = ConstPetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1) * ov->e… in DMDAVecRestoreKokkosOffsetViewDOF()