Searched refs:garray1 (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ ! |
| H A D | mpiaijkok.kokkos.cxx | 265 static PetscErrorCode ReduceTwoSetsOfGlobalIndices(PetscInt n1, const PetscInt *garray1, PetscInt m… in ReduceTwoSetsOfGlobalIndices() argument
281 PetscCall(PetscHMapIGetWithDefault(g2l, garray1[i], -1, &val)); in ReduceTwoSetsOfGlobalIndices()
282 if (val < 0) PetscCall(PetscHMapISet(g2l, garray1[i], tot++)); in ReduceTwoSetsOfGlobalIndices()
308 …for (PetscInt i = 0; i < n1; i++) PetscCall(PetscHMapIGetWithDefault(g2l, garray1[i], -1, &map[i])… in ReduceTwoSetsOfGlobalIndices()
342 …ix A, KokkosCsrMatrix B, PetscInt cstart, PetscInt cend, const PetscInt *garray1, PetscSF ownerSF,… in MatMPIAIJKokkosReduceBegin() argument
373 if (garray1[*it] < cstart) { // map local to global in MatMPIAIJKokkosReduceBegin()
454 buf[j] = garray1[Bj[Bi[i] + j]]; // left B, in global in MatMPIAIJKokkosReduceBegin()
458 buf[j] = garray1[Bj[Bi[i] + j - alen]]; // right B, in global in MatMPIAIJKokkosReduceBegin()
597 PetscCall(ReduceTwoSetsOfGlobalIndices(n1, garray1, Fonz, Foj, &n2, &garray2, map)); in MatMPIAIJKokkosReduceBegin()
674 …ix A, KokkosCsrMatrix B, PetscInt cstart, PetscInt cend, const PetscInt *garray1, PetscSF ownerSF,… in MatMPIAIJKokkosReduceEnd() argument
[all …]
|