Searched refs:kh1 (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 127 KernelHandle kh1; // compute C1, add C1+C3 or C1+Fd member
1077 PetscCallCXX(mm->kh1.create_spgemm_handle(spgemm_alg)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1101 PetscCallCXX(KokkosSparse::spgemm_symbolic(mm->kh1, Adt, false, Bd, false, mm->C1)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1103 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh1, Adt, false, Bd, false, mm->C1)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1119 PetscCallCXX(mm->kh1.create_spadd_handle(true)); // C1, Fd are sorted in MatProductSymbolic_MPIAIJKokkos_AtB()
1121 PetscCallCXX(KokkosSparse::spadd_symbolic(&mm->kh1, mm->C1, mm->Fd, mm->Cd)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1123 PetscCallCXX(KokkosSparse::spadd_numeric(&mm->kh1, 1.0, mm->C1, 1.0, mm->Fd, mm->Cd)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1150 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh1, Adt, false, Bd, false, mm->C1)); in MatProductNumeric_MPIAIJKokkos_AtB()
1156 PetscCallCXX(KokkosSparse::spadd_numeric(&mm->kh1, 1.0, mm->C1, 1.0, mm->Fd, mm->Cd)); in MatProductNumeric_MPIAIJKokkos_AtB()
1191 mm->kh1.create_spgemm_handle(spgemm_alg); in MatProductSymbolic_MPIAIJKokkos_AB()
[all …]
|