Home
last modified time | relevance | path

Searched refs:exec (Results 1 – 13 of 13) sorted by relevance

/petsc/src/sys/objects/device/tests/ !
H A Dex1k.kokkos.cxx38 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in main() local
39 Kokkos::RangePolicy<Kokkos::DefaultExecutionSpace> policy(exec, 0, N); in main()
41 PetscCallCXX(exec.fence()); // Initialize device runtime to get more accurate timing below in main()
46 PetscCallCXX(exec.fence()); in main()
54 PetscCallCXX(exec.fence()); in main()
/petsc/include/petsc/private/ !
H A Dkokkosimpl.hpp130 …KokkosDualViewSyncHost(Kokkos::DualView<Type *> &v_dual, const Kokkos::DefaultExecutionSpace &exec) in KokkosDualViewSyncHost() argument
136 PetscCallCXX(v_dual.sync_host(exec)); in KokkosDualViewSyncHost()
142 PetscCallCXX(exec.fence()); in KokkosDualViewSyncHost()
147 …kkosDualViewSyncDevice(Kokkos::DualView<Type *> &v_dual, const Kokkos::DefaultExecutionSpace &exec) in KokkosDualViewSyncDevice() argument
153 PetscCallCXX(v_dual.sync_device(exec)); in KokkosDualViewSyncDevice()
/petsc/src/vec/is/sf/impls/basic/kokkos/ !
H A Dsfkok.kokkos.cxx238 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in Pack() local
242 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in Pack()
261 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in UnpackAndOp() local
265 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in UnpackAndOp()
280 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FetchAndOp() local
284 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in FetchAndOp()
299 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in ScatterAndOp() local
328 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in ScatterAndOp()
362 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in ScatterAndInsert() local
371 Kokkos::deep_copy(exec, dbuf, sbuf); in ScatterAndInsert()
[all …]
/petsc/src/mat/impls/aij/seq/kokkos/ !
H A Daijkok.hpp121 auto exec = PetscGetKokkosExecutionSpace(); in Mat_SeqAIJKokkos() local
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()
142 if (copyValues) (void)KokkosDualViewSyncDevice(a_dual, exec); in Mat_SeqAIJKokkos()
H A Daijkok.kokkos.cxx143 auto exec = PetscGetKokkosExecutionSpace(); in MatSeqAIJKokkosSyncHost() local
150 PetscCall(KokkosDualViewSyncHost(aijkok->a_dual, exec)); in MatSeqAIJKokkosSyncHost()
1265 auto exec = PetscGetKokkosExecutionSpace(); in MatAXPY_SeqAIJKokkos() local
1294 KokkosBlas::axpy(exec, alpha, Xa, Ya); in MatAXPY_SeqAIJKokkos()
1301 Kokkos::TeamPolicy<>(exec, Y->rmap->n, 1), KOKKOS_LAMBDA(const KokkosTeamMemberType &t) { in MatAXPY_SeqAIJKokkos()
1557 auto exec = PetscGetKokkosExecutionSpace(); in MatSetSeqAIJKokkosWithCSRMatrix() local
1571 PetscCall(KokkosDualViewSyncHost(akok->i_dual, exec)); /* We always need sync'ed i, j on host */ in MatSetSeqAIJKokkosWithCSRMatrix()
1572 PetscCall(KokkosDualViewSyncHost(akok->j_dual, exec)); in MatSetSeqAIJKokkosWithCSRMatrix()
1782 auto exec = PetscGetKokkosExecutionSpace(); in MatSolve_SeqAIJKokkos_Cholesky() local
1805 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(exec, 0, m), KOKKOS_LAMBDA(const PetscInt … in MatSolve_SeqAIJKokkos_Cholesky()
[all …]
/petsc/src/snes/tutorials/ !
H A Dex55k.kokkos.cxx46 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FormFunctionLocalVec() local
61 …"FormFunctionLocalVec", MDRangePolicy<Rank<2, Iterate::Right, Iterate::Right>>(exec, {ys, xs}, {ys… in FormFunctionLocalVec()
123 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FormObjectiveLocalVec() local
140 …"FormObjectiveLocalVec", MDRangePolicy<Rank<2, Iterate::Right, Iterate::Right>>(exec, {ys, xs}, {y… in FormObjectiveLocalVec()
259 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FormJacobianLocalVec() local
264 …"FormJacobianLocalVec", MDRangePolicy<Rank<2, Iterate::Right, Iterate::Right>>(exec, {ys, xs}, {ys… in FormJacobianLocalVec()
/petsc/src/vec/vec/impls/seq/kokkos/ !
H A Dveckok.kokkos.cxx122 auto exec = PetscGetKokkosExecutionSpace(); in VecAbs_SeqKokkos() local
127 PetscCallCXX(KokkosBlas::abs(exec, xv, xv)); in VecAbs_SeqKokkos()
348 auto exec = PetscGetKokkosExecutionSpace(); in VecMultiDot_Private() local
356 …PetscCallCXX(Kokkos::parallel_reduce(Kokkos::RangePolicy<WorkTag>(exec, 0, N), mdot, Kokkos::subvi… in VecMultiDot_Private()
364 Kokkos::RangePolicy<WorkTag> policy(exec, 0, N); in VecMultiDot_Private()
382 …exec.fence(); /* If reduce is async, then we need this fence to make sure z is ready for use on ho… in VecMultiDot_Private()
619 auto exec = PetscGetKokkosExecutionSpace(); in VecSet_SeqKokkos() local
624 PetscCallCXX(KokkosBlas::fill(exec, xv, alpha)); in VecSet_SeqKokkos()
633 auto exec = PetscGetKokkosExecutionSpace(); in VecScale_SeqKokkos() local
643 PetscCallCXX(KokkosBlas::scal(exec, xv, alpha, xv)); in VecScale_SeqKokkos()
[all …]
/petsc/src/mat/impls/aij/mpi/kokkos/ !
H A Dmpiaijkok.kokkos.cxx1516 auto exec = PetscGetKokkosExecutionSpace(); in MatCOOStruct_MPIAIJKokkos() local
1524 …Ajmap1 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Ajmap1, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1525 …Aperm1 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Aperm1, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1526 …Bjmap1 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Bjmap1, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1527 …Bperm1 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Bperm1, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1528 …Aimap2 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Aimap2, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1529 …Ajmap2 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Ajmap2, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1530 …Aperm2 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Aperm2, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1531 …Bimap2 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Bimap2, coo_h-… in MatCOOStruct_MPIAIJKokkos()
1532 …Bjmap2 = Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Bjmap2, coo_h-… in MatCOOStruct_MPIAIJKokkos()
[all …]
/petsc/doc/changes/ !
H A D324.md25 - Add `--with-python-exec-from-env=python` to allow providing the Python to be used by mpi4py and p…
H A D319.md19 - Add `--with-coverage-exec` configure option to specify the coverage-collection tool to be used e.…
H A D318.md134 exec-name`.
/petsc/src/binding/petsc4py/src/petsc4py/PETSc/ !
H A Dlibpetsc4py.pyx161 code = compile(source.read(), path, 'exec')
162 exec(code, module.__dict__)
/petsc/doc/ !
H A Dpetsc.bib16332 …url = {http://www.amazon.ca/exec/obidos/redirect?tag=citeulike04-20{\&}path=ASIN/0201633…
16378 …url = {http://www.amazon.ca/exec/obidos/redirect?tag=citeulike04-20{\&}path=ASIN/0691005…