Home
last modified time | relevance | path

Searched refs:PetscScalarPoolSize (Results 1 – 3 of 3) sorted by relevance

/petsc/src/sys/objects/kokkos/
H A Dkinit.kokkos.cxx9 PetscInt PetscScalarPoolSize = 0; variable
19 PetscScalarPoolSize = 0; in PetscKokkosFinalize_Private()
101 if (!PetscScalarPoolSize) { // A pool for a small count of PetscScalars in PetscKokkosInitializeCheck()
102 PetscScalarPoolSize = 1024; in PetscKokkosInitializeCheck()
103 …ol = static_cast<PetscScalar *>(Kokkos::kokkos_malloc(sizeof(PetscScalar) * PetscScalarPoolSize))); in PetscKokkosInitializeCheck()
/petsc/include/petsc/private/
H A Dkokkosimpl.hpp12 PETSC_SINGLE_LIBRARY_INTERN PetscInt PetscScalarPoolSize; variable
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckok.kokkos.cxx569 if (nv > PetscScalarPoolSize) { // rare case in VecMultiDot_SeqKokkos_GEMV()
570 PetscScalarPoolSize = nv; in VecMultiDot_SeqKokkos_GEMV()
571 …arPool = static_cast<PetscScalar *>(Kokkos::kokkos_realloc(PetscScalarPool, PetscScalarPoolSize))); in VecMultiDot_SeqKokkos_GEMV()
875 if (nv > PetscScalarPoolSize) { // rare case in VecMAXPY_SeqKokkos_GEMV()
876 PetscScalarPoolSize = nv; in VecMAXPY_SeqKokkos_GEMV()
877 …arPool = static_cast<PetscScalar *>(Kokkos::kokkos_realloc(PetscScalarPool, PetscScalarPoolSize))); in VecMAXPY_SeqKokkos_GEMV()