Lines Matching refs:nx
261 PetscInt xs, ys, nx, ny, i, j, ii, gxs, gys, gnx, gny, m, n, M, N, dim, s, k, nc, col; in DMCreateColoring_DA_2d_MPIAIJ() local
277 PetscCall(DMDAGetCorners(da, &xs, &ys, NULL, &nx, &ny, NULL)); in DMCreateColoring_DA_2d_MPIAIJ()
287 PetscCall(PetscMalloc1(nc * nx * ny, &colors)); in DMCreateColoring_DA_2d_MPIAIJ()
290 for (i = xs; i < xs + nx; i++) { in DMCreateColoring_DA_2d_MPIAIJ()
295 …PetscCall(ISColoringCreate(comm, ncolors, nc * nx * ny, colors, PETSC_OWN_POINTER, &dd->localcolor… in DMCreateColoring_DA_2d_MPIAIJ()
325 …PetscInt xs, ys, nx, ny, i, j, gxs, gys, gnx, gny, m, n, p, dim, s, k, nc, col, zs, gzs, i… in DMCreateColoring_DA_3d_MPIAIJ() local
344 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &nx, &ny, &nz)); in DMCreateColoring_DA_3d_MPIAIJ()
351 PetscCall(PetscMalloc1(nc * nx * ny * nz, &colors)); in DMCreateColoring_DA_3d_MPIAIJ()
355 for (i = xs; i < xs + nx; i++) { in DMCreateColoring_DA_3d_MPIAIJ()
361 …PetscCall(ISColoringCreate(comm, ncolors, nc * nx * ny * nz, colors, PETSC_OWN_POINTER, &dd->local… in DMCreateColoring_DA_3d_MPIAIJ()
390 PetscInt xs, nx, i, i1, gxs, gnx, l, m, M, dim, s, nc, col; in DMCreateColoring_DA_1d_MPIAIJ() local
405 PetscCall(DMDAGetCorners(da, &xs, NULL, NULL, &nx, NULL, NULL)); in DMCreateColoring_DA_1d_MPIAIJ()
412 PetscCall(PetscMalloc1(nc * nx, &colors)); in DMCreateColoring_DA_1d_MPIAIJ()
417 for (i = xs; i < xs + nx; i++) { in DMCreateColoring_DA_1d_MPIAIJ()
429 for (i = xs; i < xs + nx; i++) { in DMCreateColoring_DA_1d_MPIAIJ()
434 …PetscCall(ISColoringCreate(comm, ncolors, nc * nx, colors, PETSC_OWN_POINTER, &dd->localcoloring)); in DMCreateColoring_DA_1d_MPIAIJ()
459 PetscInt xs, ys, nx, ny, i, j, ii, gxs, gys, gnx, gny, m, n, dim, s, k, nc; in DMCreateColoring_DA_2d_5pt_MPIAIJ() local
472 PetscCall(DMDAGetCorners(da, &xs, &ys, NULL, &nx, &ny, NULL)); in DMCreateColoring_DA_2d_5pt_MPIAIJ()
478 PetscCall(PetscMalloc1(nc * nx * ny, &colors)); in DMCreateColoring_DA_2d_5pt_MPIAIJ()
481 for (i = xs; i < xs + nx; i++) { in DMCreateColoring_DA_2d_5pt_MPIAIJ()
486 …PetscCall(ISColoringCreate(comm, ncolors, nc * nx * ny, colors, PETSC_OWN_POINTER, &dd->localcolor… in DMCreateColoring_DA_2d_5pt_MPIAIJ()
610 PetscInt dim, dof, nx, ny, nz, dims[3], starts[3], M, N, P; in DMCreateMatrix_DA() local
650 PetscCall(DMDAGetCorners(da, NULL, NULL, NULL, &nx, &ny, &nz)); in DMCreateMatrix_DA()
653 …PetscCall(MatSetSizes(A, dof * nx * ny * nz, dof * nx * ny * nz, dof * M * N * P, dof * M * N * P)… in DMCreateMatrix_DA()
656 if (dof * nx * ny * nz < da->bind_below) { in DMCreateMatrix_DA()
824 …PetscInt xs, ys, nx, ny, i, j, slot, gxs, gys, gnx, gny, m, n, dim, s, *cols = NULL,… in DMCreateMatrix_DA_2d_MPISELL() local
839 PetscCall(DMDAGetCorners(da, &xs, &ys, NULL, &nx, &ny, NULL)); in DMCreateMatrix_DA_2d_MPISELL()
848 MatPreallocateBegin(comm, nc * nx * ny, nc * nx * ny, dnz, onz); in DMCreateMatrix_DA_2d_MPISELL()
849 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_2d_MPISELL()
887 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_2d_MPISELL()
925 PetscInt xs, ys, nx, ny, i, j, slot, gxs, gys, gnx, gny; in DMCreateMatrix_DA_3d_MPISELL() local
941 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &nx, &ny, &nz)); in DMCreateMatrix_DA_3d_MPISELL()
950 MatPreallocateBegin(comm, nc * nx * ny * nz, nc * nx * ny * nz, dnz, onz); in DMCreateMatrix_DA_3d_MPISELL()
951 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_3d_MPISELL()
993 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_3d_MPISELL()
1036 …PetscInt xs, ys, nx, ny, i, j, slot, gxs, gys, gnx, gny, m, n, dim, s, *cols = NULL,… in DMCreateMatrix_DA_2d_MPIAIJ() local
1058 PetscCall(DMDAGetCorners(da, &xs, &ys, NULL, &nx, &ny, NULL)); in DMCreateMatrix_DA_2d_MPIAIJ()
1067 MatPreallocateBegin(comm, nc * nx * ny, nc * nx * ny, dnz, onz); in DMCreateMatrix_DA_2d_MPIAIJ()
1068 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_2d_MPIAIJ()
1106 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_2d_MPIAIJ()
1147 PetscInt xs, ys, nx, ny, i, j, slot, gxs, gys, gnx, gny; in DMCreateMatrix_DA_2d_MPIAIJ_Fill() local
1171 PetscCall(DMDAGetCorners(da, &xs, &ys, NULL, &nx, &ny, NULL)); in DMCreateMatrix_DA_2d_MPIAIJ_Fill()
1180 MatPreallocateBegin(comm, nc * nx * ny, nc * nx * ny, dnz, onz); in DMCreateMatrix_DA_2d_MPIAIJ_Fill()
1181 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_2d_MPIAIJ_Fill()
1226 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_2d_MPIAIJ_Fill()
1271 PetscInt xs, ys, nx, ny, i, j, slot, gxs, gys, gnx, gny; in DMCreateMatrix_DA_3d_MPIAIJ() local
1297 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &nx, &ny, &nz)); in DMCreateMatrix_DA_3d_MPIAIJ()
1306 MatPreallocateBegin(comm, nc * nx * ny * nz, nc * nx * ny * nz, dnz, onz); in DMCreateMatrix_DA_3d_MPIAIJ()
1307 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_3d_MPIAIJ()
1350 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_3d_MPIAIJ()
1397 PetscInt xs, nx, i, j, gxs, gnx, row, k, l; in DMCreateMatrix_DA_1d_MPIAIJ_Fill() local
1413 PetscCall(DMDAGetCorners(da, &xs, NULL, NULL, &nx, NULL, NULL)); in DMCreateMatrix_DA_1d_MPIAIJ_Fill()
1417 PetscCall(PetscCalloc2(nx * nc, &cols, nx * nc, &ocols)); in DMCreateMatrix_DA_1d_MPIAIJ_Fill()
1437 for (i = s; i < nx - s; i++) { in DMCreateMatrix_DA_1d_MPIAIJ_Fill()
1445 for (i = nx - s; i < nx; i++) { in DMCreateMatrix_DA_1d_MPIAIJ_Fill()
1447 ocols[cnt] = ((rank == (size - 1)) ? 0 : (i - nx + s + 1) * (ofill[j + 1] - ofill[j])); in DMCreateMatrix_DA_1d_MPIAIJ_Fill()
1448 cols[cnt] = dfill[j + 1] - dfill[j] + (s + nx - i - 1) * (ofill[j + 1] - ofill[j]); in DMCreateMatrix_DA_1d_MPIAIJ_Fill()
1450 if (size > 1) ocols[cnt] += (i - nx + s + 1) * (ofill[j + 1] - ofill[j]); in DMCreateMatrix_DA_1d_MPIAIJ_Fill()
1451 else cols[cnt] += (i - nx + s + 1) * (ofill[j + 1] - ofill[j]); in DMCreateMatrix_DA_1d_MPIAIJ_Fill()
1499 for (i = xs + s; i < xs + nx - s; i++) { in DMCreateMatrix_DA_1d_MPIAIJ_Fill()
1518 for (i = xs + nx - s; i < xs + nx; i++) { in DMCreateMatrix_DA_1d_MPIAIJ_Fill()
1556 PetscInt xs, nx, i, i1, slot, gxs, gnx; in DMCreateMatrix_DA_1d_MPIAIJ() local
1571 PetscCall(DMDAGetCorners(da, &xs, NULL, NULL, &nx, NULL, NULL)); in DMCreateMatrix_DA_1d_MPIAIJ()
1589 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_1d_MPIAIJ()
1620 PetscInt xs, nx, i, i1, slot, gxs, gnx; in DMCreateMatrix_DA_1d_SeqAIJ_NoPreallocation() local
1634 PetscCall(DMDAGetCorners(da, &xs, NULL, NULL, &nx, NULL, NULL)); in DMCreateMatrix_DA_1d_SeqAIJ_NoPreallocation()
1638 PetscCall(MatSeqAIJSetTotalPreallocation(J, nx * nc * col * nc)); in DMCreateMatrix_DA_1d_SeqAIJ_NoPreallocation()
1651 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_1d_SeqAIJ_NoPreallocation()
1683 PetscInt xs, ys, nx, ny, i, j, slot, gxs, gys, gnx, gny; in DMCreateMatrix_DA_2d_MPIBAIJ() local
1700 PetscCall(DMDAGetCorners(da, &xs, &ys, NULL, &nx, &ny, NULL)); in DMCreateMatrix_DA_2d_MPIBAIJ()
1709 MatPreallocateBegin(comm, nx * ny, nx * ny, dnz, onz); in DMCreateMatrix_DA_2d_MPIBAIJ()
1710 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_2d_MPIBAIJ()
1743 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_2d_MPIBAIJ()
1775 PetscInt xs, ys, nx, ny, i, j, slot, gxs, gys, gnx, gny; in DMCreateMatrix_DA_3d_MPIBAIJ() local
1792 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &nx, &ny, &nz)); in DMCreateMatrix_DA_3d_MPIBAIJ()
1801 MatPreallocateBegin(comm, nx * ny * nz, nx * ny * nz, dnz, onz); in DMCreateMatrix_DA_3d_MPIBAIJ()
1802 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_3d_MPIBAIJ()
1842 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_3d_MPIBAIJ()
1900 PetscInt xs, ys, nx, ny, i, j, slot, gxs, gys, gnx, gny; in DMCreateMatrix_DA_2d_MPISBAIJ() local
1917 PetscCall(DMDAGetCorners(da, &xs, &ys, NULL, &nx, &ny, NULL)); in DMCreateMatrix_DA_2d_MPISBAIJ()
1926 MatPreallocateBegin(comm, nx * ny, nx * ny, dnz, onz); in DMCreateMatrix_DA_2d_MPISBAIJ()
1927 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_2d_MPISBAIJ()
1959 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_2d_MPISBAIJ()
1992 PetscInt xs, ys, nx, ny, i, j, slot, gxs, gys, gnx, gny; in DMCreateMatrix_DA_3d_MPISBAIJ() local
2009 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &nx, &ny, &nz)); in DMCreateMatrix_DA_3d_MPISBAIJ()
2019 MatPreallocateBegin(comm, nx * ny * nz, nx * ny * nz, dnz, onz); in DMCreateMatrix_DA_3d_MPISBAIJ()
2020 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_3d_MPISBAIJ()
2059 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_3d_MPISBAIJ()
2098 PetscInt xs, ys, nx, ny, i, j, slot, gxs, gys, gnx, gny; in DMCreateMatrix_DA_3d_MPIAIJ_Fill() local
2126 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &nx, &ny, &nz)); in DMCreateMatrix_DA_3d_MPIAIJ_Fill()
2134 MatPreallocateBegin(comm, nc * nx * ny * nz, nc * nx * ny * nz, dnz, onz); in DMCreateMatrix_DA_3d_MPIAIJ_Fill()
2137 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_3d_MPIAIJ_Fill()
2188 for (i = xs; i < xs + nx; i++) { in DMCreateMatrix_DA_3d_MPIAIJ_Fill()