Home
last modified time | relevance | path

Searched refs:w2 (Results 1 – 18 of 18) sorted by relevance

/petsc/src/mat/impls/baij/seq/
H A Dbaijfact9.c886 __m256d a0, a1, a2, a3, a4, a5, w0, w1, w2, w3, s0, s1, s2, v0, v1, v2, v3; in MatSolve_SeqBAIJ_9_NaturalOrdering() local
925 w2 = _mm256_set1_pd((t + bs * vi[k])[2]); in MatSolve_SeqBAIJ_9_NaturalOrdering()
927 s0 = _mm256_fnmadd_pd(a0, w2, s0); in MatSolve_SeqBAIJ_9_NaturalOrdering()
929 s1 = _mm256_fnmadd_pd(a1, w2, s1); in MatSolve_SeqBAIJ_9_NaturalOrdering()
931 s2 = _mm256_fnmadd_pd(a2, w2, s2); in MatSolve_SeqBAIJ_9_NaturalOrdering()
957 w2 = _mm256_set1_pd((t + bs * vi[k])[6]); in MatSolve_SeqBAIJ_9_NaturalOrdering()
959 s0 = _mm256_fnmadd_pd(a0, w2, s0); in MatSolve_SeqBAIJ_9_NaturalOrdering()
961 s1 = _mm256_fnmadd_pd(a1, w2, s1); in MatSolve_SeqBAIJ_9_NaturalOrdering()
963 s2 = _mm256_fnmadd_pd(a2, w2, s2); in MatSolve_SeqBAIJ_9_NaturalOrdering()
1018 w2 = _mm256_set1_pd((t + bs * vi[k])[2]); in MatSolve_SeqBAIJ_9_NaturalOrdering()
[all …]
H A Dbaij2.c680 __m256d w0, w1, w2, w3; in MatMult_SeqBAIJ_9_AVX2() local
742 w2 = _mm256_set1_pd(work[j * 9 + 2]); in MatMult_SeqBAIJ_9_AVX2()
744 z0 = _mm256_fmadd_pd(a3, w2, z0); in MatMult_SeqBAIJ_9_AVX2()
746 z1 = _mm256_fmadd_pd(a4, w2, z1); in MatMult_SeqBAIJ_9_AVX2()
748 z2 = _mm256_fmadd_pd(a5, w2, z2); in MatMult_SeqBAIJ_9_AVX2()
778 w2 = _mm256_set1_pd(work[j * 9 + 6]); in MatMult_SeqBAIJ_9_AVX2()
780 z0 = _mm256_fmadd_pd(a0, w2, z0); in MatMult_SeqBAIJ_9_AVX2()
782 z1 = _mm256_fmadd_pd(a1, w2, z1); in MatMult_SeqBAIJ_9_AVX2()
784 z2 = _mm256_fmadd_pd(a2, w2, z2); in MatMult_SeqBAIJ_9_AVX2()
1334 __m256d w0, w1, w2, w3; in MatMult_SeqBAIJ_12_AVX2() local
[all …]
/petsc/src/dm/dt/tests/
H A Dex1.c111 PetscReal *x2, *w2; in CheckJacobiQuadrature() local
116 PetscCall(PetscMalloc2(npoints, &x2, npoints, &w2)); in CheckJacobiQuadrature()
117 PetscCall((*func)(npoints, -1., 1., alpha, beta, x2, w2)); in CheckJacobiQuadrature()
118 PetscCall(CheckQuadrature_Basics(npoints, alpha, beta, x2, w2)); in CheckJacobiQuadrature()
119 PetscCall(CheckQuadrature(npoints, alpha, beta, x2, w2, nexact)); in CheckJacobiQuadrature()
124 wdiff = PetscAbsReal(w[i] - w2[i]); in CheckJacobiQuadrature()
131 PetscCall(PetscFree2(x2, w2)); in CheckJacobiQuadrature()
/petsc/src/mat/impls/aij/mpi/
H A Dfdmpiaij.c39 Vec w1 = coloring->w1, w2 = coloring->w2, w3, vscale = coloring->vscale; in MatFDColoringApply_BAIJ() local
131 PetscCall(VecPlaceArray(w2, dy_i)); /* place w2 to the array dy_i */ in MatFDColoringApply_BAIJ()
132 PetscCall((*f)(sctx, w3, w2, fctx)); in MatFDColoringApply_BAIJ()
134 PetscCall(VecAXPY(w2, -1.0, w1)); in MatFDColoringApply_BAIJ()
135 PetscCall(VecResetArray(w2)); in MatFDColoringApply_BAIJ()
190 Vec w1 = coloring->w1, w2 = coloring->w2, w3, vscale = coloring->vscale; in MatFDColoringApply_AIJ() local
287 PetscCall(VecPlaceArray(w2, dy_k)); /* place w2 to the array dy_i */ in MatFDColoringApply_AIJ()
288 PetscCall((*f)(sctx, w3, w2, fctx)); in MatFDColoringApply_AIJ()
290 PetscCall(VecAXPY(w2, -1.0, w1)); in MatFDColoringApply_AIJ()
291 PetscCall(VecResetArray(w2)); in MatFDColoringApply_AIJ()
[all …]
H A Dmpiov.c495 PetscMPIInt *w1, *w2, nrqr, *w3, *w4, *onodes1, *olengths1, *onodes2, *olengths2; in MatIncreaseOverlap_MPIAIJ_Once() local
532 PetscCall(PetscCalloc4(size, &w1, size, &w2, size, &w3, size, &w4)); in MatIncreaseOverlap_MPIAIJ_Once()
557 w2[i] = 1; in MatIncreaseOverlap_MPIAIJ_Once()
573 w1[j] += w2[j] + 2 * w3[j]; in MatIncreaseOverlap_MPIAIJ_Once()
578 PetscCall(PetscGatherNumberOfMessages(comm, w2, w1, &nrqr)); in MatIncreaseOverlap_MPIAIJ_Once()
705 PetscCall(PetscFree4(w1, w2, w3, w4)); in MatIncreaseOverlap_MPIAIJ_Once()
1222 …PetscMPIInt rank, size, *req_source1, *req_source2, tag1, tag2, tag3, tag4, *w1, *w2, nrqr, nr… in MatCreateSubMatrices_MPIAIJ_SingleIS_Local() local
1278 PetscCall(PetscCalloc2(size, &w1, size, &w2)); in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1291 if (proc != rank && !w2[proc]) { in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1292 w2[proc] = 1; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
[all …]
H A Dmpiaij.c7703 …t Amat, PetscInt start, PetscInt bs, PetscInt *w0, PetscInt *w1, PetscInt *w2, PetscInt *ncollapse… in MatCollapseRows() argument
7705 PetscInt i, nprev, *cprev = w0, ncur = 0, *ccur = w1, *merged = w2, *cprevtmp; in MatCollapseRows()
7892 PetscInt *d_nnz, *o_nnz, *w0, *w1, *w2; in MatCreateGraph_Simple_AIJ() local
7907 PetscCall(PetscMalloc3(max_d_nnz, &w0, max_d_nnz, &w1, max_d_nnz, &w2)); in MatCreateGraph_Simple_AIJ()
7908 … Ii < Iend; Ii += bs, jj++) PetscCall(MatCollapseRows(Amat, Ii, bs, w0, w1, w2, &d_nnz[jj], NULL)); in MatCreateGraph_Simple_AIJ()
7909 PetscCall(PetscFree3(w0, w1, w2)); in MatCreateGraph_Simple_AIJ()
7921 PetscCall(PetscMalloc3(max_d_nnz, &w0, max_d_nnz, &w1, max_d_nnz, &w2)); in MatCreateGraph_Simple_AIJ()
7922 …d - Istart; Ii += bs, jj++) PetscCall(MatCollapseRows(Daij, Ii, bs, w0, w1, w2, &d_nnz[jj], NULL)); in MatCreateGraph_Simple_AIJ()
7923 PetscCall(PetscFree3(w0, w1, w2)); in MatCreateGraph_Simple_AIJ()
/petsc/src/ksp/pc/impls/deflation/
H A Ddeflation.c387 Vec r, w1, w2; in PCPreSolve_Deflation() local
392 w2 = def->workcoarse[1]; in PCPreSolve_Deflation()
410 PetscCall(KSPSolve(def->WtAWinv, w1, w2)); /* w2 <- (W'*A*W)^{-1}*w1 */ in PCPreSolve_Deflation()
411 PetscCall(MatMult(def->W, w2, r)); /* r <- W*w2 */ in PCPreSolve_Deflation()
427 Vec u, w1, w2; in PCApply_Deflation() local
431 w2 = def->workcoarse[1]; in PCApply_Deflation()
440 PetscCall(MatMult(def->Wt, r, w2)); /* w2 <- W'*r */ in PCApply_Deflation()
442 PetscCall(MatMultHermitianTranspose(def->W, r, w2)); /* w2 <- W'*r */ in PCApply_Deflation()
444 PetscCall(VecAXPY(w1, -1.0 * def->correctfact, w2)); /* w1 <- w1 - l*w2 */ in PCApply_Deflation()
446 PetscCall(KSPSolve(def->WtAWinv, w1, w2)); /* w2 <- (W'*A*W)^{-1}*w1 */ in PCApply_Deflation()
[all …]
/petsc/src/ts/impls/implicit/sundials/
H A Dsundials.h28 Vec w1, w2; /* work space vectors for function evaluation */ member
H A Dsundials.c54 Vec rr = cvode->w1, zz = cvode->w2; in TSPSolve_Sundials_Petsc()
89 Vec yy = cvode->w1, yyd = cvode->w2, yydot = cvode->ydot; in TSFunction_Sundials()
251 PetscCall(VecDestroy(&cvode->w2)); in TSReset_Sundials()
321 …cCreateMPIWithArray(PetscObjectComm((PetscObject)ts), 1, locsize, PETSC_DECIDE, NULL, &cvode->w2)); in TSSetUp_Sundials()
/petsc/src/mat/impls/lrc/
H A Dlrc.c40 PetscScalar *w2; in MatMult_LRC_kernel() local
57 PetscCall(VecGetArrayWrite(Na->work2, &w2)); in MatMult_LRC_kernel()
59 …PetscCallMPI(MPIU_Allreduce(w1, w2, nwork, MPIU_SCALAR, MPIU_SUM, PetscObjectComm((PetscObject)N))… in MatMult_LRC_kernel()
61 PetscCall(VecRestoreArrayWrite(Na->work2, &w2)); in MatMult_LRC_kernel()
/petsc/src/mat/impls/baij/mpi/
H A Dbaijov.c62 PetscMPIInt size, rank, tag1, tag2, *w2, *w1, nrqs, nrqr, *pa; in MatIncreaseOverlap_MPIBAIJ_Once() local
89 PetscCall(PetscCalloc4(size, &w1, size, &w2, size, &w3, size, &w4)); in MatIncreaseOverlap_MPIBAIJ_Once()
114 w2[i] = 1; in MatIncreaseOverlap_MPIBAIJ_Once()
130 w1[j] += w2[j] + 2 * w3[j]; in MatIncreaseOverlap_MPIBAIJ_Once()
135 PetscCall(PetscGatherNumberOfMessages(comm, w2, w1, &nrqr)); in MatIncreaseOverlap_MPIBAIJ_Once()
236 PetscCall(PetscFree4(w1, w2, w3, w4)); in MatIncreaseOverlap_MPIBAIJ_Once()
621 …PetscMPIInt rank, size, tag0, tag2, tag3, tag4, *w1, *w2, *w3, *w4, nrqr, nrqs = 0, *req_sour… local
755 …PetscCall(PetscCalloc4(size, &w1, size, &w2, size, &w3, size, &w4)); /* mesg size, initialize work…
789 w2[i] = 1;
801 w1[j] += w2[j] + 2 * w3[j];
[all …]
/petsc/src/mat/matfd/
H A Dfdmatrix.c458 PetscCall(VecDuplicate(c->w1, &c->w2)); in MatFDColoringCreate()
460 PetscCall(VecBindToCPU(c->w2, PETSC_TRUE)); in MatFDColoringCreate()
513 PetscCall(VecDestroy(&color->w2)); in MatFDColoringDestroy()
/petsc/src/binding/petsc4py/test/
H A Dtest_vec.py214 w2, v2 = w[e - 1], v[e - 1]
216 self.assertEqual(w2, v2)
/petsc/src/ksp/pc/impls/fieldsplit/
H A Dfieldsplit.c43 Vec *x, *y, w1, w2; member
1047 PetscCall(VecDuplicate(ilink->x, &jac->w2)); in PCSetUp_FieldSplit()
1740 PetscCall(VecDuplicate(x, &jac->w2)); in PCApply_FieldSplit()
1880 work2 = jac->w2; in PCApply_FieldSplit_GKB()
2022 PetscCall(VecDuplicate(x, &jac->w2)); in PCApplyTranspose_FieldSplit()
2030 PetscCall(VecWAXPY(jac->w2, -1.0, jac->w1, x)); in PCApplyTranspose_FieldSplit()
2031 PetscCall(FieldSplitSplitSolveAddTranspose(ilink, jac->w2, y)); in PCApplyTranspose_FieldSplit()
2036 PetscCall(VecWAXPY(jac->w2, -1.0, jac->w1, x)); in PCApplyTranspose_FieldSplit()
2037 PetscCall(FieldSplitSplitSolveAddTranspose(ilink, jac->w2, y)); in PCApplyTranspose_FieldSplit()
2047 PetscCall(VecWAXPY(jac->w2, -1.0, jac->w1, x)); in PCApplyTranspose_FieldSplit()
[all …]
/petsc/src/ts/tutorials/
H A Dex30.c1394 Vec w2, wg; in DMGetLumpedMass() local
1398 PetscCall(DMGetLocalVector(dm, &w2)); in DMGetLumpedMass()
1399 PetscCall(VecSet(w2, 0.0)); in DMGetLumpedMass()
1402 PetscCall(DMGlobalToLocal(dm, wg, INSERT_VALUES, w2)); in DMGetLumpedMass()
1403 PetscCall(VecPointwiseMult(w, w, w2)); in DMGetLumpedMass()
1405 PetscCall(DMRestoreLocalVector(dm, &w2)); in DMGetLumpedMass()
/petsc/src/dm/dt/interface/
H A Ddt.c2615 const PetscReal *x1, *w1, *x2, *w2; in PetscDTTensorQuadratureCreate() local
2631 PetscCall(PetscQuadratureGetData(q2, &dim2, &Nc2, &Np2, &x2, &w2)); in PetscDTTensorQuadratureCreate()
2775 w[qc] = w1[qa] * w2[qb]; in PetscDTTensorQuadratureCreate()
/petsc/include/petsc/private/
H A Dmatimpl.h661 Vec w1, w2, w3; /* work vectors used in computing Jacobian */ member
/petsc/share/petsc/datafiles/meshes/
H A Dtestcase3D.cas7924 (sst-sig-w2 1.168)
7954 (bsl-sig-w2 1.168)