| /petsc/src/ksp/pc/impls/tfs/ |
| H A D | gs.c | 40 PetscScalar *vals; member 134 static PetscErrorCode PCTFS_gs_gop_local_out(PCTFS_gs_id *gs, PetscScalar *vals); 136 static PetscErrorCode PCTFS_gs_gop_vec_local_out(PCTFS_gs_id *gs, PetscScalar *vals, PetscInt step); 140 static PetscErrorCode PCTFS_gs_gop_vec_local_plus(PCTFS_gs_id *gs, PetscScalar *vals, PetscInt step… 141 static PetscErrorCode PCTFS_gs_gop_vec_local_in_plus(PCTFS_gs_id *gs, PetscScalar *vals, PetscInt s… 142 static PetscErrorCode PCTFS_gs_gop_vec_tree_plus(PCTFS_gs_id *gs, PetscScalar *vals, PetscInt step); 144 static PetscErrorCode PCTFS_gs_gop_local_plus(PCTFS_gs_id *gs, PetscScalar *vals); 145 static PetscErrorCode PCTFS_gs_gop_local_in_plus(PCTFS_gs_id *gs, PetscScalar *vals); 149 static PetscErrorCode PCTFS_gs_gop_tree_plus_hc(PCTFS_gs_id *gs, PetscScalar *vals, PetscInt dim); 222 PetscInt vals[PETSC_STATIC_ARRAY_LENGTH(oprs) - 1]; in gsi_check_args() local [all …]
|
| H A D | comm.c | 62 PetscErrorCode PCTFS_giop(PetscInt *vals, PetscInt *work, PetscInt n, PetscInt *oprs) in PCTFS_giop() argument 71 …PetscCheck(vals && work && oprs, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PCTFS_giop() :: vals=%p, work=%… in PCTFS_giop() 95 … PetscCallMPI(MPI_Send(vals, n, MPIU_INT, edge_not_pow_2, MSGTAG0 + PCTFS_my_id, MPI_COMM_WORLD)); in PCTFS_giop() 98 PetscCall((*fp)(vals, work, n, oprs)); in PCTFS_giop() 107 PetscCallMPI(MPI_Send(vals, n, MPIU_INT, dest, MSGTAG2 + PCTFS_my_id, MPI_COMM_WORLD)); in PCTFS_giop() 110 PetscCall((*fp)(vals, work, n, oprs)); in PCTFS_giop() 120 PetscCallMPI(MPI_Send(vals, n, MPIU_INT, dest, MSGTAG4 + PCTFS_my_id, MPI_COMM_WORLD)); in PCTFS_giop() 122 …PetscCallMPI(MPI_Recv(vals, n, MPIU_INT, MPI_ANY_SOURCE, MSGTAG4 + dest, MPI_COMM_WORLD, &status)); in PCTFS_giop() 130 …PetscCallMPI(MPI_Recv(vals, n, MPIU_INT, MPI_ANY_SOURCE, MSGTAG5 + edge_not_pow_2, MPI_COMM_WORLD,… in PCTFS_giop() 132 … PetscCallMPI(MPI_Send(vals, n, MPIU_INT, edge_not_pow_2, MSGTAG5 + PCTFS_my_id, MPI_COMM_WORLD)); in PCTFS_giop() [all …]
|
| H A D | xyt.c | 159 PetscInt vals[9], work[9]; in XYT_stats() local 172 vals[0] = vals[1] = vals[2] = xyt_handle->info->nnz; in XYT_stats() 173 vals[3] = vals[4] = vals[5] = xyt_handle->mvi->n; in XYT_stats() 174 vals[6] = vals[7] = vals[8] = xyt_handle->info->msg_buf_sz; in XYT_stats() 175 PetscCall(PCTFS_giop(vals, work, PETSC_STATIC_ARRAY_LENGTH(op) - 1, op)); in XYT_stats() 181 …ll(PetscPrintf(PETSC_COMM_WORLD, "%d :: min xyt_nnz=%" PetscInt_FMT "\n", PCTFS_my_id, vals[0])); in XYT_stats() 182 …ll(PetscPrintf(PETSC_COMM_WORLD, "%d :: max xyt_nnz=%" PetscInt_FMT "\n", PCTFS_my_id, vals[1])); in XYT_stats() 183 …PETSC_COMM_WORLD, "%d :: avg xyt_nnz=%g\n", PCTFS_my_id, (double)(1.0 * vals[2] / PCTFS_num_node… in XYT_stats() 184 …ll(PetscPrintf(PETSC_COMM_WORLD, "%d :: tot xyt_nnz=%" PetscInt_FMT "\n", PCTFS_my_id, vals[2])); in XYT_stats() 185 …M_WORLD, "%d :: xyt C(2d) =%g\n", PCTFS_my_id, (double)(vals[2] / (PetscPowReal(1.0 * vals[5], … in XYT_stats() [all …]
|
| /petsc/src/snes/tutorials/ |
| H A D | ex70.c | 256 PetscScalar vals[5]; in StokesSetupMatBlock00() local 270 PetscCall(StokesStencilLaplacian(s, i, j, &sz, cols, vals)); in StokesSetupMatBlock00() 275 for (i = 0; i < sz; i++) vals[i] = -1.0 * vals[i]; /* dynamic viscosity coef mu=-1 */ in StokesSetupMatBlock00() 276 PetscCall(MatSetValues(s->subA[0], 1, &row, sz, cols, vals, INSERT_VALUES)); in StokesSetupMatBlock00() 287 PetscScalar vals[5]; in StokesSetupMatBlock01() local 304 PetscCall(StokesStencilGradientX(s, i, j, &sz, cols, vals)); in StokesSetupMatBlock01() 306 PetscCall(StokesStencilGradientY(s, i, j, &sz, cols, vals)); in StokesSetupMatBlock01() 308 PetscCall(MatSetValues(s->subA[1], 1, &row, sz, cols, vals, INSERT_VALUES)); in StokesSetupMatBlock01() 380 …tencilLaplacian(Stokes *s, PetscInt i, PetscInt j, PetscInt *sz, PetscInt *cols, PetscScalar *vals) in StokesStencilLaplacian() argument 392 vals[0] = -(ae + awb + asb + an); in StokesStencilLaplacian() [all …]
|
| /petsc/src/mat/tutorials/ |
| H A D | ex15f.F90 | 13 PetscScalar, pointer, dimension(:) :: vals 32 allocate (vals(2)) 33 vals = 1.0 35 PetscCallA(MatSetValues(A, one, [r], two, cols, vals, INSERT_VALUES, ierr)) 37 deallocate (vals) 39 allocate (vals(2)) 40 vals = 1.0 42 PetscCallA(MatSetValues(A, one, [r], two, cols, vals, INSERT_VALUES, ierr)) 44 deallocate (vals) 46 allocate (vals(3)) [all …]
|
| H A D | ex15.c | 36 PetscScalar vals[2]; in main() local 40 vals[0] = 1.0; in main() 41 vals[1] = use_edge_weights ? 2.0 : 1.0; in main() 43 PetscCall(MatSetValues(A, 1, &r, 2, cols, vals, INSERT_VALUES)); in main() 46 PetscScalar vals[2]; in main() local 50 vals[0] = use_edge_weights ? 3.0 : 1.0; in main() 51 vals[1] = 1.0; in main() 53 PetscCall(MatSetValues(A, 1, &r, 2, cols, vals, INSERT_VALUES)); in main() 56 PetscScalar vals[3]; in main() local 62 vals[0] = use_edge_weights ? (cols[0] == 0 ? 2.0 : 5.0) : 1.0; in main() [all …]
|
| /petsc/src/ksp/ksp/tests/ |
| H A D | ex13.c | 85 PetscScalar *vals; in main() local 88 PetscCall(PetscMalloc2(nnz, &cols, nnz, &vals)); in main() 91 vals[0] = -2.0; in main() 92 vals[1] = 4.0; in main() 93 vals[2] = -2.0; in main() 98 PetscCall(MatSetValues(A, 1, &row, 3, cols, vals, INSERT_VALUES)); in main() 103 for (PetscInt i = 0; i < 9; i++) vals[i] = 1.0; in main() 108 PetscCall(MatSetValues(A, 1, &row, 9, cols, vals, INSERT_VALUES)); in main() 115 vals[0] = row - 10.0; in main() 116 PetscCall(MatSetValue(A, row, row, vals[0], INSERT_VALUES)); in main() [all …]
|
| /petsc/src/dm/tests/ |
| H A D | noflux_check.c | 10 PetscScalar vals[7]; in globalKMat_3d() local 25 vals[ncols++] = -6.; //ncols=1 in globalKMat_3d() 30 vals[ncols++] = 1.; //ncols=2 in globalKMat_3d() 35 vals[ncols++] = 1.; //ncols=3 in globalKMat_3d() 40 vals[ncols++] = 1.; //ncols=4 in globalKMat_3d() 45 vals[ncols++] = 1.; //ncols=5 in globalKMat_3d() 50 vals[ncols++] = 1.; //ncols=6 in globalKMat_3d() 55 vals[ncols++] = 1.; //ncols=7 in globalKMat_3d() 57 PetscCall(MatSetValuesStencil(K, 1, &row, ncols, col, vals, ADD_VALUES)); in globalKMat_3d() 69 PetscScalar vals[5]; in globalKMat_2d() local [all …]
|
| /petsc/src/dm/dt/fe/tests/ |
| H A D | ex4.c | 56 PetscScalar vals[6]; in main() local 88 PetscCall(VecSum(work, &vals[3])); in main() 91 PetscCall(VecSum(work, &vals[4])); in main() 92 PetscCall(VecSum(lM, &vals[5])); in main() 93 PetscCall(DMPlexComputeIntegralFEM(dm, ones, vals, NULL)); in main() 94 CheckVals(vals[0], vals[1], rtol, atol, "Error volume"); in main() 95 CheckVals((3 + dim) * vals[0], vals[3], rtol, atol, "Error mass"); in main() 96 CheckVals((3 + dim) * vals[0], vals[4], rtol, atol, "Error local lumped mass"); in main() 97 CheckVals((3 + dim) * vals[0], vals[5], rtol, atol, "Error lumped mass"); in main()
|
| /petsc/src/ksp/pc/impls/kaczmarz/ |
| H A D | kaczmarz.c | 20 const PetscScalar *vals, *xarray; in PCApply_Kaczmarz() local 34 PetscCall(MatGetRow(pc->pmat, i, &ncols, &cols, &vals)); in PCApply_Kaczmarz() 38 if (cols[j] >= ys && cols[j] < ye) r -= yarray[cols[j] - ys] * vals[j]; in PCApply_Kaczmarz() 39 anrm += PetscRealPart(PetscSqr(vals[j])); in PCApply_Kaczmarz() 43 if (cols[j] >= ys && cols[j] < ye) yarray[cols[j] - ys] += vals[j] * lambda * r / anrm; in PCApply_Kaczmarz() 46 PetscCall(MatRestoreRow(pc->pmat, i, &ncols, &cols, &vals)); in PCApply_Kaczmarz() 50 PetscCall(MatGetRow(pc->pmat, i, &ncols, &cols, &vals)); in PCApply_Kaczmarz() 54 if (cols[j] >= ys && cols[j] < ye) r -= yarray[cols[j] - ys] * vals[j]; in PCApply_Kaczmarz() 55 anrm += PetscRealPart(PetscSqr(vals[j])); in PCApply_Kaczmarz() 59 if (cols[j] >= ys && cols[j] < ye) yarray[cols[j] - ys] += vals[j] * lambda * r / anrm; in PCApply_Kaczmarz() [all …]
|
| /petsc/src/sys/tests/ |
| H A D | ex40.c | 25 PetscInt n, v, koff, keys[4], voff, vals[4], na, nb, i, size, *karray, off; in PETSC_HASH_MAP() local 93 vals[0] = vals[1] = 0; in PETSC_HASH_MAP() 94 PetscCall(PetscHMapIGetVals(ht, &voff, vals)); in PETSC_HASH_MAP() 95 PetscCall(PetscSortInt(voff, vals)); in PETSC_HASH_MAP() 97 PetscTestCheck(vals[0] == 24); in PETSC_HASH_MAP() 98 PetscTestCheck(vals[1] == 42); in PETSC_HASH_MAP() 103 vals[0] = vals[1] = 0; in PETSC_HASH_MAP() 106 PetscCall(PetscHMapIGetVals(ht, &voff, vals)); in PETSC_HASH_MAP() 108 PetscCall(PetscSortInt(voff, vals)); in PETSC_HASH_MAP() 113 PetscTestCheck(vals[0] == 24); in PETSC_HASH_MAP() [all …]
|
| /petsc/src/sys/classes/random/impls/random123/ |
| H A D | random123.c | 99 static PetscErrorCode PetscRandomGetValuesReal_Random123(PetscRandom r, PetscInt n, PetscReal vals[… in PetscRandomGetValuesReal_Random123() argument 111 for (PetscInt i = 0; i < peel_start; i++) PetscCall(PetscRandomGetValueReal(r, &vals[i])); in PetscRandomGetValuesReal_Random123() 114 vals += peel_start; in PetscRandomGetValuesReal_Random123() 123 for (PetscInt i = 0; i < lim; i += 4, vals += 4) { in PetscRandomGetValuesReal_Random123() 124 vals[0] = r123_copy.result.v[0] * scale + shift; in PetscRandomGetValuesReal_Random123() 125 vals[1] = r123_copy.result.v[1] * scale + shift; in PetscRandomGetValuesReal_Random123() 126 vals[2] = r123_copy.result.v[2] * scale + shift; in PetscRandomGetValuesReal_Random123() 127 vals[3] = r123_copy.result.v[3] * scale + shift; in PetscRandomGetValuesReal_Random123() 136 for (PetscInt i = 0; i < rem; i++) PetscCall(PetscRandomGetValueReal(r, &vals[i])); in PetscRandomGetValuesReal_Random123() 140 static PetscErrorCode PetscRandomGetValues_Random123(PetscRandom r, PetscInt n, PetscScalar vals[]) in PetscRandomGetValues_Random123() argument [all …]
|
| /petsc/src/mat/tests/ |
| H A D | ex170.c | 61 PetscScalar vals[3] = {1.0, 1.0, 1.0}; in CreateGraph() local 73 …if ((row >= rStart) && (row < rEnd)) PetscCall(MatSetValues(G, 1, &row, 2, cols, vals, INSERT_VALU… in CreateGraph() 77 …if ((row >= rStart) && (row < rEnd)) PetscCall(MatSetValues(G, 1, &row, 2, cols, vals, INSERT_VALU… in CreateGraph() 81 …if ((row >= rStart) && (row < rEnd)) PetscCall(MatSetValues(G, 1, &row, 2, cols, vals, INSERT_VALU… in CreateGraph() 85 …if ((row >= rStart) && (row < rEnd)) PetscCall(MatSetValues(G, 1, &row, 2, cols, vals, INSERT_VALU… in CreateGraph() 89 …if ((row >= rStart) && (row < rEnd)) PetscCall(MatSetValues(G, 1, &row, 2, cols, vals, INSERT_VALU… in CreateGraph() 94 PetscScalar vals[3] = {1.0, 1.0, 1.0}; in CreateGraph() local 106 …if ((row >= rStart) && (row < rEnd)) PetscCall(MatSetValues(G, 1, &row, 2, cols, vals, INSERT_VALU… in CreateGraph() 110 …if ((row >= rStart) && (row < rEnd)) PetscCall(MatSetValues(G, 1, &row, 2, cols, vals, INSERT_VALU… in CreateGraph() 114 …if ((row >= rStart) && (row < rEnd)) PetscCall(MatSetValues(G, 1, &row, 2, cols, vals, INSERT_VALU… in CreateGraph() [all …]
|
| H A D | ex241f.F90 | 16 PetscScalar vals(1) 38 cols = [row]; vals = [one]; rows = [row] 39 PetscCallA(MatSetValues(L, i1, rows, i1, cols, vals, ADD_VALUES, ierr)) 55 vals = [zero] 57 PetscCallA(MatSetValues(L, i1, rows, i1, cols, vals, ADD_VALUES, ierr)) 59 cols = [row]; vals = [one]; rows = [row] 60 PetscCallA(MatSetValues(L, i1, rows, i1, cols, vals, ADD_VALUES, ierr))
|
| H A D | ex266.c | 10 PetscScalar *vals; in main() local 48 PetscCall(PetscMalloc1(mycoo.n, &vals)); in main() 50 vals[k] = mycoo.j[k]; in main() 51 PetscCall(MatSetValue(A, mycoo.i[k], mycoo.j[k], vals[k], ADD_VALUES)); in main() 60 …for (k = 0; k < mycoo.n / 2; k++) PetscCall(MatSetValue(B, mycoo.i[k], mycoo.j[k], vals[k], ADD_VA… in main() 66 …for (k = mycoo.n / 2; k < mycoo.n; k++) PetscCall(MatSetValue(C, mycoo.i[k], mycoo.j[k], vals[k], … in main() 74 PetscCall(PetscFree(vals)); in main()
|
| H A D | ex156.c | 8 PetscScalar vals[16]; in Assemble() local 13 for (i = 0; i < 16; ++i) vals[i] = 1; in Assemble() 20 PetscCall(MatSetValues(mat, 4, idx, 4, idx, vals, ADD_VALUES)); in Assemble() 26 PetscCall(MatSetValues(mat, 4, idx, 4, idx, vals, ADD_VALUES)); in Assemble() 33 PetscCall(MatSetValues(mat, 4, idx, 4, idx, vals, ADD_VALUES)); in Assemble() 39 PetscCall(MatSetValues(mat, 4, idx, 4, idx, vals, ADD_VALUES)); in Assemble()
|
| H A D | ex86.c | 8 PetscScalar value[3], *vals; in main() local 46 PetscCall(PetscMalloc3(bs * bs, &vals, bs, &rows, bs, &cols)); in main() 48 for (i = 0; i < bs * bs; i++) vals[i] = 2.0; in main() 54 PetscCall(MatSetValues(seqmat, bs, rows, bs, cols, vals, INSERT_VALUES)); in main() 57 for (i = 0; i < bs * bs; i++) vals[i] = -1.0; in main() 63 PetscCall(MatSetValues(seqmat, bs, rows, bs, cols, vals, INSERT_VALUES)); in main() 66 PetscCall(PetscFree3(vals, rows, cols)); in main()
|
| H A D | ex238.c | 10 PetscScalar rval, *vals; in main() local 28 PetscCall(PetscMalloc1(bs * bs, &vals)); in main() 37 vals[k * bs + l] = rval; in main() 40 PetscCall(MatSetValuesBlocked(A, 1, &row, 1, &col, vals, INSERT_VALUES)); in main() 45 PetscCall(PetscFree(vals)); in main()
|
| H A D | ex19.c | 39 PetscScalar one = 1.0, Ke[16], *vals; in main() local 108 PetscCall(PetscMalloc1(nrsub * ncsub, &vals)); in main() 113 PetscCall(MatGetValues(C, nrsub, rsub, ncsub, csub, vals)); in main() 118 if (PetscImaginaryPart(vals[i * ncsub + j]) != 0.0) { in main() 119 … %g i\n", rsub[i], csub[j], (double)PetscRealPart(vals[i * ncsub + j]), (double)PetscImaginaryPart… in main() 121 …_FMT ", %" PetscInt_FMT "] = %g\n", rsub[i], csub[j], (double)PetscRealPart(vals[i * ncsub + j]))); in main() 128 PetscCall(PetscFree(vals)); in main()
|
| H A D | ex254.c | 10 PetscScalar *vals; in main() local 76 PetscCall(PetscMalloc1(mycoo.n, &vals)); in main() 78 vals[k] = mycoo.j[k]; in main() 79 PetscCall(MatSetValue(A, mycoo.i[k], mycoo.j[k], vals[k], ADD_VALUES)); in main() 92 PetscCall(MatSetValuesCOO(B, vals, ADD_VALUES)); in main() 99 PetscCall(MatSetValuesCOO(C, vals, ADD_VALUES)); in main() 113 PetscCall(PetscFree(vals)); in main()
|
| /petsc/src/mat/utils/ |
| H A D | axpy.c | 167 const PetscScalar *vals; in MatAXPY_Basic_Preallocate() local 169 PetscCall(MatGetRow(Y, r, &ncols, &row, &vals)); in MatAXPY_Basic_Preallocate() 170 PetscCall(MatSetValues(preallocator, 1, &r, ncols, row, vals, INSERT_VALUES)); in MatAXPY_Basic_Preallocate() 171 PetscCall(MatRestoreRow(Y, r, &ncols, &row, &vals)); in MatAXPY_Basic_Preallocate() 172 PetscCall(MatGetRow(X, r, &ncols, &row, &vals)); in MatAXPY_Basic_Preallocate() 173 PetscCall(MatSetValues(preallocator, 1, &r, ncols, row, vals, INSERT_VALUES)); in MatAXPY_Basic_Preallocate() 174 PetscCall(MatRestoreRow(X, r, &ncols, &row, &vals)); in MatAXPY_Basic_Preallocate() 205 const PetscScalar *vals; in MatAXPY_Basic() local 213 PetscCall(MatGetRow(X, i, &ncols, &row, &vals)); in MatAXPY_Basic() 214 PetscCall(MatSetValues(Y, 1, &i, ncols, row, vals, ADD_VALUES)); in MatAXPY_Basic() [all …]
|
| /petsc/src/vec/vec/tests/ |
| H A D | ex2k.c | 25 PetscScalar *vals; in main() local 57 PetscCall(PetscMalloc1(maxN, &vals)); in main() 58 for (j = 0; j < maxN; j++) vals[j] = 3.14 + j; // same across all processes in main() 81 if (testMDot) PetscCall(VecMDot(x, N, ys, vals)); in main() 82 else if (testMAXPY) PetscCall(VecMAXPY(x, N, vals, ys)); in main() 90 if (testMDot) PetscCall(VecMDot(x, N, ys, vals)); in main() 91 else if (testMAXPY) PetscCall(VecMAXPY(x, N, vals, ys)); in main() 119 PetscCall(PetscFree(vals)); in main()
|
| /petsc/src/vec/vec/tutorials/ |
| H A D | ex44.c | 11 PetscScalar *vals; in main() local 29 PetscCall(PetscMalloc2(x_size, &idx, x_size, &vals)); in main() 32 vals[j] = (PetscScalar)(shift + j + 1); in main() 35 …PetscCall(VecSetValues(x[i], x_size, (const PetscInt *)idx, (const PetscScalar *)vals, INSERT_VALU… in main() 38 PetscCall(PetscFree2(idx, vals)); in main() 46 PetscCall(PetscMalloc2(y_size, &idx, y_size, &vals)); in main() 49 vals[j] = (PetscScalar)(j + 1); in main() 51 …PetscCall(VecSetValues(y, y_size, (const PetscInt *)idx, (const PetscScalar *)vals, INSERT_VALUES)… in main() 54 PetscCall(PetscFree2(idx, vals)); in main()
|
| /petsc/src/ts/tutorials/ |
| H A D | ex15.c | 252 PetscScalar vals[5], hx, hy, sx, sy; in FormIJacobian() local 271 vals[nc++] = 1.0; in FormIJacobian() 276 vals[nc++] = 1.0; in FormIJacobian() 279 vals[nc++] = -1.0; in FormIJacobian() 283 vals[nc++] = 1.0; in FormIJacobian() 286 vals[nc++] = -1.0; in FormIJacobian() 290 vals[nc++] = 1.0; in FormIJacobian() 293 vals[nc++] = -1.0; in FormIJacobian() 297 vals[nc++] = 1.0; in FormIJacobian() 300 vals[nc++] = -1.0; in FormIJacobian() [all …]
|
| /petsc/src/ksp/pc/impls/spai/ |
| H A D | dspai.c | 28 const PetscScalar *vals; in MatDumpSPAI() local 31 PetscCall(MatGetRow(A, i, &nz, &cols, &vals)); in MatDumpSPAI() 32 … j++) fprintf(file, "%" PetscInt_FMT " %d" PetscInt_FMT " %16.14e\n", i + 1, cols[j] + 1, vals[j]); in MatDumpSPAI() 33 PetscCall(MatRestoreRow(A, i, &nz, &cols, &vals)); in MatDumpSPAI()
|