Home
last modified time | relevance | path

Searched refs:PetscGetKokkosExecutionSpace (Results 1 – 12 of 12) sorted by relevance

/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckok.kokkos.cxx122 auto exec = PetscGetKokkosExecutionSpace(); in VecAbs_SeqKokkos()
142 …"VecReciprocal", Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, xin->map->n), KOKKOS_LAM… in VecReciprocal_SeqKokkos()
159 "VecMin", Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, xin->map->n), in VecMin_SeqKokkos()
183 "VecMax", Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, xin->map->n), in VecMax_SeqKokkos()
205 PetscCallCXX(*sum = KokkosBlas::sum(PetscGetKokkosExecutionSpace(), xv)); in VecSum_SeqKokkos()
218 …PetscCallCXX(Kokkos::parallel_for("VecShift", Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace()… in VecShift_SeqKokkos()
243 PetscCallCXX(KokkosBlas::axpy(PetscGetKokkosExecutionSpace(), alpha, xv, yv)); in VecAXPY_SeqKokkos()
274 …lCXX(Kokkos::parallel_reduce("VecTDot", Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, x… in VecTDot_SeqKokkos()
348 auto exec = PetscGetKokkosExecutionSpace(); in VecMultiDot_Private()
392 Kokkos::RangePolicy<> policy(PetscGetKokkosExecutionSpace(), 0, N); in VecMultiDot_Verbose()
[all …]
/petsc/src/ksp/pc/impls/pbjacobi/kokkos/
H A Dpbjacobi_kok.kokkos.cxx13 … diag_d = Kokkos::create_mirror_view_and_copy(PetscGetKokkosExecutionSpace(), diag… in PC_PBJacobi_Kokkos()
22 PetscCall(KokkosDualViewSyncDevice(diag_dual, PetscGetKokkosExecutionSpace())); in Update()
48 …label, Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, bs * mbs), KOKKOS_LAMBDA(PetscInt … in PCApplyOrTranspose_PBJacobi_Kokkos()
/petsc/src/ksp/pc/impls/vpbjacobi/kokkos/
H A Dvpbjacobi_kok.kokkos.cxx63 PetscCall(KokkosDualViewSyncDevice(bs_dual, PetscGetKokkosExecutionSpace())); in BuildHelperArrays()
64 PetscCall(KokkosDualViewSyncDevice(bs2_dual, PetscGetKokkosExecutionSpace())); in BuildHelperArrays()
65 PetscCall(KokkosDualViewSyncDevice(blkMap_dual, PetscGetKokkosExecutionSpace())); in BuildHelperArrays()
91 …label, Kokkos::TeamPolicy<>(PetscGetKokkosExecutionSpace(), jac->nblocks, Kokkos::AUTO()), KOKKOS_… in PCApplyOrTranspose_VPBJacobi_Kokkos()
108 …label, Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, pckok->n), KOKKOS_LAMBDA(PetscInt … in PCApplyOrTranspose_VPBJacobi_Kokkos()
/petsc/include/
H A Dpetsc_kokkos.hpp15 inline Kokkos::DefaultExecutionSpace PetscGetKokkosExecutionSpace(void) in PetscGetKokkosExecutionSpace() function
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.kokkos.cxx118 PetscCall(KokkosDualViewSyncDevice(aijkok->a_dual, PetscGetKokkosExecutionSpace())); in MatSeqAIJKokkosSyncDevice()
143 auto exec = PetscGetKokkosExecutionSpace(); in MatSeqAIJKokkosSyncHost()
165 PetscCall(KokkosDualViewSyncHost(aijkok->a_dual, PetscGetKokkosExecutionSpace())); in MatSeqAIJGetArray_SeqAIJKokkos()
188 PetscCall(KokkosDualViewSyncHost(aijkok->a_dual, PetscGetKokkosExecutionSpace())); in MatSeqAIJGetArrayRead_SeqAIJKokkos()
238 PetscCall(KokkosDualViewSyncDevice(aijkok->a_dual, PetscGetKokkosExecutionSpace())); in MatSeqAIJGetCSRAndMemType_SeqAIJKokkos()
319 …PetscCall(KokkosDualViewSyncDevice(akok->a_dual, PetscGetKokkosExecutionSpace())); // Sync A's val… in MatSeqAIJKokkosGenerateTranspose_Private()
332 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KO… in MatSeqAIJKokkosGenerateTranspose_Private()
339 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KO… in MatSeqAIJKokkosGenerateTranspose_Private()
357 …PetscCall(KokkosDualViewSyncDevice(akok->a_dual, PetscGetKokkosExecutionSpace())); // Sync A's val… in MatSeqAIJKokkosGenerateHermitian_Private()
370 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KO… in MatSeqAIJKokkosGenerateHermitian_Private()
[all …]
H A Daijkok.hpp121 auto exec = PetscGetKokkosExecutionSpace(); in Mat_SeqAIJKokkos()
/petsc/src/sys/objects/device/tests/
H A Dex1k.kokkos.cxx38 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in main()
/petsc/src/vec/is/sf/impls/basic/kokkos/
H A Dsfkok.kokkos.cxx238 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in Pack()
261 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in UnpackAndOp()
280 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FetchAndOp()
299 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in ScatterAndOp()
362 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in ScatterAndInsert()
387 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FetchAndOpLocal()
568 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in PetscSFLinkSyncStream_Kokkos()
577 DeviceExecutionSpace exec = PetscGetKokkosExecutionSpace(); in PetscSFLinkMemcpy_Kokkos()
/petsc/src/snes/tutorials/
H A Dex55k.kokkos.cxx46 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FormFunctionLocalVec()
123 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FormObjectiveLocalVec()
259 Kokkos::DefaultExecutionSpace exec = PetscGetKokkosExecutionSpace(); in FormJacobianLocalVec()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx647 …Kokkos::TeamPolicy<>(PetscGetKokkosExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAM… in MatMPIAIJKokkosReduceBegin()
693 …Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, Fdnz), KOKKOS_LAMBDA(const MatRowMapType … in MatMPIAIJKokkosReduceEnd()
700 …Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, Fonz), KOKKOS_LAMBDA(const MatRowMapType … in MatMPIAIJKokkosReduceEnd()
950 PetscCall(KokkosDualViewSyncDevice(akok->a_dual, PetscGetKokkosExecutionSpace())); in MatMPIAIJKokkosBcastBegin()
951 PetscCall(KokkosDualViewSyncDevice(bkok->a_dual, PetscGetKokkosExecutionSpace())); in MatMPIAIJKokkosBcastBegin()
974 …Kokkos::TeamPolicy<>(PetscGetKokkosExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAM… in MatMPIAIJKokkosBcastBegin()
1025 …Kokkos::TeamPolicy<>(PetscGetKokkosExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAM… in MatMPIAIJKokkosBcastEnd()
1115 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, oldj.ex… in MatProductSymbolic_MPIAIJKokkos_AtB()
1227 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, oldj.ex… in MatProductSymbolic_MPIAIJKokkos_AB()
1516 auto exec = PetscGetKokkosExecutionSpace(); in MatCOOStruct_MPIAIJKokkos()
[all …]
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx177 …Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, vecmpi->sendlen), KO… in VecSetValuesCOO_MPIKokkos()
184 Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, m), KOKKOS_LAMBDA(const PetscCount i) { in VecSetValuesCOO_MPIKokkos()
194 …Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, vecmpi->nnz2), KOKKOS_LAMBDA(PetscCount i… in VecSetValuesCOO_MPIKokkos()
/petsc/doc/changes/
H A D323.md18 - Change `PetscGetKokkosExecutionSpace()` to return the `Kokkos::DefaultExecutionSpace` object inst…