[petsc-users] building kokkos matrices on the device

Steven Dargaville dargaville.steven at gmail.com
Tue Feb 25 15:39:03 CST 2025


Hi

I'm just wondering if there is any possibility of making:
MatSetSeqAIJKokkosWithCSRMatrix or MatCreateSeqAIJKokkosWithCSRMatrix in
src/mat/impls/aij/seq/kokkos/aijkok.kokkos.cxx
MatSetMPIAIJKokkosWithSplitSeqAIJKokkosMatrices in
src/mat/impls/aij/mpi/kokkos/mpiaijkok.kokkos.cxx

publicly accessible outside of petsc, or if there is an interface I have
missed for creating Kokkos matrices entirely on the device?
MatCreateSeqAIJKokkosWithCSRMatrix for example is marked as PETSC_INTERN so
I can't link to it.

I've currently just copied the code inside of those methods so that I can
build without any preallocation on the host (e.g., through the COO
interface) and it works really well.

Thanks for your help
Steven
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://lists.mcs.anl.gov/pipermail/petsc-users/attachments/20250225/88849863/attachment.html>


More information about the petsc-users mailing list