Lines Matching refs:Ao
97 Mat Ad, Ao;
101 PetscCall(MatMPIAIJGetSeqAIJ(mat, &Ad, &Ao, &cmap));
102 PetscCall(MatSeqAIJKokkosMergeMats(Ad, Ao, reuse, C));
106 PetscCall(MatGetLocalSize(Ao, NULL, &on));
1054 KokkosCsrMatrix Adt, Aot, Ad, Ao, Bd, Bo;
1063 PetscCall(MatSeqAIJKokkosGetKokkosCsrMatrix(ampi->B, &Ao));
1173 KokkosCsrMatrix Ad, Ao, Bd, Bo;
1177 PetscCall(MatSeqAIJKokkosGetKokkosCsrMatrix(ampi->B, &Ao));
1215 PetscCallCXX(KokkosSparse::spgemm_symbolic(mm->kh3, Ao, false, mm->Fd, false, mm->C3));
1216 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh3, Ao, false, mm->Fd, false, mm->C3));
1217 PetscCallCXX(KokkosSparse::spgemm_symbolic(mm->kh4, Ao, false, mm->Fo, false, mm->C4));
1218 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh4, Ao, false, mm->Fo, false, mm->C4));
1244 KokkosCsrMatrix Ad, Ao, Bd, Bo;
1248 PetscCall(MatSeqAIJKokkosGetKokkosCsrMatrix(ampi->B, &Ao));
1262 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh3, Ao, false, mm->Fd, false, mm->C3));
1263 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh4, Ao, false, mm->Fo, false, mm->C4));