Lines Matching refs:Kokkos

13 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()
60   PetscCallCXX(Kokkos::parallel_for(  in FormFunctionLocalVec()
123   Kokkos::DefaultExecutionSpace      exec = PetscGetKokkosExecutionSpace();  in FormObjectiveLocalVec()
139   PetscCallCXX(Kokkos::parallel_reduce(  in FormObjectiveLocalVec()
259   Kokkos::DefaultExecutionSpace      exec = PetscGetKokkosExecutionSpace();  in FormJacobianLocalVec()
263   PetscCallCXX(Kokkos::parallel_for(  in FormJacobianLocalVec()