Lines Matching refs:Fm
346 PetscInt Em = A.numRows(), Fm;
349 PetscCall(PetscSFGetGraph(ownerSF, &Fm, NULL, NULL, NULL)); // Fm = #rows of F = nroots of ownerSF
475 MatRowMapKokkosViewHost Fdi_h(NoInit("Fdi_h"), Fm + 1), Foi_h(NoInit("Foi_h"), Fm + 1);
531 for (PetscInt i = 0; i < Fm; i++) {
535 Fdnz = Fdi[Fm];
536 Fonz = Foi[Fm];
617 PetscCallCXX(mm->Fd = KokkosCsrMatrix("Fd", Fm, cend - cstart, Fdnz, Fda_d, Fdi_d, Fdj_d));
618 PetscCallCXX(mm->Fo = KokkosCsrMatrix("Fo", Fm, n2, Fonz, Foa_d, Foi_d, Foj_d)); // Fo's column size is n2, length of garray2[]
782 PetscInt *Fi, Fm, Fnz;
783 PetscCall(PetscSFGetGraph(ownerSF, NULL, &Fm, NULL, NULL)); // Fm = #rows of F = nleaves of ownerSF
784 PetscCall(PetscMalloc1(Fm + 1, &Fi));
788 for (PetscInt i = 0; i < Fm; i++) Fi[i + 1] += Fi[i];
789 Fnz = Fi[Fm];
861 MatRowMapKokkosViewHost Fdi_h(NoInit("Fdi_h"), Fm + 1), Foi_h(NoInit("Foi_h"), Fm + 1); // row pointer of Fd, Fo
862 MatColIdxKokkosViewHost F_NzLeft_h(NoInit("F_NzLeft_h"), Fm); // split each row of F into Left, Diag, Right. We only need to record #nz in Left and Diag.
867 for (PetscInt i = 0; i < Fm; i++) {
878 for (PetscInt i = 0; i < Fm; i++) {
884 PetscInt Fdnz = Fdi[Fm], Fonz = Foi[Fm];
888 for (PetscInt i = 0; i < Fm; i++) {
930 PetscCallCXX(mm->Fd = KokkosCsrMatrix("Fd", Fm, cend - cstart, Fdnz, Fda_d, Fdi_d, Fdj_d));
931 PetscCallCXX(mm->Fo = KokkosCsrMatrix("Fo", Fm, n2, Fonz, Foa_d, Foi_d, Foj_d));
943 PetscCall(MatMergeGetLaunchParameters<DefaultExecutionSpace>(Fm, Fnz, -1, teamSize, vectorLength, rowsPerTeam));
1010 auto Fm = Fd.numRows();
1019 PetscInt workSets = (Fm + rowsPerTeam - 1) / rowsPerTeam;
1028 if (i < Fm) {