| /petsc/src/sys/objects/kokkos/ |
| H A D | kinit.kokkos.cxx | 11 Kokkos::DefaultExecutionSpace *PetscKokkosExecutionSpacePtr = nullptr;
85 …ICE_CUDA) PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExecutionSpace(PetscDefau… in PetscKokkosInitializeCheck()
87 …VICE_HIP) PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExecutionSpace(PetscDefau… in PetscKokkosInitializeCheck()
92 …PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExecutionSpace(*(sycl::queue *)hand… in PetscKokkosInitializeCheck()
97 PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExecutionSpace()); in PetscKokkosInitializeCheck()
|
| /petsc/include/ |
| H A D | petsc_kokkos.hpp | 7 extern Kokkos::DefaultExecutionSpace *PetscKokkosExecutionSpacePtr;
15 inline Kokkos::DefaultExecutionSpace PetscGetKokkosExecutionSpace(void) in PetscGetKokkosExecutionSpace()
|
| /petsc/include/petsc/private/ |
| H A D | kokkosimpl.hpp | 14 using DefaultExecutionSpace = Kokkos::DefaultExecutionSpace; typedef
15 using DefaultMemorySpace = Kokkos::DefaultExecutionSpace::memory_space;
130 …KokkosDualViewSyncHost(Kokkos::DualView<Type *> &v_dual, const Kokkos::DefaultExecutionSpace &exec) in KokkosDualViewSyncHost()
147 …kkosDualViewSyncDevice(Kokkos::DualView<Type *> &v_dual, const Kokkos::DefaultExecutionSpace &exec) in KokkosDualViewSyncDevice()
|
| H A D | pcbjkokkosimpl.h | 32 typedef Kokkos::DefaultExecutionSpace exec_space;
|
| /petsc/src/sys/objects/device/tests/ |
| H A D | ex1k.kokkos.cxx | 38 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in main()
39 Kokkos::RangePolicy<Kokkos::DefaultExecutionSpace> policy(exec, 0, N); in main()
|
| /petsc/src/dm/tests/ |
| H A D | ex10k.kokkos.cxx | 10 …rimental::OffsetView<PetscScalar **, Kokkos::LayoutRight, Kokkos::DefaultExecutionSpace::memory_sp…
11 …al::OffsetView<const PetscScalar **, Kokkos::LayoutRight, Kokkos::DefaultExecutionSpace::memory_sp…
13 …imental::OffsetView<PetscScalar ***, Kokkos::LayoutRight, Kokkos::DefaultExecutionSpace::memory_sp…
14 …l::OffsetView<const PetscScalar ***, Kokkos::LayoutRight, Kokkos::DefaultExecutionSpace::memory_sp…
24 …s::Experimental::OffsetView<Node **, Kokkos::LayoutRight, Kokkos::DefaultExecutionSpace::memory_sp…
25 …erimental::OffsetView<const Node **, Kokkos::LayoutRight, Kokkos::DefaultExecutionSpace::memory_sp…
|
| /petsc/src/mat/tutorials/ |
| H A D | ex18k.kokkos.cxx | 5 using DefaultMemorySpace = Kokkos::DefaultExecutionSpace::memory_space;
|
| /petsc/src/snes/tutorials/ |
| H A D | ex55k.kokkos.cxx | 13 using DefaultMemorySpace = Kokkos::DefaultExecutionSpace::memory_space;
46 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FormFunctionLocalVec()
123 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FormObjectiveLocalVec()
259 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FormJacobianLocalVec()
|
| H A D | ex3k.kokkos.cxx | 84 using DefaultExecutionSpace = Kokkos::DefaultExecutionSpace; typedef
85 using DefaultMemorySpace = Kokkos::DefaultExecutionSpace::memory_space;
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.hpp | 53 …:KokkosKernelsHandle<MatRowMapType, MatColIdxType, MatScalarType, DefaultExecutionSpace, DefaultMe…
55 using KokkosTeamMemberType = Kokkos::TeamPolicy<DefaultExecutionSpace>::member_type;
|
| H A D | aijkok.kokkos.cxx | 1727 …w, MatColIdxKokkosView, MatScalarKokkosView, MatRowMapKokkosView, DefaultExecutionSpace>(n, n, fac… in MatSeqAIJKokkosTransposeSolveCheck()
1731 …PetscCallCXX(sort_crs_matrix<DefaultExecutionSpace, MatRowMapKokkosView, MatColIdxKokkosView, MatS… in MatSeqAIJKokkosTransposeSolveCheck()
1764 …w, MatColIdxKokkosView, MatScalarKokkosView, MatRowMapKokkosView, DefaultExecutionSpace>(n, n, fac… in MatSeqAIJKokkosTransposeSolveCheck()
1768 …PetscCallCXX(sort_crs_matrix<DefaultExecutionSpace, MatRowMapKokkosView, MatColIdxKokkosView, MatS… in MatSeqAIJKokkosTransposeSolveCheck()
|
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkos.kokkos.cxx | 538 …const PetscInt conc = Kokkos::DefaultExecutionSpace().concurrency(), openmp = !!(conc < 1000),… in PCApply_BJKOKKOS()
598 using scr_mem_t = Kokkos::DefaultExecutionSpace::scratch_memory_space; in PCApply_BJKOKKOS()
600 …Kokkos::View<Batch_MetaData *, Kokkos::DefaultExecutionSpace> d_metadata("solver meta data", nBlk); in PCApply_BJKOKKOS()
626 …Kokkos::View<PetscScalar *, Kokkos::DefaultExecutionSpace> d_work_vecs_k("workvectors", global_buf… in PCApply_BJKOKKOS()
966 …ew<PetscInt *, Kokkos::LayoutRight>(Kokkos::create_mirror(Kokkos::DefaultExecutionSpace::memory_sp… in PCSetUp_BJKOKKOS()
974 …const PetscInt conc = Kokkos::DefaultExecutionSpace().concurrency(), openmp = !!(conc < 1000),… in PCSetUp_BJKOKKOS()
|
| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 116 …= new Kokkos::View<P4estVertexMaps>(Kokkos::create_mirror(Kokkos::DefaultExecutionSpace::memory_sp… in LandauKokkosCreateMatMaps()
268 … = new Kokkos::View<PetscScalar *, Kokkos::LayoutRight, Kokkos::DefaultExecutionSpace>("coo_vals… in LandauKokkosStaticDataSet()
334 …tic_cast<Kokkos::View<PetscScalar *, Kokkos::LayoutRight, Kokkos::DefaultExecutionSpace> *>((void … in LandauKokkosStaticDataClear()
383 using scr_mem_t = Kokkos::DefaultExecutionSpace::scratch_memory_space; in LandauKokkosJacobian()
394 …const int conc = Kokkos::DefaultExecutionSpace().concurrency(), openmp = !!(conc < 1000), t… in LandauKokkosJacobian()
|
| /petsc/src/dm/impls/plex/kokkos/ |
| H A D | plexlocalizationletkf.kokkos.cxx | 124 using ExecSpace = Kokkos::DefaultExecutionSpace; in DMPlexGetLETKFLocalizationMatrix()
|
| /petsc/doc/changes/ |
| H A D | 323.md | 18 - Change `PetscGetKokkosExecutionSpace()` to return the `Kokkos::DefaultExecutionSpace` object inst…
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 624 …PetscCall(MatMergeGetLaunchParameters<DefaultExecutionSpace>(Em, Enz, -1, teamSize, vectorLength, … in MatMPIAIJKokkosReduceBegin()
937 …PetscCall(MatMergeGetLaunchParameters<DefaultExecutionSpace>(mm->irootloc.extent(0), mm->rootBuf.e… in MatMPIAIJKokkosBcastBegin()
943 …PetscCall(MatMergeGetLaunchParameters<DefaultExecutionSpace>(Fm, Fnz, -1, teamSize, vectorLength, … in MatMPIAIJKokkosBcastBegin()
1586 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in MatSetValuesCOO_MPIAIJKokkos()
|
| /petsc/src/vec/is/sf/impls/basic/kokkos/ |
| H A D | sfkok.kokkos.cxx | 6 using DeviceExecutionSpace = Kokkos::DefaultExecutionSpace;
|