| /petsc/src/ksp/ksp/tutorials/ |
| H A D | bench_kspsolve.c | 121 PetscScalar *coo_v; in FillCOO() local
129 PetscCall(PetscCalloc3(user->nnz, &coo_i, user->nnz, &coo_j, user->nnz, &coo_v)); in FillCOO()
138 coo_v[count] = vcorn; in FillCOO()
144 coo_v[count] = vcorn; in FillCOO()
150 coo_v[count] = vcorn; in FillCOO()
156 coo_v[count] = vcorn; in FillCOO()
162 coo_v[count] = vcorn; in FillCOO()
168 coo_v[count] = vcorn; in FillCOO()
174 coo_v[count] = vcorn; in FillCOO()
180 coo_v[count] = vcorn; in FillCOO()
[all …]
|
| /petsc/src/snes/tutorials/ |
| H A D | ex55k.kokkos.cxx | 257 PetscScalarKokkosView coo_v("coo_v", user->ncoo); in FormJacobianLocalVec() local
268 coo_v(p + 2) = 2.0 * (hydhx + hxdhy); in FormJacobianLocalVec()
271 if (j - 1 != 0) coo_v(p + 0) = -hxdhy; in FormJacobianLocalVec()
272 if (i - 1 != 0) coo_v(p + 1) = -hydhx; in FormJacobianLocalVec()
274 coo_v(p + 2) = 2.0 * (hydhx + hxdhy) - sc * PetscExpScalar(xv(j, i)); in FormJacobianLocalVec()
276 if (i + 1 != mx - 1) coo_v(p + 3) = -hydhx; in FormJacobianLocalVec()
277 if (j + 1 != mx - 1) coo_v(p + 4) = -hxdhy; in FormJacobianLocalVec()
280 PetscCall(MatSetValuesCOO(jacpre, coo_v.data(), INSERT_VALUES)); in FormJacobianLocalVec()
|
| /petsc/src/mat/tests/ |
| H A D | ex123.c | 177 PetscScalar *coo_v; in main() local
190 PetscCall(PetscMalloc3(nnz, &coo_i, nnz, &coo_j, nnz, &coo_v)); in main()
196 coo_v[nnz] = vA[j]; in main()
203 coo_v[nnz] = vB[j]; in main()
213 PetscCall(MatSetValuesCOO(A, coo_v, ADD_VALUES)); in main()
217 PetscCall(MatSetValuesCOO(A, coo_v, INSERT_VALUES)); in main()
234 PetscCall(PetscFree3(coo_i, coo_j, coo_v)); in main()
|
| /petsc/src/mat/utils/ |
| H A D | gcreate.c | 613 PetscErrorCode MatSetValuesCOO_Basic(Mat A, const PetscScalar coo_v[], InsertMode imode) in MatSetValuesCOO_Basic() argument
631 …for (n = 0; n < n_i; n++) PetscCall(MatSetValue(A, coo_i[n], coo_j[n], coo_v ? coo_v[n] : zero, AD… in MatSetValuesCOO_Basic()
808 PetscErrorCode MatSetValuesCOO(Mat A, const PetscScalar coo_v[], InsertMode imode) in MatSetValuesCOO() argument
821 …PetscCall((*f)(A, coo_v, imode)); // all known COO implementations do not use MatStash. They do th… in MatSetValuesCOO()
825 …PetscCall(MatSetValuesCOO_Basic(A, coo_v, imode)); // fall back to MatSetValues, which might use M… in MatSetValuesCOO()
|
| /petsc/src/vec/vec/interface/ |
| H A D | vector.c | 280 PetscErrorCode VecSetValuesCOO(Vec x, const PetscScalar coo_v[], InsertMode imode) in VecSetValuesCOO() argument
288 PetscUseTypeMethod(x, setvaluescoo, coo_v, imode); in VecSetValuesCOO()
296 PetscCall(PetscGetMemType(coo_v, &mtype)); in VecSetValuesCOO()
303 PetscCall(VecSetValues(x, ncoo, coo_i, coo_v, ADD_VALUES)); in VecSetValuesCOO()
|
| /petsc/src/vec/vec/impls/seq/ |
| H A D | bvec2.c | 617 PetscErrorCode VecSetValuesCOO_Seq(Vec x, const PetscScalar coo_v[], InsertMode imode) in VecSetValuesCOO_Seq() argument
629 for (PetscCount j = jmap1[i]; j < jmap1[i + 1]; j++) sum += coo_v[perm1[j]]; in VecSetValuesCOO_Seq()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 6965 …PetscScalar *coo_v, *coo_w; /* store on-process and off-process COO scalars, and used as MPI recv/… member
6985 PetscCall(PetscSFFree(mmdata->sf, mmdata->mtype, mmdata->coo_v)); in MatProductCtxDestroy_MatMatMPIAIJBACKEND()
7058 PetscCall(MatSeqAIJCopySubArray(mmdata->mp[i], nown, mmdata->own[i], mmdata->coo_v + n_d)); in MatProductNumeric_MPIAIJBACKEND()
7064 PetscCall(MatSeqAIJCopySubArray(mmdata->mp[i], mm->nz, NULL, mmdata->coo_v + n_d)); in MatProductNumeric_MPIAIJBACKEND()
7069 PetscCall(PetscSFGatherBegin(mmdata->sf, MPIU_SCALAR, mmdata->coo_w, mmdata->coo_v + n_d)); in MatProductNumeric_MPIAIJBACKEND()
7070 PetscCall(PetscSFGatherEnd(mmdata->sf, MPIU_SCALAR, mmdata->coo_w, mmdata->coo_v + n_d)); in MatProductNumeric_MPIAIJBACKEND()
7072 PetscCall(MatSetValuesCOO(C, mmdata->coo_v, INSERT_VALUES)); in MatProductNumeric_MPIAIJBACKEND()
7567 …all(PetscSFMalloc(mmdata->sf, mmdata->mtype, ncoo * sizeof(PetscScalar), (void **)&mmdata->coo_v)); in MatProductSymbolic_MPIAIJBACKEND()
|
| /petsc/src/binding/petsc4py/src/petsc4py/PETSc/ |
| H A D | Mat.pyx | 2756 coo_v: Sequence[Scalar],
2764 coo_v
2773 matsetvalues_coo(self.mat, coo_v, addv)
|