mpi/kokkos/mpiaijkok.kokkos.cxx

1052   Mat_MPIAIJ     *ampi = static_cast<Mat_MPIAIJ *>(A->data);
1060   PetscCall(MatSeqAIJKokkosGenerateTranspose_Private(ampi->A, &Adt));
1061   PetscCall(MatSeqAIJKokkosGenerateTranspose_Private(ampi->B, &Aot));
1062   PetscCall(MatSeqAIJKokkosGetKokkosCsrMatrix(ampi->A, &Ad));
1063   PetscCall(MatSeqAIJKokkosGetKokkosCsrMatrix(ampi->B, &Ao));
1098   PetscCall(MatMPIAIJKokkosReduceBegin(comm, mm->C3, mm->C4, cstart, cend, bmpi->garray, ampi->Mvctx, MAT_INITIAL_MATRIX, map_h.data(), mm));
1110   PetscCall(MatMPIAIJKokkosReduceEnd(comm, mm->C3, mm->C4, cstart, cend, bmpi->garray, ampi->Mvctx, MAT_INITIAL_MATRIX, map_h.data(), mm));
1130   Mat_MPIAIJ     *ampi = static_cast<Mat_MPIAIJ *>(A->data);
1137   PetscCall(MatSeqAIJKokkosGenerateTranspose_Private(ampi->A, &Adt));
1138   PetscCall(MatSeqAIJKokkosGenerateTranspose_Private(ampi->B, &Aot));
1171   Mat_MPIAIJ     *ampi = static_cast<Mat_MPIAIJ *>(A->data);
1176   PetscCall(MatSeqAIJKokkosGetKokkosCsrMatrix(ampi->A, &Ad));
1177   PetscCall(MatSeqAIJKokkosGetKokkosCsrMatrix(ampi->B, &Ao));
1198   PetscCall(MatMPIAIJKokkosBcastBegin(B, ampi->Mvctx, MAT_INITIAL_MATRIX, map_h.data(), mm));
1212   PetscCall(MatMPIAIJKokkosBcastEnd(B, ampi->Mvctx, MAT_INITIAL_MATRIX, map_h.data(), mm));
1242   Mat_MPIAIJ     *ampi = static_cast<Mat_MPIAIJ *>(A->data);
1247   PetscCall(MatSeqAIJKokkosGetKokkosCsrMatrix(ampi->A, &Ad));
1248   PetscCall(MatSeqAIJKokkosGetKokkosCsrMatrix(ampi->B, &Ao));