seq/kokkos/aijkok.kokkos.cxx

98   } else if (A->rmap->n && aijkok->diag_dual.extent(0) == 0) { // MatProduct might directly produce AIJ on device, but not the diag.
746     auto ca = MatScalarKokkosView("a", aa.extent(0) + ba.extent(0));
747     auto ci = MatRowMapKokkosView("i", ai.extent(0));
748     auto cj = MatColIdxKokkosView("j", aj.extent(0) + bj.extent(0));
1029   PetscCall(PetscLogGpuFlops(aijkok->a_dual.extent(0)));
1252   PetscCallCXX(akok = new Mat_SeqAIJKokkos(m, n, j_d.extent(0), i_d, j_d, a_d));
1339   PetscCall(PetscLogGpuFlops(xkok->a_dual.extent(0) * 2)); // Because we scaled X and then added it to Y
1495   PetscInt          nblocks = bs.extent(0) - 1;
1679   const PetscBool             has_lower = factors->iL_d.extent(0) ? PETSC_TRUE : PETSC_FALSE; // false with Choleksy
1680   const PetscBool             has_upper = factors->iU_d.extent(0) ? PETSC_TRUE : PETSC_FALSE; // true with LU and Choleksy
1695   const PetscBool             has_lower = factors->iL_d.extent(0) ? PETSC_TRUE : PETSC_FALSE; // false with Choleksy
1696   const PetscBool             has_upper = factors->iU_d.extent(0) ? PETSC_TRUE : PETSC_FALSE; // true with LU or Choleksy
1701       if (!factors->iUt_d.extent(0)) {                                 // Allocate Ut on device if not yet
1703         factors->jUt_d = MatColIdxKokkosView(NoInit("factors->jUt_d"), factors->jU_d.extent(0));
1704         factors->aUt_d = MatScalarKokkosView(NoInit("factors->aUt_d"), factors->aU_d.extent(0));
1707       if (factors->iU_h.extent(0)) { // If U is on host (factorization was done on host), we also compute the transpose on host
1738       if (!factors->iLt_d.extent(0)) {                                 // Allocate Lt on device if not yet
1740         factors->jLt_d = MatColIdxKokkosView(NoInit("factors->jLt_d"), factors->jL_d.extent(0));
1741         factors->aLt_d = MatScalarKokkosView(NoInit("factors->aLt_d"), factors->aL_d.extent(0));
1744       if (factors->iL_h.extent(0)) { // If L is on host, we also compute the transpose on host
1790   PetscBool                   identity = rowperm.extent(0) ? PETSC_FALSE : PETSC_TRUE;
1801     B = PetscScalarKokkosView(const_cast<PetscScalar *>(b.data()), b.extent(0));
1844   PetscBool                   row_identity = rowperm.extent(0) ? PETSC_FALSE : PETSC_TRUE;
1845   PetscBool                   col_identity = colperm.extent(0) ? PETSC_FALSE : PETSC_TRUE;
1855     B = PetscScalarKokkosView(const_cast<PetscScalar *>(b.data()), b.extent(0));
1895   PetscBool                   row_identity = rowperm.extent(0) ? PETSC_FALSE : PETSC_TRUE;
1896   PetscBool                   col_identity = colperm.extent(0) ? PETSC_FALSE : PETSC_TRUE;
1906     B = PetscScalarKokkosView(const_cast<PetscScalar *>(b.data()), b.extent(0));
1945     if (factors->iL_h.extent(0) == 0) { // Allocate memory and copy the L, U structure for the first time
2189     if (factors->iU_h.extent(0) == 0) { // First time of numeric factorization