| /petsc/src/ksp/ksp/impls/gmres/agmres/ |
| H A D | agmresleja.c | 23 static PetscErrorCode KSPAGMRESLejaCfpdMax(PetscScalar *rm, PetscScalar *im, PetscInt *spos, PetscI… in KSPAGMRESLejaCfpdMax() argument 35 id = im[i] - im[spos[j]]; in KSPAGMRESLejaCfpdMax() 47 PetscErrorCode KSPAGMRESLejaOrdering(PetscScalar *re, PetscScalar *im, PetscScalar *rre, PetscScala… in KSPAGMRESLejaOrdering() argument 59 if (im[j] != 0.0) { /* complex eigenvalue */ in KSPAGMRESLejaOrdering() 60 if (im[j] < 0.0) { /* change the order */ in KSPAGMRESLejaOrdering() 61 temp = im[j + 1]; in KSPAGMRESLejaOrdering() 62 im[j + 1] = im[j]; in KSPAGMRESLejaOrdering() 63 im[j] = temp; in KSPAGMRESLejaOrdering() 69 for (i = 0; i < m; i++) n_cmpl[i] = PetscSqrtReal(re[i] * re[i] + im[i] * im[i]); in KSPAGMRESLejaOrdering() 72 if (im[pos] >= 0.0) { in KSPAGMRESLejaOrdering() [all …]
|
| /petsc/src/dm/tests/ |
| H A D | ex30.c | 76 PetscInt row[2], col[9], im, ip; in main() local 80 im = (i == 0) ? n : i - 1; in main() 84 col[0] = im; in main() 108 col[0] = im; in main() 115 col[0] = 2 * im; in main() 116 col[1] = 2 * im + 1; in main()
|
| /petsc/src/mat/tests/ |
| H A D | ex116.c | 129 PetscBLASInt minMN, maxMN, im, in; in main() local 150 PetscCall(PetscBLASIntCast(m, &im)); in main() 153 …LAPACKgesvd_("S", "S", &im, &in, arrayA, &im, evals, arrayU, &minMN, arrayVT, &minMN, work, &lwork… in main() 168 …BLASgemm_("N", "N", &im, &minMN, &minMN, &alpha, arrayU, &im, arrayVT, &minMN, &beta, arrayErr, &i… in main()
|
| H A D | ex192.c | 133 PetscScalar im = PetscSqrtScalar((PetscScalar)-1.); in main() local 135 val = val + im; in main()
|
| /petsc/src/sys/classes/random/impls/random123/ |
| H A D | random123.c | 70 PetscReal im = PetscRandom123Step(r123); in PetscRandomGetValue_Random123() local 74 im = im * PetscImaginaryPart(r->width) + PetscImaginaryPart(r->low); in PetscRandomGetValue_Random123() 77 rscal = PetscCMPLX(re, im); in PetscRandomGetValue_Random123()
|
| /petsc/src/mat/impls/shell/ |
| H A D | shellcnv.c | 11 PetscInt im, i, m, n, M, N, *rows, start; in MatConvert_Shell() local 62 for (j = 0, im = 0; j < m; j++) { in MatConvert_Shell() 64 rows[im] = j + start; in MatConvert_Shell() 65 array[im] = array[j]; in MatConvert_Shell() 66 im++; in MatConvert_Shell() 68 PetscCall(MatSetValues(mat, im, rows, 1, &i, array, INSERT_VALUES)); in MatConvert_Shell()
|
| /petsc/src/mat/impls/baij/seq/ |
| H A D | baijfact3.c | 159 PetscInt *bdiag, row, nnz, nzi, reallocs = 0, nzbd, *im; in MatLUFactorSymbolic_SeqBAIJ() local 188 PetscCall(PetscMalloc2(n + 1, &bi_ptr, n + 1, &im)); in MatLUFactorSymbolic_SeqBAIJ() 209 PetscCall(PetscLLAddSortedLU(ajtmp, row, &nlnk, lnk, lnkbt, i, nzbd, im)); in MatLUFactorSymbolic_SeqBAIJ() 214 im[i] = nzi; in MatLUFactorSymbolic_SeqBAIJ() 249 PetscCall(PetscFree2(bi_ptr, im)); in MatLUFactorSymbolic_SeqBAIJ()
|
| /petsc/src/sys/tests/ |
| H A D | ex15.cxx | 75 InsertMode im; in main() local 155 im = INSERT_VALUES; in main() 203 (void)im; in main()
|
| /petsc/src/tao/pde_constrained/tutorials/ |
| H A D | elliptic.c | 544 PetscInt im, indx1, indx2, indy1, indy2, indz1, indz2, nx, ny, nz; in EllipticInitialize() local 1073 im = 0; in EllipticInitialize() 1074 xim = x[im]; in EllipticInitialize() 1075 while (xri > xim && im < nx) { in EllipticInitialize() 1076 im = im + 1; in EllipticInitialize() 1077 xim = x[im]; in EllipticInitialize() 1079 indx1 = im - 1; in EllipticInitialize() 1080 indx2 = im; in EllipticInitialize() 1085 im = 0; in EllipticInitialize() 1086 yim = y[im]; in EllipticInitialize() [all …]
|
| H A D | parabolic.c | 663 PetscInt im, indx1, indx2, indy1, indy2, indz1, indz2, nx, ny, nz; in ParabolicInitialize() local 1091 im = 0; in ParabolicInitialize() 1092 xim = x[im]; in ParabolicInitialize() 1093 while (xri > xim && im < nx) { in ParabolicInitialize() 1094 im = im + 1; in ParabolicInitialize() 1095 xim = x[im]; in ParabolicInitialize() 1097 indx1 = im - 1; in ParabolicInitialize() 1098 indx2 = im; in ParabolicInitialize() 1103 im = 0; in ParabolicInitialize() 1104 yim = y[im]; in ParabolicInitialize() [all …]
|
| /petsc/src/mat/impls/baij/mpi/ |
| H A D | mpibaij.c | 240 PetscErrorCode MatSetValues_MPIBAIJ(Mat mat, PetscInt m, const PetscInt im[], PetscInt n, const Pet… in MatSetValues_MPIBAIJ() argument 267 if (im[i] < 0) continue; in MatSetValues_MPIBAIJ() 268 …PetscCheck(im[i] < mat->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %"… in MatSetValues_MPIBAIJ() 269 if (im[i] >= rstart_orig && im[i] < rend_orig) { in MatSetValues_MPIBAIJ() 270 row = im[i] - rstart_orig; in MatSetValues_MPIBAIJ() 276 MatSetValues_SeqBAIJ_A_Private(row, col, value, addv, im[i], in[j]); in MatSetValues_MPIBAIJ() 301 …RANGE, "Inserting a new nonzero (%" PetscInt_FMT ", %" PetscInt_FMT ") into matrix", im[i], in[j]); in MatSetValues_MPIBAIJ() 307 MatSetValues_SeqBAIJ_B_Private(row, col, value, addv, im[i], in[j]); in MatSetValues_MPIBAIJ() 312 …w %" PetscInt_FMT " even though MatSetOption(,MAT_NO_OFF_PROC_ENTRIES,PETSC_TRUE) was set", im[i]); in MatSetValues_MPIBAIJ() 316 PetscCall(MatStashValuesRow_Private(&mat->stash, im[i], n, in, v + i * n, PETSC_FALSE)); in MatSetValues_MPIBAIJ() [all …]
|
| /petsc/src/mat/impls/sbaij/mpi/ |
| H A D | mpisbaij.c | 284 static PetscErrorCode MatSetValues_MPISBAIJ(Mat mat, PetscInt m, const PetscInt im[], PetscInt n, c… in MatSetValues_MPISBAIJ() argument 328 if (im[i] < 0) continue; in MatSetValues_MPISBAIJ() 329 …PetscCheck(im[i] < mat->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %"… in MatSetValues_MPISBAIJ() 330 if (im[i] >= rstart_orig && im[i] < rend_orig) { /* this processor entry */ in MatSetValues_MPISBAIJ() 331 row = im[i] - rstart_orig; /* local row index */ in MatSetValues_MPISBAIJ() 333 if (im[i] / bs > in[j] / bs) { in MatSetValues_MPISBAIJ() 344 MatSetValues_SeqSBAIJ_A_Private(row, col, value, addv, im[i], in[j]); in MatSetValues_MPISBAIJ() 374 MatSetValues_SeqSBAIJ_B_Private(row, col, value, addv, im[i], in[j]); in MatSetValues_MPISBAIJ() 379 …w %" PetscInt_FMT " even though MatSetOption(,MAT_NO_OFF_PROC_ENTRIES,PETSC_TRUE) was set", im[i]); in MatSetValues_MPISBAIJ() 384 if (im[i] / bs > in[j] / bs) continue; /* ignore lower triangular blocks */ in MatSetValues_MPISBAIJ() [all …]
|
| /petsc/src/ksp/pc/impls/parms/ |
| H A D | parms.c | 43 int *maptmp, *mapptr, *ia, *ja, *ja1, *im; in PCSetUp_PARMS() local 116 PetscCall(PetscMalloc1(lsize, &im)); in PCSetUp_PARMS() 117 PetscCall(PetscArraycpy(im, &maptmp[mapptr[rank] - 1], lsize)); in PCSetUp_PARMS() 123 parms_MatSetValues(parms->A, (int)lsize, im, ia, ja, aa, INSERT); in PCSetUp_PARMS() 131 PetscCall(PetscFree(im)); in PCSetUp_PARMS()
|
| /petsc/src/mat/impls/sell/mpi/ |
| H A D | mpisell.c | 150 static PetscErrorCode MatSetValues_MPISELL(Mat mat, PetscInt m, const PetscInt im[], PetscInt n, co… in MatSetValues_MPISELL() argument 169 if (im[i] < 0) continue; in MatSetValues_MPISELL() 170 …PetscCheck(im[i] < mat->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %"… in MatSetValues_MPISELL() 171 if (im[i] >= rstart && im[i] < rend) { in MatSetValues_MPISELL() 172 row = im[i] - rstart; in MatSetValues_MPISELL() 194 …MatSetValue_SeqSELL_Private(A, row, col, value, addv, im[i], in[j], cp1, vp1, lastcol1, low1, high… in MatSetValues_MPISELL() 224 …ew nonzero at global row/column (%" PetscInt_FMT ", %" PetscInt_FMT ") into matrix", im[i], in[j]); in MatSetValues_MPISELL() 227 …MatSetValue_SeqSELL_Private(B, row, col, value, addv, im[i], in[j], cp2, vp2, lastcol2, low2, high… in MatSetValues_MPISELL() 234 …w %" PetscInt_FMT " even though MatSetOption(,MAT_NO_OFF_PROC_ENTRIES,PETSC_TRUE) was set", im[i]); in MatSetValues_MPISELL() 238 …PetscCall(MatStashValuesRow_Private(&mat->stash, im[i], n, in, v + i * n, (PetscBool)(ignorezeroen… in MatSetValues_MPISELL() [all …]
|
| /petsc/src/binding/petsc4py/test/ |
| H A D | test_mat_py.py | 271 def setDiagonal(self, mat, vd, im): argument 272 if isinstance(im, bool): 273 addv = im 278 elif im == PETSc.InsertMode.INSERT_VALUES: 280 elif im == PETSc.InsertMode.ADD_VALUES: 283 raise ValueError('wrong InsertMode %d' % im)
|
| /petsc/src/sys/classes/viewer/interface/ |
| H A D | viewreg.c | 277 PetscBool tk, im; in PetscOptionsCreateViewers_Single() local 279 PetscCall(PetscStrcmp(loc1_fname, "image", &im)); in PetscOptionsCreateViewers_Single() 280 if (tk || im) { in PetscOptionsCreateViewers_Single()
|
| /petsc/src/binding/petsc4py/demo/python_types/ |
| H A D | matpython_protocol.py | 82 def setDiagonal(self, A: Mat, d: Vec, im: InsertMode) -> None:
|
| /petsc/src/snes/tutorials/ |
| H A D | ex30.c | 965 PetscInt i, j, is, js, im, jm; in Initialize() local 975 PetscCall(DMDAGetCorners(da, &is, &js, NULL, &im, &jm, NULL)); in Initialize() 980 for (i = is; i < is + im; i++) { in Initialize() 1100 PetscInt i, j, is, js, im, jm, ilim, jlim, ivt; in ViscosityField() local 1122 PetscCall(DMDAGetCorners(da, &is, &js, NULL, &im, &jm, NULL)); in ViscosityField() 1124 for (i = is; i < is + im; i++) { in ViscosityField() 1154 PetscInt i, j, is, js, im, jm; in StressField() local 1162 PetscCall(DMDAGetCorners(da, &is, &js, NULL, &im, &jm, NULL)); in StressField() 1172 for (i = is; i < is + im; i++) { in StressField()
|
| /petsc/src/binding/petsc4py/src/petsc4py/PETSc/ |
| H A D | DMDA.pyx | 875 cdef PetscInsertMode im = insertmode(addv) 876 CHKERR(DMDAGlobalToNaturalBegin(self.dm, vg.vec, im, vn.vec)) 877 CHKERR(DMDAGlobalToNaturalEnd(self.dm, vg.vec, im, vn.vec)) 903 cdef PetscInsertMode im = insertmode(addv) 904 CHKERR(DMDANaturalToGlobalBegin(self.dm, vn.vec, im, vg.vec)) 905 CHKERR(DMDANaturalToGlobalEnd(self.dm, vn.vec, im, vg.vec))
|
| H A D | DM.pyx | 990 cdef PetscInsertMode im = insertmode(addv) 991 CHKERR(DMGlobalToLocalBegin(self.dm, vg.vec, im, vl.vec)) 992 CHKERR(DMGlobalToLocalEnd (self.dm, vg.vec, im, vl.vec)) 1013 cdef PetscInsertMode im = insertmode(addv) 1014 CHKERR(DMLocalToGlobalBegin(self.dm, vl.vec, im, vg.vec)) 1015 CHKERR(DMLocalToGlobalEnd(self.dm, vl.vec, im, vg.vec)) 1036 cdef PetscInsertMode im = insertmode(addv) 1037 CHKERR(DMLocalToLocalBegin(self.dm, vl.vec, im, vlg.vec)) 1038 CHKERR(DMLocalToLocalEnd (self.dm, vl.vec, im, vlg.vec))
|
| /petsc/src/ksp/ksp/impls/cheby/ |
| H A D | cheby.c | 21 PetscReal *re, *im, min, max; in KSPChebyshevComputeExtremeEigenvalues_Private() local 25 PetscCall(PetscMalloc2(n, &re, n, &im)); in KSPChebyshevComputeExtremeEigenvalues_Private() 26 PetscCall(KSPComputeEigenvalues(kspest, n, re, im, &neig)); in KSPChebyshevComputeExtremeEigenvalues_Private() 33 PetscCall(PetscFree2(re, im)); in KSPChebyshevComputeExtremeEigenvalues_Private()
|
| /petsc/src/snes/tests/ |
| H A D | ex8.c | 583 PetscInt k, dim, d, im, fm; in CheckTransfer() local 590 PetscCall(VecGetLocalSize(iV, &im)); in CheckTransfer() 592 …PetscCall(MatCreateDense(PetscObjectComm((PetscObject)dm), im, PETSC_DECIDE, PETSC_DECIDE, user->K… in CheckTransfer()
|
| /petsc/src/ts/characteristic/interface/ |
| H A D | characteristic.c | 516 PetscScalar im = interpIndices[0]; in CharacteristicSolve() local 519 …im >= (PetscScalar)is - 1.) && (im <= (PetscScalar)ie) && (jm >= (PetscScalar)js - 1.) && (jm <= (… in CharacteristicSolve()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 529 PetscErrorCode MatSetValues_MPIAIJ(Mat mat, PetscInt m, const PetscInt im[], PetscInt n, const Pets… in MatSetValues_MPIAIJ() argument 554 if (im[i] < 0) continue; in MatSetValues_MPIAIJ() 555 …PetscCheck(im[i] < mat->rmap->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %"… in MatSetValues_MPIAIJ() 556 if (im[i] >= rstart && im[i] < rend) { in MatSetValues_MPIAIJ() 557 row = im[i] - rstart; in MatSetValues_MPIAIJ() 575 if (ignorezeroentries && value == 0.0 && (addv == ADD_VALUES) && im[i] != in[j]) continue; in MatSetValues_MPIAIJ() 579 MatSetValues_SeqAIJ_A_Private(row, col, value, addv, im[i], in[j]); in MatSetValues_MPIAIJ() 612 …ew nonzero at global row/column (%" PetscInt_FMT ", %" PetscInt_FMT ") into matrix", im[i], in[j]); in MatSetValues_MPIAIJ() 613 …of matrix %g(%" PetscInt_FMT ",%" PetscInt_FMT ")\n", (double)PetscRealPart(value), im[i], in[j])); in MatSetValues_MPIAIJ() 617 MatSetValues_SeqAIJ_B_Private(row, col, value, addv, im[i], in[j]); in MatSetValues_MPIAIJ() [all …]
|
| /petsc/src/mat/impls/sbaij/seq/ |
| H A D | sbaij.c | 510 PetscErrorCode MatGetValues_SeqSBAIJ(Mat A, PetscInt m, const PetscInt im[], PetscInt n, const Pets… in MatGetValues_SeqSBAIJ() argument 520 row = im[k]; in MatGetValues_SeqSBAIJ() 575 PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A, PetscInt m, const PetscInt im[], PetscInt n, con… in MatSetValuesBlocked_SeqSBAIJ() argument 589 row = im[k]; in MatSetValuesBlocked_SeqSBAIJ() 737 PetscErrorCode MatSetValues_SeqSBAIJ(Mat A, PetscInt m, const PetscInt im[], PetscInt n, const Pets… in MatSetValues_SeqSBAIJ() argument 748 row = im[k]; /* row number */ in MatSetValues_SeqSBAIJ()
|