Searched refs:kh2 (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 128 KernelHandle kh2; // compute C2, add C2+C4 or C2+Fo member
1078 PetscCallCXX(mm->kh2.create_spgemm_handle(spgemm_alg)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1102 PetscCallCXX(KokkosSparse::spgemm_symbolic(mm->kh2, Adt, false, Bo, false, mm->C2_mid)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1104 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh2, Adt, false, Bo, false, mm->C2_mid)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1120 PetscCallCXX(mm->kh2.create_spadd_handle(true)); // C2, Fo are sorted in MatProductSymbolic_MPIAIJKokkos_AtB()
1122 PetscCallCXX(KokkosSparse::spadd_symbolic(&mm->kh2, mm->C2, mm->Fo, mm->Co)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1124 PetscCallCXX(KokkosSparse::spadd_numeric(&mm->kh2, 1.0, mm->C2, 1.0, mm->Fo, mm->Co)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1151 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh2, Adt, false, Bo, false, mm->C2_mid)); in MatProductNumeric_MPIAIJKokkos_AtB()
1157 PetscCallCXX(KokkosSparse::spadd_numeric(&mm->kh2, 1.0, mm->C2, 1.0, mm->Fo, mm->Co)); in MatProductNumeric_MPIAIJKokkos_AtB()
1192 mm->kh2.create_spgemm_handle(spgemm_alg); in MatProductSymbolic_MPIAIJKokkos_AB()
[all …]
|