Home
last modified time | relevance | path

Searched refs:C4 (Results 1 – 2 of 2) sorted by relevance

/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx120 KokkosCsrMatrix C1, C2, C3, C4; // intermediate mat products member
1084 PetscCallCXX(KokkosSparse::spgemm_symbolic(mm->kh4, Aot, false, Bo, false, mm->C4)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1088 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh4, Aot, false, Bo, false, mm->C4)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1092 PetscCallCXX(sort_crs_matrix(mm->C4)); in MatProductSymbolic_MPIAIJKokkos_AtB()
1098 …PetscCall(MatMPIAIJKokkosReduceBegin(comm, mm->C3, mm->C4, cstart, cend, bmpi->garray, ampi->Mvctx… in MatProductSymbolic_MPIAIJKokkos_AtB()
1110 …PetscCall(MatMPIAIJKokkosReduceEnd(comm, mm->C3, mm->C4, cstart, cend, bmpi->garray, ampi->Mvctx, … in MatProductSymbolic_MPIAIJKokkos_AtB()
1144 PetscCallCXX(KokkosSparse::spgemm_numeric(mm->kh4, Aot, false, Bo, false, mm->C4)); in MatProductNumeric_MPIAIJKokkos_AtB()
1147 …PetscCall(MatMPIAIJKokkosReduceBegin(comm, mm->C3, mm->C4, 0, 0, NULL, NULL, MAT_REUSE_MATRIX, NUL… in MatProductNumeric_MPIAIJKokkos_AtB()
1153 …PetscCall(MatMPIAIJKokkosReduceEnd(comm, mm->C3, mm->C4, 0, 0, NULL, NULL, MAT_REUSE_MATRIX, NULL,… in MatProductNumeric_MPIAIJKokkos_AtB()
1217 PetscCallCXX(KokkosSparse::spgemm_symbolic(mm->kh4, Ao, false, mm->Fo, false, mm->C4)); in MatProductSymbolic_MPIAIJKokkos_AB()
[all …]
/petsc/src/mat/impls/baij/seq/
H A Dbaij.h980 …A2, A3, A4, A5, A6, A7, A8, B0, B1, B2, B3, B4, B5, B6, B7, B8, C0, C1, C2, C3, C4, C5, C6, C7, C8; in PetscKernel_A_gets_A_minus_B_times_C_9() local
1013 C4 = _mm256_broadcast_sd(C + 10); in PetscKernel_A_gets_A_minus_B_times_C_9()
1018 A3 = _mm256_fnmadd_pd(B3, C4, A3); in PetscKernel_A_gets_A_minus_B_times_C_9()
1019 A4 = _mm256_fnmadd_pd(B4, C4, A4); in PetscKernel_A_gets_A_minus_B_times_C_9()
1020 A5 = _mm256_fnmadd_pd(B5, C4, A5); in PetscKernel_A_gets_A_minus_B_times_C_9()
1061 C4 = _mm256_broadcast_sd(C + 13); in PetscKernel_A_gets_A_minus_B_times_C_9()
1066 A3 = _mm256_fnmadd_pd(B3, C4, A3); in PetscKernel_A_gets_A_minus_B_times_C_9()
1067 A4 = _mm256_fnmadd_pd(B4, C4, A4); in PetscKernel_A_gets_A_minus_B_times_C_9()
1068 A5 = _mm256_fnmadd_pd(B5, C4, A5); in PetscKernel_A_gets_A_minus_B_times_C_9()
1109 C4 = _mm256_broadcast_sd(C + 16); in PetscKernel_A_gets_A_minus_B_times_C_9()
[all …]