Searched refs:garray2 (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/dm/tests/ |
| H A D | ex10k.kokkos.cxx | 114 Node **garray2; /* Node arrays of g, l */ in main() local
132 PetscCall(DMDAVecGetArrayWrite(da, g, &garray2)); in main()
135 garray2[j][i].u = 100 * j + 10 * (i + 1) + 111; in main()
136 garray2[j][i].v = 100 * j + 10 * (i + 1) + 222; in main()
137 garray2[j][i].w = 100 * j + 10 * (i + 1) + 333; in main()
140 PetscCall(DMDAVecRestoreArrayWrite(da, g, &garray2)); in main()
158 PetscCall(DMDAVecGetArrayWrite(da, g, &garray2)); in main()
162 …garray2[j][i].u = (larray2[j][i - 1].u + larray2[j][i + 1].u + larray2[j - 1][i].u + larray2[j + 1… in main()
163 …garray2[j][i].v = (larray2[j][i - 1].v + larray2[j][i + 1].v + larray2[j - 1][i].v + larray2[j + 1… in main()
164 …garray2[j][i].w = (larray2[j][i - 1].w + larray2[j][i + 1].w + larray2[j - 1][i].w + larray2[j + 1… in main()
[all …]
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 270 PetscInt n2, *garray2; in ReduceTwoSetsOfGlobalIndices() local
287 PetscCall(PetscMalloc1(n2, &garray2)); in ReduceTwoSetsOfGlobalIndices()
293 garray2[tot++] = key; in ReduceTwoSetsOfGlobalIndices()
297 PetscCall(PetscSortInt(n2, garray2)); in ReduceTwoSetsOfGlobalIndices()
299 …for (PetscInt i = 0; i < tot; i++) PetscCall(PetscHMapISet(g2l, garray2[i], i)); // i is the local… in ReduceTwoSetsOfGlobalIndices()
311 *garray2_ = garray2; in ReduceTwoSetsOfGlobalIndices()
595 PetscInt n2, *garray2; in MatMPIAIJKokkosReduceBegin() local
597 PetscCall(ReduceTwoSetsOfGlobalIndices(n1, garray1, Fonz, Foj, &n2, &garray2, map)); in MatMPIAIJKokkosReduceBegin()
601 mm->garray = garray2; // give ownership, so no free in MatMPIAIJKokkosReduceBegin()
906 PetscInt n2, *garray2; in MatMPIAIJKokkosBcastBegin() local
[all …]
|