Searched refs:Aot (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 1054 KokkosCsrMatrix Adt, Aot, Ad, Ao, Bd, Bo; in MatProductSymbolic_MPIAIJKokkos_AtB() local
1061 PetscCall(MatSeqAIJKokkosGenerateTranspose_Private(ampi->B, &Aot)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1083 PetscCallCXX(KokkosSparse::spgemm_symbolic(mm->kh3, Aot, false, Bd, false, mm->C3)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1084 PetscCallCXX(KokkosSparse::spgemm_symbolic(mm->kh4, Aot, false, Bo, false, mm->C4)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1087 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh3, Aot, false, Bd, false, mm->C3)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1088 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh4, Aot, false, Bo, false, mm->C4)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1132 KokkosCsrMatrix Adt, Aot, Bd, Bo; in MatProductNumeric_MPIAIJKokkos_AtB() local
1138 PetscCall(MatSeqAIJKokkosGenerateTranspose_Private(ampi->B, &Aot)); in MatProductNumeric_MPIAIJKokkos_AtB()
1143 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh3, Aot, false, Bd, false, mm->C3)); in MatProductNumeric_MPIAIJKokkos_AtB()
1144 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh4, Aot, false, Bo, false, mm->C4)); in MatProductNumeric_MPIAIJKokkos_AtB()
|