Home
last modified time | relevance | path

Searched refs:w1 (Results 1 – 17 of 17) sorted by relevance

/petsc/src/mat/impls/dense/mpi/
H A Dmmdense.c58 PetscInt *nrow, *ncol, *w1, *w3, *w4, start, inrqr; in MatCreateSubMatrices_MPIDense_Local() local
103 PetscCall(PetscMalloc3(2 * size, &w1, size, &w3, size, &w4)); in MatCreateSubMatrices_MPIDense_Local()
104 PetscCall(PetscArrayzero(w1, size * 2)); /* initialize work vector*/ in MatCreateSubMatrices_MPIDense_Local()
117 w1[2 * j] += w4[j]; in MatCreateSubMatrices_MPIDense_Local()
125 w1[2 * rank] = 0; /* no mesg sent to self */ in MatCreateSubMatrices_MPIDense_Local()
128 if (w1[2 * i]) { in MatCreateSubMatrices_MPIDense_Local()
129 w1[2 * i + 1] = 1; in MatCreateSubMatrices_MPIDense_Local()
135 if (w1[2 * i]) { in MatCreateSubMatrices_MPIDense_Local()
144 w1[2 * j] += w1[2 * j + 1] + 2 * w3[j]; in MatCreateSubMatrices_MPIDense_Local()
145 msz += w1[2 * j]; in MatCreateSubMatrices_MPIDense_Local()
[all …]
/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
917 w1 = _mm256_set1_pd((t + bs * vi[k])[1]); in MatSolve_SeqBAIJ_9_NaturalOrdering()
919 s0 = _mm256_fnmadd_pd(a3, w1, s0); in MatSolve_SeqBAIJ_9_NaturalOrdering()
921 s1 = _mm256_fnmadd_pd(a4, w1, s1); in MatSolve_SeqBAIJ_9_NaturalOrdering()
923 s2 = _mm256_fnmadd_pd(a5, w1, s2); in MatSolve_SeqBAIJ_9_NaturalOrdering()
949 w1 = _mm256_set1_pd((t + bs * vi[k])[5]); in MatSolve_SeqBAIJ_9_NaturalOrdering()
951 s0 = _mm256_fnmadd_pd(a3, w1, s0); in MatSolve_SeqBAIJ_9_NaturalOrdering()
953 s1 = _mm256_fnmadd_pd(a4, w1, s1); in MatSolve_SeqBAIJ_9_NaturalOrdering()
955 s2 = _mm256_fnmadd_pd(a5, w1, s2); in MatSolve_SeqBAIJ_9_NaturalOrdering()
1009 w1 = _mm256_set1_pd((t + bs * vi[k])[1]); in MatSolve_SeqBAIJ_9_NaturalOrdering()
[all …]
H A Dbaij2.c680 __m256d w0, w1, w2, w3; in MatMult_SeqBAIJ_9_AVX2() local
733 w1 = _mm256_set1_pd(work[j * 9 + 1]); in MatMult_SeqBAIJ_9_AVX2()
735 z0 = _mm256_fmadd_pd(a0, w1, z0); in MatMult_SeqBAIJ_9_AVX2()
737 z1 = _mm256_fmadd_pd(a1, w1, z1); in MatMult_SeqBAIJ_9_AVX2()
739 z2 = _mm256_fmadd_pd(a2, w1, z2); in MatMult_SeqBAIJ_9_AVX2()
769 w1 = _mm256_set1_pd(work[j * 9 + 5]); in MatMult_SeqBAIJ_9_AVX2()
771 z0 = _mm256_fmadd_pd(a0, w1, z0); in MatMult_SeqBAIJ_9_AVX2()
773 z1 = _mm256_fmadd_pd(a1, w1, z1); in MatMult_SeqBAIJ_9_AVX2()
775 z2 = _mm256_fmadd_pd(a2, w1, z2); in MatMult_SeqBAIJ_9_AVX2()
1334 __m256d w0, w1, w2, w3; in MatMult_SeqBAIJ_12_AVX2() local
[all …]
/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()
102 w1[j] += w4[j]; in MatIncreaseOverlap_MPIBAIJ_Once()
110 w1[rank] = 0; /* no mesg sent to itself */ in MatIncreaseOverlap_MPIBAIJ_Once()
113 if (w1[i]) { in MatIncreaseOverlap_MPIBAIJ_Once()
121 if (w1[i]) { in MatIncreaseOverlap_MPIBAIJ_Once()
130 w1[j] += w2[j] + 2 * w3[j]; in MatIncreaseOverlap_MPIBAIJ_Once()
131 msz += w1[j]; in MatIncreaseOverlap_MPIBAIJ_Once()
135 PetscCall(PetscGatherNumberOfMessages(comm, w2, w1, &nrqr)); in MatIncreaseOverlap_MPIBAIJ_Once()
136 PetscCall(PetscGatherMessageLengths(comm, nrqs, nrqr, w1, &onodes1, &olengths1)); in MatIncreaseOverlap_MPIBAIJ_Once()
[all …]
/petsc/src/ksp/pc/impls/deflation/
H A Ddeflation.c387 Vec r, w1, w2; in PCPreSolve_Deflation() local
391 w1 = def->workcoarse[0]; in PCPreSolve_Deflation()
406 PetscCall(MatMult(def->Wt, r, w1)); /* w1 <- W'*r */ in PCPreSolve_Deflation()
408 PetscCall(MatMultHermitianTranspose(def->W, r, w1)); /* w1 <- W'*r */ in PCPreSolve_Deflation()
410 PetscCall(KSPSolve(def->WtAWinv, w1, w2)); /* w2 <- (W'*A*W)^{-1}*w1 */ in PCPreSolve_Deflation()
427 Vec u, w1, w2; in PCApply_Deflation() local
430 w1 = def->workcoarse[0]; in PCApply_Deflation()
437 PetscCall(MatMult(def->WtA, z, w1)); /* w1 <- W'*A*z */ 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()
/petsc/src/ts/impls/implicit/sundials/
H A Dsundials.h28 Vec w1, w2; /* work space vectors for function evaluation */ member
H A Dsundials.c20 Vec yy = cvode->w1, yydot = cvode->ydot; in TSPrecond_Sundials_Petsc()
54 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()
199 PetscCall(VecPlaceArray(cvode->w1, y_data)); in TSStep_Sundials()
200 PetscCall(VecCopy(cvode->w1, cvode->update)); in TSStep_Sundials()
201 PetscCall(VecResetArray(cvode->w1)); in TSStep_Sundials()
250 PetscCall(VecDestroy(&cvode->w1)); in TSReset_Sundials()
320 …cCreateMPIWithArray(PetscObjectComm((PetscObject)ts), 1, locsize, PETSC_DECIDE, NULL, &cvode->w1)); in TSSetUp_Sundials()
/petsc/src/mat/matfd/
H A Dfdmatrix.c13 PetscCall(VecCopy(F, fd->w1)); in MatFDColoringSetF()
455 PetscCall(MatCreateVecs(mat, NULL, &c->w1)); in MatFDColoringCreate()
457 PetscCall(VecBindToCPU(c->w1, PETSC_TRUE)); in MatFDColoringCreate()
458 PetscCall(VecDuplicate(c->w1, &c->w2)); in MatFDColoringCreate()
512 PetscCall(VecDestroy(&color->w1)); in MatFDColoringDestroy()
/petsc/src/mat/impls/aij/mpi/
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()
545 w1[j] += w4[j]; in MatIncreaseOverlap_MPIAIJ_Once()
553 w1[rank] = 0; /* no mesg sent to intself */ in MatIncreaseOverlap_MPIAIJ_Once()
556 if (w1[i]) { in MatIncreaseOverlap_MPIAIJ_Once()
564 if (w1[i]) { in MatIncreaseOverlap_MPIAIJ_Once()
573 w1[j] += w2[j] + 2 * w3[j]; in MatIncreaseOverlap_MPIAIJ_Once()
574 msz += w1[j]; in MatIncreaseOverlap_MPIAIJ_Once()
578 PetscCall(PetscGatherNumberOfMessages(comm, w2, w1, &nrqr)); in MatIncreaseOverlap_MPIAIJ_Once()
579 PetscCall(PetscGatherMessageLengths(comm, nrqs, nrqr, w1, &onodes1, &olengths1)); in MatIncreaseOverlap_MPIAIJ_Once()
[all …]
H A Dfdmpiaij.c39 Vec w1 = coloring->w1, w2 = coloring->w2, w3, vscale = coloring->vscale; in MatFDColoringApply_BAIJ() local
53 PetscCall((*f)(sctx, x1, w1, fctx)); in MatFDColoringApply_BAIJ()
134 PetscCall(VecAXPY(w2, -1.0, w1)); in MatFDColoringApply_BAIJ()
190 Vec w1 = coloring->w1, w2 = coloring->w2, w3, vscale = coloring->vscale; in MatFDColoringApply_AIJ() local
207 PetscCall((*f)(sctx, x1, w1, fctx)); in MatFDColoringApply_AIJ()
290 PetscCall(VecAXPY(w2, -1.0, w1)); in MatFDColoringApply_AIJ()
361 PetscCall(VecAXPY(w2, -1.0, w1)); in MatFDColoringApply_AIJ()
H A Dmpiaij.c7703 …ollapseRows(Mat Amat, PetscInt start, PetscInt bs, PetscInt *w0, PetscInt *w1, PetscInt *w2, Petsc… 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 … 0; Ii < Iend; Ii += bs, jj++) PetscCall(MatCollapseRows(Amat, Ii, bs, w0, w1, w2, &d_nnz[jj], NUL… 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 …Iend - Istart; Ii += bs, jj++) PetscCall(MatCollapseRows(Daij, Ii, bs, w0, w1, w2, &d_nnz[jj], NUL… in MatCreateGraph_Simple_AIJ()
7923 PetscCall(PetscFree3(w0, w1, w2)); in MatCreateGraph_Simple_AIJ()
/petsc/src/mat/impls/lrc/
H A Dlrc.c39 const PetscScalar *w1; in MatMult_LRC_kernel() local
56 PetscCall(VecGetArrayRead(Na->work1, &w1)); in MatMult_LRC_kernel()
59 …PetscCallMPI(MPIU_Allreduce(w1, w2, nwork, MPIU_SCALAR, MPIU_SUM, PetscObjectComm((PetscObject)N))… in MatMult_LRC_kernel()
60 PetscCall(VecRestoreArrayRead(Na->work1, &w1)); in MatMult_LRC_kernel()
/petsc/src/binding/petsc4py/test/
H A Dtest_vec.py213 w1, v1 = w[s], v[s]
215 self.assertEqual(w1, v1)
/petsc/src/ksp/pc/impls/fieldsplit/
H A Dfieldsplit.c43 Vec *x, *y, w1, w2; member
1053 PetscCall(VecDuplicate(ilink->x, &jac->w1)); in PCSetUp_FieldSplit()
1738 if (!jac->w1) { in PCApply_FieldSplit()
1739 PetscCall(VecDuplicate(x, &jac->w1)); in PCApply_FieldSplit()
1879 work1 = jac->w1; in PCApply_FieldSplit_GKB()
2020 if (!jac->w1) { in PCApplyTranspose_FieldSplit()
2021 PetscCall(VecDuplicate(x, &jac->w1)); in PCApplyTranspose_FieldSplit()
2029 PetscCall(MatMultTranspose(pc->mat, y, jac->w1)); in PCApplyTranspose_FieldSplit()
2030 PetscCall(VecWAXPY(jac->w2, -1.0, jac->w1, x)); in PCApplyTranspose_FieldSplit()
2035 PetscCall(MatMultTranspose(pc->mat, y, jac->w1)); in PCApplyTranspose_FieldSplit()
[all …]
/petsc/src/dm/dt/interface/
H A Ddt.c2006 PetscReal *p1, *w1; in PetscDTStroudConicalQuadrature() local
2031 PetscCall(PetscMalloc2(npoints, &p1, npoints, &w1)); in PetscDTStroudConicalQuadrature()
2037 PetscCall(PetscDTGaussJacobiQuadrature(npoints, -1., 1., i, 0.0, p1, w1)); in PetscDTStroudConicalQuadrature()
2043 for (PetscInt c = 0; c < Nc; c++) w[pt * Nc + c] *= mul * w1[j]; in PetscDTStroudConicalQuadrature()
2050 PetscCall(PetscFree2(p1, w1)); in PetscDTStroudConicalQuadrature()
2615 const PetscReal *x1, *w1, *x2, *w2; in PetscDTTensorQuadratureCreate() local
2629 PetscCall(PetscQuadratureGetData(q1, &dim1, &Nc1, &Np1, &x1, &w1)); 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.cas7925 (sst-sig-w1 2.)
7955 (bsl-sig-w1 2.)