Home
last modified time | relevance | path

Searched refs:nn (Results 1 – 25 of 55) sorted by relevance

123

/petsc/src/ksp/ksp/tutorials/ !
H A Dex56.c27 PetscInt m, nn, M, Istart, Iend, i, j, k, ii, jj, kk, ic, ne = 4, id; in main() local
71 nn = ne + 1; in main()
73 M = 3 * nn * nn * nn; /* global number of equations */ in main()
76 else m = nn * nn * nn; in main()
79 m = nn * nn * nn / npe; in main()
80 if (mype == npe - 1) m = nn * nn * nn - (npe - 1) * m; in main()
91 const PetscInt Ni0 = ipx * (nn / NP), Nj0 = ipy * (nn / NP), Nk0 = ipz * (nn / NP); in main()
92 … const PetscInt Ni1 = Ni0 + (m > 0 ? (nn / NP) : 0), Nj1 = Nj0 + (nn / NP), Nk1 = Nk0 + (nn / NP); in main()
93 const PetscInt NN = nn / NP, id0 = ipz * nn * nn * NN + ipy * nn * NN * NN + ipx * NN * NN * NN; in main()
97 …PetscCheck(nn == NP * (nn / NP), comm, PETSC_ERR_ARG_WRONG, "-ne %" PetscInt_FMT ": (ne+1)%%(npe^{… in main()
[all …]
H A Dex19.c153 PetscInt nn; in main() local
155 PetscCall(ISGetSize(loc_blocks[bid], &nn)); // size only in main()
156 PetscCall(ISCreateStride(PETSC_COMM_SELF, nn, gid0, 1, &new_loc_blocks)); in main()
159 gid0 += nn; // start of next block in main()
/petsc/src/dm/impls/da/ !
H A Dda3.c210 PetscInt left, right, up, down, bottom, top, i, j, k, *idx, nn; in DMSetUp_DA_3D() local
460 nn = x * y * z; in DMSetUp_DA_3D()
462 PetscCallMPI(MPI_Allgather(&nn, 1, MPIU_INT, ldims, 1, MPIU_INT, comm)); in DMSetUp_DA_3D()
810 nn = 0; in DMSetUp_DA_3D()
820 for (j = 0; j < s_x; j++) idx[nn++] = s_t++; in DMSetUp_DA_3D()
828 for (j = 0; j < x_t; j++) idx[nn++] = s_t++; in DMSetUp_DA_3D()
836 for (j = 0; j < s_x; j++) idx[nn++] = s_t++; in DMSetUp_DA_3D()
847 for (j = 0; j < s_x; j++) idx[nn++] = s_t++; in DMSetUp_DA_3D()
856 for (j = 0; j < x_t; j++) idx[nn++] = s_t++; in DMSetUp_DA_3D()
858 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + j + x * i + (s_z - k - 1) * x * y; in DMSetUp_DA_3D()
[all …]
H A Dda2.c203 PetscInt up, down, left, right, i, *idx, nn; in DMSetUp_DA_2D() local
373 nn = x * y; in DMSetUp_DA_2D()
375 PetscCallMPI(MPI_Allgather(&nn, 1, MPIU_INT, ldims, 1, MPIU_INT, comm)); in DMSetUp_DA_2D()
529 nn = 0; in DMSetUp_DA_2D()
536 for (j = 0; j < s_x; j++) idx[nn++] = s_t++; in DMSetUp_DA_2D()
543 for (j = 0; j < x_t; j++) idx[nn++] = s_t++; in DMSetUp_DA_2D()
545 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + x * (s_y - i + 1) + j; in DMSetUp_DA_2D()
552 for (j = 0; j < s_x; j++) idx[nn++] = s_t++; in DMSetUp_DA_2D()
561 for (j = 0; j < s_x; j++) idx[nn++] = s_t++; in DMSetUp_DA_2D()
563 for (j = 0; j < s_x; j++) idx[nn++] = bases[rank] + x * i + s_x - j; in DMSetUp_DA_2D()
[all …]
H A Dda1.c138 PetscInt i, *idx, nn, left, xs, xe, x, Xs, Xe, start, m, IXs, IXe; in DMSetUp_DA_1D() local
239 nn = IXs - Xs; in DMSetUp_DA_1D()
242 if ((xs - sDist + i) >= 0) idx[nn++] = xs - sDist + i; in DMSetUp_DA_1D()
243 else idx[nn++] = M + (xs - sDist + i); in DMSetUp_DA_1D()
246 for (i = 0; i < x; i++) idx[nn++] = xs + i; /* Non-ghost points */ in DMSetUp_DA_1D()
249 if ((xe + i) < M) idx[nn++] = xe + i; in DMSetUp_DA_1D()
250 else idx[nn++] = (xe + i) - M; in DMSetUp_DA_1D()
254 if ((xs - sDist + i) >= 0) idx[nn++] = xs - sDist + i; in DMSetUp_DA_1D()
255 else idx[nn++] = sDist - i; in DMSetUp_DA_1D()
258 for (i = 0; i < x; i++) idx[nn++] = xs + i; /* Non-ghost points */ in DMSetUp_DA_1D()
[all …]
H A Ddagetelem.c47 PetscInt corners[4], nn = 0; in DMDAGetElements_2D() local
61 nn = da->nen; in DMDAGetElements_2D()
74 PetscCall(PetscMalloc1(1 + nn * da->ne, &da->e)); in DMDAGetElements_2D()
82 for (c = 0; c < ns * nn; c++) da->e[cnt++] = cell[split[c]]; in DMDAGetElements_2D()
85 for (c = 0; c < ns * nn; c++) da->e[cnt++] = cell[c]; in DMDAGetElements_2D()
113 PetscInt corners[8], nn = 0; in DMDAGetElements_3D() local
127 nn = da->nen; in DMDAGetElements_3D()
143 PetscCall(PetscMalloc1(1 + nn * da->ne, &da->e)); in DMDAGetElements_3D()
156 for (c = 0; c < ns * nn; c++) da->e[cnt++] = cell[split[c]]; in DMDAGetElements_3D()
159 for (c = 0; c < ns * nn; c++) da->e[cnt++] = cell[c]; in DMDAGetElements_3D()
H A Dgrvtk.c172 PetscMPIInt nn; in DMDAVTKWriteAll_VTS() local
175 PetscCallMPI(MPI_Get_count(&status, MPIU_SCALAR, &nn)); in DMDAVTKWriteAll_VTS()
176 PetscCheck(nn == nnodes * cdim, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Array size mismatch"); in DMDAVTKWriteAll_VTS()
219 PetscMPIInt nn; in DMDAVTKWriteAll_VTS() local
222 PetscCallMPI(MPI_Get_count(&status, MPIU_SCALAR, &nn)); in DMDAVTKWriteAll_VTS()
223 …PetscCheck(nn == nnodes * bs, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Array size mismatch receiving from… in DMDAVTKWriteAll_VTS()
389 PetscMPIInt nn; in DMDAVTKWriteAll_VTR() local
392 PetscCallMPI(MPI_Get_count(&status, MPIU_SCALAR, &nn)); in DMDAVTKWriteAll_VTR()
393 PetscCheck(nn == xm + ym + zm, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Array size mismatch"); in DMDAVTKWriteAll_VTR()
462 PetscMPIInt nn; in DMDAVTKWriteAll_VTR() local
[all …]
/petsc/src/binding/petsc4py/demo/legacy/taosolve/ !
H A Drosenbrock.py35 nn = self.size // 2
37 for i in range(nn):
46 nn = self.size // 2
48 for i in range(nn):
57 nn = self.size // 2
60 for i in range(nn):
71 nn = self.size // 2
76 for i in range(nn):
/petsc/src/tao/unconstrained/tutorials/ !
H A Drosenbrock1f.F9040 PetscInt i, nn
47 nn = n/2
55 do i = 0, nn - 1
68 PetscCall(PetscLogFlops(15.0d0*nn, ierr))
109 PetscInt i, nn, ind(0:1), i2
115 nn = n/2
128 do i = 0, nn - 1
147 PetscCall(PetscLogFlops(9.0d0*nn, ierr))
H A Drosenbrock1.c140 PetscInt i, nn = user->n / 2; in FormFunctionGradient() local
160 for (i = 0; i < nn; i++) { in FormFunctionGradient()
174 PetscCall(PetscLogFlops(15.0 * nn)); in FormFunctionGradient()
H A Drosenbrock2.c131 PetscInt i, nn = user->n / 2; in FormFunctionGradient() local
151 for (i = 0; i < nn; i++) { in FormFunctionGradient()
165 PetscCall(PetscLogFlops(15.0 * nn)); in FormFunctionGradient()
H A Drosenbrock3.c136 PetscInt i, nn = user->n / 2; in FormFunctionGradient() local
156 for (i = 0; i < nn; i++) { in FormFunctionGradient()
170 PetscCall(PetscLogFlops(15.0 * nn)); in FormFunctionGradient()
/petsc/src/ts/tutorials/ !
H A Dex11.h174 PetscReal nn[DIM]; in PhysicsRiemann_SW_Rusanov() local
202 nn[0] = n[0]; in PhysicsRiemann_SW_Rusanov()
203 nn[1] = n[1]; in PhysicsRiemann_SW_Rusanov()
204 Normalize2Real(nn); in PhysicsRiemann_SW_Rusanov()
205 ierr = SWFlux(phys, nn, uL, &fL.swnode); in PhysicsRiemann_SW_Rusanov()
211 ierr = SWFlux(phys, nn, uR, &fR.swnode); in PhysicsRiemann_SW_Rusanov()
219 …speed = PetscMax(PetscAbsReal(Dot2Real(uL->uh, nn) / uL->h) + cL, PetscAbsReal(Dot2Real(uR->uh, nn… in PhysicsRiemann_SW_Rusanov()
274 PetscReal nn[DIM]; in PhysicsRiemann_SW_HLL() local
299 nn[0] = n[0]; in PhysicsRiemann_SW_HLL()
300 nn[1] = n[1]; in PhysicsRiemann_SW_HLL()
[all …]
/petsc/src/sys/classes/random/impls/curand/ !
H A Dcurand.c23 size_t nn = n < 0 ? (size_t)(-2 * n) : (size_t)n; /* handle complex case */ in PetscRandomGetValuesReal_CURAND() local
27 PetscCallCURAND(curandGenerateUniform(curand->gen, val, nn)); in PetscRandomGetValuesReal_CURAND()
29 PetscCallCURAND(curandGenerateUniformDouble(curand->gen, val, nn)); in PetscRandomGetValuesReal_CURAND()
31 if (r->iset) PetscCall(PetscRandomCurandScale_Private(r, nn, val, (PetscBool)(n < 0))); in PetscRandomGetValuesReal_CURAND()
/petsc/src/sys/classes/draw/tests/ !
H A Dex9.c12 PetscInt w = 400, h = 300, nn = 20, b = 8, c = PETSC_DRAW_GREEN; in main() local
26 PetscCall(PetscOptionsGetInt(NULL, NULL, "-n", &nn, NULL)); in main()
32 PetscCall(PetscCIntCast(nn, &n)); in main()
/petsc/src/mat/impls/baij/seq/ftn-kernels/ !
H A Dfsolvebaij.F9090 PetscInt :: jstart, jend, idx, ax, jdx, kdx, nn
127 nn = 4*(jend - jstart + 1) - 1
129 do jj = 0, nn
155 nn = 4*(jend - jstart + 1) - 1
157 do jj = 0, nn
/petsc/src/vec/is/is/utils/ !
H A Discoloring.c230 PetscErrorCode ISColoringGetIS(ISColoring iscoloring, PetscCopyMode mode, PetscInt *nn, IS *isis[]) in ISColoringGetIS() argument
235 if (nn) *nn = iscoloring->n; in ISColoringGetIS()
675 PetscMPIInt size, *sizes = NULL, *offsets = NULL, nn; in ISAllGather() local
692 PetscCall(PetscMPIIntCast(n, &nn)); in ISAllGather()
693 PetscCallMPI(MPI_Allgather(&nn, 1, MPI_INT, sizes, 1, MPI_INT, comm)); in ISAllGather()
703 …PetscCallMPI(MPI_Allgatherv((void *)lindices, nn, MPIU_INT, indices, sizes, offsets, MPIU_INT, com… in ISAllGather()
738 PetscMPIInt size, *offsets = NULL, *sizes = NULL, nn; in ISAllGatherColors() local
741 PetscCall(PetscMPIIntCast(n, &nn)); in ISAllGatherColors()
745 PetscCallMPI(MPI_Allgather(&nn, 1, MPI_INT, sizes, 1, MPI_INT, comm)); in ISAllGatherColors()
752 …PetscCallMPI(MPI_Allgatherv(lindices, nn, MPIU_COLORING_VALUE, indices, sizes, offsets, MPIU_COLOR… in ISAllGatherColors()
/petsc/src/vec/vec/tests/ !
H A Dex49.c73 const PetscInt nn = ln[neighbor]; in main() local
77 for (j = 0; j < nn / 4; j += 3) { in main()
/petsc/src/mat/impls/sell/seq/ !
H A Dfdsell.c10 …PetscInt oshift, PetscBool symmetric, PetscBool inodecompressed, PetscInt *nn, const PetscInt *ia[… in MatGetColumnIJ_SeqSELL_Color() argument
20 *nn = n; in MatGetColumnIJ_SeqSELL_Color()
/petsc/src/mat/tests/ !
H A Dex120.c20 PetscBLASInt nn, nevs, il, iu, *iwork, *ifail, lwork, lierr, bn, one = 1; in main() local
162 PetscCall(PetscBLASIntCast(n, &nn)); in main()
163 …&bn, arrayA, &bn, &vl, &vu, &il, &iu, &abstol, &nevs, evals, evecs_array, &nn, work, &lwork, rwork… in main()
191 PetscCall(PetscBLASIntCast(n, &nn)); in main()
192 …&bn, arrayB, &bn, &vl, &vu, &il, &iu, &abstol, &nevs, evals, evecs_array, &nn, work, &lwork, rwork… in main()
H A Dex91.c8 PetscInt bs = 1, m = 43, ov = 1, i, j, k, *rows, *cols, M, nd = 5, *idx, mm, nn; in main() local
84 PetscCall(MatGetSize(A, &mm, &nn)); in main()
141 PetscCall(MatGetSize(submatA[i], &mm, &nn)); in main()
165 PetscCall(MatGetSize(submatA[i], &mm, &nn)); in main()
/petsc/include/petsc/private/kernels/ !
H A Dpetscaxpy.h35 PetscBLASInt nn = (PetscBLASInt)n; \
36 PetscCallBLAS("BLASaxpy", BLASaxpy_(&nn, &a1, p1, &one, U, &one)); \
176 PetscBLASInt nn = (PetscBLASInt)n; \
177 PetscCallBLAS("BLASaxpy", BLASaxpy_(&nn, &a1, p1, &one, U, &one)); \
/petsc/src/mat/impls/aij/seq/matlab/ !
H A Daijmatlab.c42 mwIndex nnz, nn, nm, *ii, *jj; in MatSeqAIJFromMatlab() local
46 nn = mxGetN(mmat); /* rows of transpose of matrix */ in MatSeqAIJFromMatlab()
48 nnz = (mxGetJc(mmat))[nn]; in MatSeqAIJFromMatlab()
51 n = (PetscInt)nn; in MatSeqAIJFromMatlab()
/petsc/src/tao/util/ !
H A Dtao_util.c249 PetscInt i, nn; in MatDFischer() local
255 PetscCall(VecGetLocalSize(X, &nn)); in MatDFischer()
265 for (i = 0; i < nn; i++) { in MatDFischer()
288 for (i = 0; i < nn; i++) { in MatDFischer()
395 PetscInt i, nn; in MatDSFischer() local
405 PetscCall(VecGetLocalSize(X, &nn)); in MatDSFischer()
414 for (i = 0; i < nn; ++i) { in MatDSFischer()
/petsc/src/mat/graphops/color/utils/ !
H A Dvalid.c100 PetscInt nn, c, i, j, M, N, nc, nnz, col, row; in MatISColoringTest() local
109 PetscCall(ISColoringGetIS(iscoloring, PETSC_USE_POINTER, &nn, &isis)); in MatISColoringTest()
120 for (c = 0; c < nn; c++) { /* for each color */ in MatISColoringTest()

123