xref: /petsc/include/petsc_kokkos.hpp (revision 524fe776c8aa733ff2ef43b738fa4e354b69f6ec)
1*524fe776SJunchao Zhang #ifndef PETSC_KOKKOS_HPP
2*524fe776SJunchao Zhang #define PETSC_KOKKOS_HPP
3*524fe776SJunchao Zhang 
4*524fe776SJunchao Zhang #include <Kokkos_Core.hpp>
5*524fe776SJunchao Zhang 
6*524fe776SJunchao Zhang /* SUBMANSEC = Sys */
7*524fe776SJunchao Zhang 
8*524fe776SJunchao Zhang extern Kokkos::DefaultExecutionSpace *PetscKokkosExecutionSpacePtr;
9*524fe776SJunchao Zhang 
10*524fe776SJunchao Zhang /*MC
11*524fe776SJunchao Zhang   PetscGetKokkosExecutionSpace - Return the Kokkos execution space that petsc is using
12*524fe776SJunchao Zhang 
13*524fe776SJunchao Zhang   Level: beginner
14*524fe776SJunchao Zhang 
15*524fe776SJunchao Zhang M*/
16*524fe776SJunchao Zhang inline Kokkos::DefaultExecutionSpace &PetscGetKokkosExecutionSpace(void)
17*524fe776SJunchao Zhang {
18*524fe776SJunchao Zhang   return *PetscKokkosExecutionSpacePtr;
19*524fe776SJunchao Zhang }
20*524fe776SJunchao Zhang 
21*524fe776SJunchao Zhang #endif
22