mpi/kokkos/mpiaijkok.kokkos.cxx

120   KokkosCsrMatrix     C1, C2, C3, C4; // intermediate mat products
127   KernelHandle kh1; // compute C1, add C1+C3 or C1+Fd
1101   PetscCallCXX(KokkosSparse::spgemm_symbolic(mm->kh1, Adt, false, Bd, false, mm->C1));
1103   PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh1, Adt, false, Bd, false, mm->C1));
1106   PetscCallCXX(sort_crs_matrix(mm->C1));
1118   // C = (C1+Fd, C2+Fo)
1119   PetscCallCXX(mm->kh1.create_spadd_handle(true)); // C1, Fd are sorted
1121   PetscCallCXX(KokkosSparse::spadd_symbolic(&mm->kh1, mm->C1, mm->Fd, mm->Cd));
1123   PetscCallCXX(KokkosSparse::spadd_numeric(&mm->kh1, 1.0, mm->C1, 1.0, mm->Fd, mm->Cd));
1150   PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh1, Adt, false, Bd, false, mm->C1));
1155   // C = (C1+Fd, C2+Fo)
1156   PetscCallCXX(KokkosSparse::spadd_numeric(&mm->kh1, 1.0, mm->C1, 1.0, mm->Fd, mm->Cd));
1201   PetscCallCXX(KokkosSparse::spgemm_symbolic(mm->kh1, Ad, false, Bd, false, mm->C1));
1205   PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh1, Ad, false, Bd, false, mm->C1));
1208   PetscCallCXX(sort_crs_matrix(mm->C1));
1230   // C = (Cd, Co) = (C1+C3, C2+C4)
1231   mm->kh1.create_spadd_handle(true); // C1, C3 are sorted
1233   PetscCallCXX(KokkosSparse::spadd_symbolic(&mm->kh1, mm->C1, mm->C3, mm->Cd));
1235   PetscCallCXX(KokkosSparse::spadd_numeric(&mm->kh1, 1.0, mm->C1, 1.0, mm->C3, mm->Cd));
1256   PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh1, Ad, false, Bd, false, mm->C1));
1265   // C = (Cd, Co) = (C1+C3, C2+C4)
1266   PetscCallCXX(KokkosSparse::spadd_numeric(&mm->kh1, 1.0, mm->C1, 1.0, mm->C3, mm->Cd));