Searched refs:reduceSF (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 388 PetscSF reduceSF; in MatMPIAIJKokkosReduceBegin() local
441 PetscCall(PetscSFCreate(comm, &reduceSF)); in MatMPIAIJKokkosReduceBegin()
442 …PetscCall(PetscSFSetGraph(reduceSF, nroots, nleaves, NULL, PETSC_OWN_POINTER, iremote, PETSC_OWN_P… in MatMPIAIJKokkosReduceBegin()
462 …PetscCall(PetscSFReduceWithMemTypeBegin(reduceSF, MPIU_INT, PETSC_MEMTYPE_HOST, sendCol, PETSC_MEM… in MatMPIAIJKokkosReduceBegin()
463 PetscCall(PetscSFReduceEnd(reduceSF, MPIU_INT, sendCol, recvCol, MPI_REPLACE)); in MatMPIAIJKokkosReduceBegin()
598 mm->sf = reduceSF; in MatMPIAIJKokkosReduceBegin()
638 PetscSF reduceSF = mm->sf; in MatMPIAIJKokkosReduceBegin() local
669 …PetscCall(PetscSFReduceWithMemTypeBegin(reduceSF, MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, leafBuf.data(… in MatMPIAIJKokkosReduceBegin()
686 PetscSF reduceSF = mm->sf; in MatMPIAIJKokkosReduceEnd() local
689 PetscCall(PetscSFReduceEnd(reduceSF, MPIU_SCALAR, leafBuf.data(), rootBuf.data(), MPI_REPLACE)); in MatMPIAIJKokkosReduceEnd()
|