Searched refs:sendCol (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 445 PetscInt *sendCol, *recvCol; in MatMPIAIJKokkosReduceBegin() local
446 PetscCall(PetscMalloc2(nleaves, &sendCol, nroots, &recvCol)); in MatMPIAIJKokkosReduceBegin()
449 PetscInt *buf = &sendCol[Ai[i] + Bi[i]]; 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()
537 PetscCall(PetscFree2(sendCol, recvCol)); in MatMPIAIJKokkosReduceBegin()
|