mpi/kokkos/mpiaijkok.kokkos.cxx

191   Kokkos::TeamPolicy<ExecutionSpace> teamPolicy(128, Kokkos::AUTO);
352     auto                 Ai_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), A.graph.row_map);
353     auto                 Aj_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), A.graph.entries);
354     auto                 Bi_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), B.graph.row_map);
355     auto                 Bj_h = Kokkos::create_mirror_view_and_copy(HostMirrorMemorySpace(), B.graph.entries);
631   auto       &Aa           = A.values;
632   auto       &Ba           = B.values;
633   const auto &Ai           = A.graph.row_map;
634   const auto &Bi           = B.graph.row_map;
635   const auto &E_NzLeft     = mm->E_NzLeft;
636   auto       &leafBuf      = mm->leafBuf;
637   auto       &rootBuf      = mm->rootBuf;
676   auto       &leafBuf  = mm->leafBuf;
677   auto       &rootBuf  = mm->rootBuf;
678   auto       &Fda      = mm->Fd.values;
679   const auto &Fdjmap   = mm->Fdjmap;
680   const auto &Fdjperm  = mm->Fdjperm;
681   auto        Fdnz     = mm->Fd.nnz();
682   auto       &Foa      = mm->Fo.values;
683   const auto &Fojmap   = mm->Fojmap;
684   const auto &Fojperm  = mm->Fojperm;
685   auto        Fonz     = mm->Fo.nnz();
954   const auto &Aa = akok->a_dual.view_device();
955   const auto &Ba = bkok->a_dual.view_device();
956   const auto &Ai = akok->i_dual.view_device();
957   const auto &Bi = bkok->i_dual.view_device();
1004   const auto &Fd  = mm->Fd;
1005   const auto &Fo  = mm->Fo;
1006   const auto &Fdi = Fd.graph.row_map;
1007   const auto &Foi = Fo.graph.row_map;
1008   auto       &Fda = Fd.values;
1009   auto       &Foa = Fo.values;
1010   auto        Fm  = Fd.numRows();
1068   auto spgemm_alg = KokkosSparse::SPGEMMAlgorithm::SPGEMM_DEFAULT; // default is TPL if enabled, otherwise KK
1182   auto spgemm_alg = KokkosSparse::SPGEMMAlgorithm::SPGEMM_DEFAULT; // default is TPL if enabled, otherwise KK
1361     auto mmAB = new MatMatStruct_AB();
1365     auto mmAtB = new MatMatStruct_AtB();
1371     auto mmAB = new MatMatStruct_AB();
1383     auto mmAtB = new MatMatStruct_AtB();
1516     auto exec = PetscGetKokkosExecutionSpace();
1592   const auto &n      = coo->n;
1593   const auto &Annz   = coo->Annz;
1594   const auto &Annz2  = coo->Annz2;
1595   const auto &Bnnz   = coo->Bnnz;
1596   const auto &Bnnz2  = coo->Bnnz2;
1597   const auto &vsend  = coo->sendbuf;
1598   const auto &v2     = coo->recvbuf;
1599   const auto &Ajmap1 = coo->Ajmap1;
1600   const auto &Ajmap2 = coo->Ajmap2;
1601   const auto &Aimap2 = coo->Aimap2;
1602   const auto &Bjmap1 = coo->Bjmap1;
1603   const auto &Bjmap2 = coo->Bjmap2;
1604   const auto &Bimap2 = coo->Bimap2;
1605   const auto &Aperm1 = coo->Aperm1;
1606   const auto &Aperm2 = coo->Aperm2;
1607   const auto &Bperm1 = coo->Bperm1;
1608   const auto &Bperm2 = coo->Bperm2;
1609   const auto &Cperm1 = coo->Cperm1;