Home
last modified time | relevance | path

Searched refs:PETSC_MEMTYPE_KOKKOS (Results 1 – 9 of 9) sorted by relevance

/petsc/include/
H A Dpetscdevicetypes.h54 #define PETSC_MEMTYPE_KOKKOS PETSC_MEMTYPE_CUDA macro
56 #define PETSC_MEMTYPE_KOKKOS PETSC_MEMTYPE_HIP macro
58 #define PETSC_MEMTYPE_KOKKOS PETSC_MEMTYPE_SYCL macro
60 #define PETSC_MEMTYPE_KOKKOS PETSC_MEMTYPE_HOST macro
207 return PETSC_MEMTYPE_KOKKOS; in PetscOffloadMaskToMemType()
/petsc/src/mat/tests/
H A Dex217.c32 …else if (iskokkos) PetscCheck(memtype == PETSC_MEMTYPE_KOKKOS, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "… in main()
/petsc/src/vec/is/sf/tests/
H A Dex4k.kokkos.cxx69 else if (isKokkos) mtype = PETSC_MEMTYPE_KOKKOS; in main()
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx178 …eWithMemTypeBegin(vecmpi->coo_sf, MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, sendbuf.data(), PETSC_MEMTYPE… in VecSetValuesCOO_MPIKokkos()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx669 …FReduceWithMemTypeBegin(reduceSF, MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, leafBuf.data(), PETSC_MEMTYPE… in MatMPIAIJKokkosReduceBegin()
996 …cSFBcastWithMemTypeBegin(bcastSF, MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, rootBuf.data(), PETSC_MEMTYPE… in MatMPIAIJKokkosBcastBegin()
1631 …cSFReduceWithMemTypeBegin(coo->sf, MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, vsend.data(), PETSC_MEMTYPE_… in MatSetValuesCOO_MPIAIJKokkos()
/petsc/src/binding/petsc4py/src/lib-petsc/
H A Dcustom.h186 else if (iskok) *m = PETSC_MEMTYPE_KOKKOS; in VecGetCurrentMemType()
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckok.kokkos.cxx1426 …if (mtype) *mtype = PETSC_MEMTYPE_KOKKOS; // Could be PETSC_MEMTYPE_HOST when Kokkos was not confi… in VecGetArrayAndMemType_SeqKokkos()
1435 if (PetscMemTypeHost(PETSC_MEMTYPE_KOKKOS)) { in VecRestoreArrayAndMemType_SeqKokkos()
1452 …if (mtype) *mtype = PETSC_MEMTYPE_KOKKOS; // Could be PETSC_MEMTYPE_HOST when Kokkos was not confi… in VecGetArrayWriteAndMemType_SeqKokkos()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.kokkos.cxx241 if (mtype) *mtype = PETSC_MEMTYPE_KOKKOS; in MatSeqAIJGetCSRAndMemType_SeqAIJKokkos()
248 *mtype = PETSC_MEMTYPE_KOKKOS; in MatGetCurrentMemType_SeqAIJKokkos()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.c7393 else if (iskokk) mmdata->mtype = PETSC_MEMTYPE_KOKKOS; in MatProductSymbolic_MPIAIJBACKEND()