seq/kokkos/aijkok.kokkos.cxx

66   Mat_SeqAIJKokkos *aijkok;
73   aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
96     aijkok   = new Mat_SeqAIJKokkos(A, A->rmap->n, A->cmap->n, aijseq, A->nonzerostate, PETSC_FALSE /* don't copy mat values to device */);
112   Mat_SeqAIJKokkos *aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
116   PetscCheck(aijkok, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "Unexpected NULL (Mat_SeqAIJKokkos*)A->spptr");
128   Mat_SeqAIJKokkos *aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
142   Mat_SeqAIJKokkos *aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
156   Mat_SeqAIJKokkos *aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
175   Mat_SeqAIJKokkos *aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
184   Mat_SeqAIJKokkos *aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
205   Mat_SeqAIJKokkos *aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
218   Mat_SeqAIJKokkos *aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
230   Mat_SeqAIJKokkos *aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
313   Mat_SeqAIJKokkos *akok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
318   PetscCheck(akok, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "Unexpected NULL (Mat_SeqAIJKokkos*)A->spptr");
351   Mat_SeqAIJKokkos *akok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
356   PetscCheck(akok, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "Unexpected NULL (Mat_SeqAIJKokkos*)A->spptr");
388   Mat_SeqAIJKokkos          *aijkok;
397   aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
410   Mat_SeqAIJKokkos          *aijkok;
425     aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
440   Mat_SeqAIJKokkos          *aijkok;
455     aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
470   Mat_SeqAIJKokkos          *aijkok;
480   aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
492   Mat_SeqAIJKokkos          *aijkok;
508     aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
523   Mat_SeqAIJKokkos          *aijkok;
539     aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
553   Mat_SeqAIJKokkos *aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
589       PetscCheck(!A->spptr, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "Expect NULL (Mat_SeqAIJKokkos*)A->spptr");
590       A->spptr = new Mat_SeqAIJKokkos(A, A->rmap->n, A->cmap->n, aseq, A->nonzerostate, PETSC_FALSE);
602   Mat_SeqAIJKokkos *akok = static_cast<Mat_SeqAIJKokkos *>(A->spptr), *bkok;
611     bkok = new Mat_SeqAIJKokkos(mat, mat->rmap->n, mat->cmap->n, bseq, mat->nonzerostate, PETSC_FALSE);
640   Mat_SeqAIJKokkos *atkok, *bkok;
647     PetscCallCXX(atkok = new Mat_SeqAIJKokkos(KokkosCsrMatrix("csrmat", internT)));
653       bkok = static_cast<Mat_SeqAIJKokkos *>((*B)->spptr);
669   Mat_SeqAIJKokkos *aijkok;
673     aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
714   Mat_SeqAIJKokkos   *akok, *bkok, *ckok;
733   akok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
734   bkok = static_cast<Mat_SeqAIJKokkos *>(B->spptr);
771     PetscCallCXX(ckok = new Mat_SeqAIJKokkos(m, n, nnz, ci, cj, ca));
776     ckok = static_cast<Mat_SeqAIJKokkos *>((*C)->spptr);
806   Mat_SeqAIJKokkos           *akok, *bkok, *ckok;
845   akok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
846   bkok = static_cast<Mat_SeqAIJKokkos *>(B->spptr);
847   ckok = static_cast<Mat_SeqAIJKokkos *>(C->spptr);
892   Mat_SeqAIJKokkos           *akok, *bkok, *ckok;
905   akok    = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
906   bkok    = static_cast<Mat_SeqAIJKokkos *>(B->spptr);
971     calling new Mat_SeqAIJKokkos().
982   PetscCallCXX(ckok = new Mat_SeqAIJKokkos(csrmatC));
1021   Mat_SeqAIJKokkos *aijkok;
1026   aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
1045     const auto  aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
1074     const auto  aijkok = static_cast<Mat_SeqAIJKokkos *>(Y->spptr);
1101   const auto  aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
1135   Mat_SeqAIJKokkos *aijkok;
1138   aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
1150   Mat_SeqAIJKokkos     *aijkok;
1160   aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
1179   Mat_SeqAIJKokkos *aijkok;
1186   aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
1202   Mat_SeqAIJKokkos *aijkok;
1209   aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
1226   Mat_SeqAIJKokkos *aijkok;
1232   aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
1249   Mat_SeqAIJKokkos *akok;
1252   PetscCallCXX(akok = new Mat_SeqAIJKokkos(m, n, j_d.extent(0), i_d, j_d, a_d));
1262   Mat_SeqAIJKokkos        *xkok, *ykok, *zkok;
1288   ykok = static_cast<Mat_SeqAIJKokkos *>(Y->spptr);
1289   xkok = static_cast<Mat_SeqAIJKokkos *>(X->spptr);
1333     zkok = new Mat_SeqAIJKokkos(zcsr);
1369   Mat_SeqAIJKokkos          *akok;
1378   akok = static_cast<Mat_SeqAIJKokkos *>(mat->spptr);
1380   mat->spptr = akok = new Mat_SeqAIJKokkos(mat, mat->rmap->n, mat->cmap->n, aseq, mat->nonzerostate + 1, PETSC_FALSE);
1494   Mat_SeqAIJKokkos *akok    = static_cast<Mat_SeqAIJKokkos *>(A->spptr);
1553 PETSC_INTERN PetscErrorCode MatSetSeqAIJKokkosWithCSRMatrix(Mat A, Mat_SeqAIJKokkos *akok)
1599   *csr = static_cast<Mat_SeqAIJKokkos *>(A->spptr)->csrmat;
1605   Mat_SeqAIJKokkos *akok;
1608   PetscCallCXX(akok = new Mat_SeqAIJKokkos(csr));
1614 /* Crete a SEQAIJKOKKOS matrix with a Mat_SeqAIJKokkos data structure
1618 PETSC_INTERN PetscErrorCode MatCreateSeqAIJKokkosWithCSRMatrix(MPI_Comm comm, Mat_SeqAIJKokkos *akok, Mat *A)
2054   Mat_SeqAIJKokkos           *aijkok   = (Mat_SeqAIJKokkos *)A->spptr;
2087   Mat_SeqAIJKokkos           *aijkok;
2109   aijkok   = (Mat_SeqAIJKokkos *)A->spptr;