[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