Home
last modified time | relevance | path

Searched refs:lvec (Results 1 – 25 of 46) sorted by relevance

12

/petsc/src/ksp/pc/impls/gamg/
H A Dutil.c38 PetscCall(VecGetLocalSize(mpimat->lvec, &num_ghosts)); in PCGAMGGetDataWithGhosts()
56 … PetscCall(VecScatterBegin(mpimat->Mvctx, tmp_crds, mpimat->lvec, INSERT_VALUES, SCATTER_FORWARD)); in PCGAMGGetDataWithGhosts()
57 PetscCall(VecScatterEnd(mpimat->Mvctx, tmp_crds, mpimat->lvec, INSERT_VALUES, SCATTER_FORWARD)); in PCGAMGGetDataWithGhosts()
58 PetscCall(VecGetArray(mpimat->lvec, &data_arr)); in PCGAMGGetDataWithGhosts()
60 PetscCall(VecRestoreArray(mpimat->lvec, &data_arr)); in PCGAMGGetDataWithGhosts()
H A Dagg.c931 …PetscCall(VecScatterBegin(mpimat_1->Mvctx, tempVec, mpimat_1->lvec, INSERT_VALUES, SCATTER_FORWARD… in fixAggregatesWithSquare()
932 …PetscCall(VecScatterEnd(mpimat_1->Mvctx, tempVec, mpimat_1->lvec, INSERT_VALUES, SCATTER_FORWARD)); in fixAggregatesWithSquare()
933 PetscCall(VecGetArray(mpimat_1->lvec, &cpcol_1_state)); in fixAggregatesWithSquare()
935 …PetscCall(VecScatterBegin(mpimat_2->Mvctx, tempVec, mpimat_2->lvec, INSERT_VALUES, SCATTER_FORWARD… in fixAggregatesWithSquare()
936 …PetscCall(VecScatterEnd(mpimat_2->Mvctx, tempVec, mpimat_2->lvec, INSERT_VALUES, SCATTER_FORWARD)); in fixAggregatesWithSquare()
937 PetscCall(VecGetArray(mpimat_2->lvec, &cpcol_2_state)); in fixAggregatesWithSquare()
946 PetscCall(VecDuplicate(mpimat_2->lvec, &ghost_par_orig2)); in fixAggregatesWithSquare()
1050 PetscCall(VecGetSize(mpimat_2->lvec, &nghost_2)); in fixAggregatesWithSquare()
1057 PetscCall(VecDuplicate(mpimat_2->lvec, &ghostparents2)); in fixAggregatesWithSquare()
1070 PetscCall(VecDuplicate(mpimat_2->lvec, &ghostgids2)); in fixAggregatesWithSquare()
[all …]
/petsc/src/binding/petsc4py/demo/legacy/ode/
H A Dheat.py41 self.lvec = PETSc.Vec().create(comm=PETSc.COMM_SELF)
42 self.lvec.setSizes(self.n + 2)
43 self.lvec.setUp()
46 self.g2l = PETSc.Scatter().create(self.gvec, isg, self.lvec, None)
58 self.g2l.scatter(self.gvec, self.lvec, PETSc.InsertMode.INSERT)
62 self.lvec.view()
72 self.g2l.scatter(x, self.lvec, PETSc.InsertMode.INSERT) # lvec is a work vector
74 with self.lvec as u, xdot as udot:
/petsc/src/mat/tests/
H A Dex211.c14 Vec lvec = a->lvec, lcmap; in ISGetSeqIS_SameColDist_Private() local
31 PetscCall(VecDuplicate(lvec, &lcmap)); in ISGetSeqIS_SameColDist_Private()
78 PetscCall(VecScatterBegin(Mvctx, x, lvec, INSERT_VALUES, SCATTER_FORWARD)); in ISGetSeqIS_SameColDist_Private()
79 PetscCall(VecScatterEnd(Mvctx, x, lvec, INSERT_VALUES, SCATTER_FORWARD)); in ISGetSeqIS_SameColDist_Private()
91 PetscCall(VecGetArray(lvec, &xarray)); in ISGetSeqIS_SameColDist_Private()
101 PetscCall(VecRestoreArray(lvec, &xarray)); in ISGetSeqIS_SameColDist_Private()
105 PetscCall(VecView(lvec, 0)); in ISGetSeqIS_SameColDist_Private()
/petsc/src/mat/impls/aij/mpi/mpiviennacl/
H A Dmpiaijviennacl.cxx36 PetscCall(PetscObjectTypeCompare((PetscObject)b->lvec, VECSEQVIENNACL, &v)); in MatAssemblyEnd_MPIAIJViennaCL()
39 PetscCall(VecGetSize(b->lvec, &m)); in MatAssemblyEnd_MPIAIJViennaCL()
40 PetscCall(VecDestroy(&b->lvec)); in MatAssemblyEnd_MPIAIJViennaCL()
41 PetscCall(VecCreateSeqViennaCL(PETSC_COMM_SELF, m, &b->lvec)); in MatAssemblyEnd_MPIAIJViennaCL()
/petsc/src/mat/impls/aij/mpi/mpihipsparse/
H A Dmpiaijhipsparse.hip.cxx55 PetscCall(VecDestroy(&mpiaij->lvec)); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
287 PetscCall(VecDestroy(&b->lvec)); in MatMPIAIJSetPreallocation_MPIAIJHIPSPARSE()
320 PetscCall(VecScatterBegin(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMult_MPIAIJHIPSPARSE()
322 PetscCall(VecScatterEnd(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMult_MPIAIJHIPSPARSE()
323 PetscCall((*a->B->ops->multadd)(a->B, a->lvec, yy, yy)); in MatMult_MPIAIJHIPSPARSE()
342 PetscCall(VecScatterBegin(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMultAdd_MPIAIJHIPSPARSE()
344 PetscCall(VecScatterEnd(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMultAdd_MPIAIJHIPSPARSE()
345 PetscCall((*a->B->ops->multadd)(a->B, a->lvec, zz, zz)); in MatMultAdd_MPIAIJHIPSPARSE()
354 PetscCall((*a->B->ops->multtranspose)(a->B, xx, a->lvec)); in MatMultTranspose_MPIAIJHIPSPARSE()
356 PetscCall(VecScatterBegin(a->Mvctx, a->lvec, yy, ADD_VALUES, SCATTER_REVERSE)); in MatMultTranspose_MPIAIJHIPSPARSE()
[all …]
/petsc/src/mat/impls/aij/mpi/mpicusparse/
H A Dmpiaijcusparse.cu55 PetscCall(VecDestroy(&mpiaij->lvec)); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
287 PetscCall(VecDestroy(&b->lvec)); in MatMPIAIJSetPreallocation_MPIAIJCUSPARSE()
320 PetscCall(VecScatterBegin(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMult_MPIAIJCUSPARSE()
322 PetscCall(VecScatterEnd(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMult_MPIAIJCUSPARSE()
323 PetscCall((*a->B->ops->multadd)(a->B, a->lvec, yy, yy)); in MatMult_MPIAIJCUSPARSE()
342 PetscCall(VecScatterBegin(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMultAdd_MPIAIJCUSPARSE()
344 PetscCall(VecScatterEnd(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMultAdd_MPIAIJCUSPARSE()
345 PetscCall((*a->B->ops->multadd)(a->B, a->lvec, zz, zz)); in MatMultAdd_MPIAIJCUSPARSE()
354 PetscCall((*a->B->ops->multtranspose)(a->B, xx, a->lvec)); in MatMultTranspose_MPIAIJCUSPARSE()
356 PetscCall(VecScatterBegin(a->Mvctx, a->lvec, yy, ADD_VALUES, SCATTER_REVERSE)); in MatMultTranspose_MPIAIJCUSPARSE()
[all …]
/petsc/src/binding/petsc4py/test/
H A Dtest_dmshell.py43 lvec = self.dm.createLocalVector()
44 self.assertEqual(vec.getSizes(), lvec.getSizes())
45 lsize, gsize = lvec.getSizes()
47 self.assertEqual(lvec.comm, PETSc.COMM_SELF)
57 lvec = self.dm.createLocalVector()
58 lsize, gsize = lvec.getSizes()
61 self.assertEqual(lvec.comm, PETSc.COMM_SELF)
H A Dtest_vec.py176 lvec = self.vec.createLocalVector()
177 self.vec.getLocalVector(lvec)
178 self.assertEqual(abs(lvec.sum()), (rank + 1) * ln)
179 self.vec.restoreLocalVector(lvec)
180 self.vec.getLocalVector(lvec, readonly=True)
181 self.assertEqual(abs(lvec.sum()), (rank + 1) * ln)
182 self.vec.restoreLocalVector(lvec, readonly=True)
183 lvec.destroy()
/petsc/src/mat/impls/sell/mpi/
H A Dmpisell.c374 PetscCall(VecScatterBegin(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMult_MPISELL()
376 PetscCall(VecScatterEnd(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMult_MPISELL()
377 PetscCall((*a->B->ops->multadd)(a->B, a->lvec, yy, yy)); in MatMult_MPISELL()
395 PetscCall(VecScatterBegin(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMultAdd_MPISELL()
397 PetscCall(VecScatterEnd(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMultAdd_MPISELL()
398 PetscCall((*a->B->ops->multadd)(a->B, a->lvec, zz, zz)); in MatMultAdd_MPISELL()
408 PetscCall((*a->B->ops->multtranspose)(a->B, xx, a->lvec)); in MatMultTranspose_MPISELL()
412 PetscCall(VecScatterBegin(a->Mvctx, a->lvec, yy, ADD_VALUES, SCATTER_REVERSE)); in MatMultTranspose_MPISELL()
413 PetscCall(VecScatterEnd(a->Mvctx, a->lvec, yy, ADD_VALUES, SCATTER_REVERSE)); in MatMultTranspose_MPISELL()
463 PetscCall((*a->B->ops->multtranspose)(a->B, xx, a->lvec)); in MatMultTransposeAdd_MPISELL()
[all …]
H A Dmpisell.h26 Vec lvec; /* local vector */ member
H A Dmmsell.c26 PetscCall(VecDestroy(&sell->lvec)); in MatDisAssemble_MPISELL()
175 PetscCall(VecCreateSeq(PETSC_COMM_SELF, ec, &sell->lvec)); in MatSetUpMultiply_MPISELL()
185 PetscCall(VecScatterCreate(gvec, from, sell->lvec, to, &sell->Mvctx)); in MatSetUpMultiply_MPISELL()
/petsc/src/mat/impls/aij/seq/bas/
H A Dspbas_cholesky.h250 PetscScalar *lvec; in spbas_incomplete_cholesky() local
291 PetscCall(PetscCalloc1(nrows, &lvec)); in spbas_incomplete_cholesky()
326 lvec[k] = diag[k] * r_val[j]; in spbas_incomplete_cholesky()
327 val[i] -= r_val[j] * lvec[k]; in spbas_incomplete_cholesky()
364 for (jL = 0; jL < retval.row_nnz[k]; jL++) val[k] -= r1_val[jL] * lvec[r1_icol[jL]]; in spbas_incomplete_cholesky()
387 for (j = 0; j < r_nnz; j++) lvec[r_icol[j]] = 0; in spbas_incomplete_cholesky()
390 PetscCall(PetscFree(lvec)); in spbas_incomplete_cholesky()
/petsc/src/mat/impls/sell/mpi/mpicuda/
H A Dmpisellcuda.cu41 …d && mode == MAT_FINAL_ASSEMBLY) PetscCall(VecSetType(((Mat_MPISELL *)A->data)->lvec, VECSEQCUDA)); in MatAssemblyEnd_MPISELLCUDA()
70 if (a->lvec) PetscCall(VecSetType(a->lvec, VECSEQCUDA)); in MatConvert_MPISELL_MPISELLCUDA()
/petsc/src/mat/impls/sell/mpi/mpihip/
H A Dmpisellhip.hip.cxx41 …ed && mode == MAT_FINAL_ASSEMBLY) PetscCall(VecSetType(((Mat_MPISELL *)A->data)->lvec, VECSEQHIP)); in MatAssemblyEnd_MPISELLHIP()
70 if (a->lvec) PetscCall(VecSetType(a->lvec, VECSEQHIP)); in MatConvert_MPISELL_MPISELLHIP()
/petsc/src/binding/petsc4py/src/petsc4py/PETSc/
H A Dpetscdmshell.pxi119 cdef Vec lvec = ref_Vec(l)
125 begin(Dm, gvec, mode, lvec, *args, **kargs)
135 cdef Vec lvec = ref_Vec(l)
141 end(Dm, gvec, mode, lvec, *args, **kargs)
151 cdef Vec lvec = ref_Vec(l)
157 begin(Dm, gvec, mode, lvec, *args, **kargs)
167 cdef Vec lvec = ref_Vec(l)
173 end(Dm, gvec, mode, lvec, *args, **kargs)
183 cdef Vec lvec = ref_Vec(l)
189 begin(Dm, gvec, mode, lvec, *args, **kargs)
[all …]
/petsc/src/dm/impls/redundant/
H A Ddmredundant.c72 static PetscErrorCode DMCreateLocalVector_Redundant(DM dm, Vec *lvec) in DMCreateLocalVector_Redundant() argument
78 PetscAssertPointer(lvec, 2); in DMCreateLocalVector_Redundant()
79 *lvec = NULL; in DMCreateLocalVector_Redundant()
80 PetscCall(VecCreate(PETSC_COMM_SELF, lvec)); in DMCreateLocalVector_Redundant()
81 PetscCall(VecSetSizes(*lvec, red->N, red->N)); in DMCreateLocalVector_Redundant()
82 PetscCall(VecSetType(*lvec, dm->vectype)); in DMCreateLocalVector_Redundant()
83 PetscCall(VecSetDM(*lvec, dm)); in DMCreateLocalVector_Redundant()
/petsc/src/mat/impls/aij/mpi/
H A Dmmaij.c99 if (!aij->lvec) { in MatSetUpMultiply_MPIAIJ()
101 PetscCall(MatCreateVecs(aij->B, &aij->lvec, NULL)); in MatSetUpMultiply_MPIAIJ()
103 PetscCall(VecGetSize(aij->lvec, &ec)); in MatSetUpMultiply_MPIAIJ()
115 PetscCall(VecScatterCreate(gvec, from, aij->lvec, to, &aij->Mvctx)); in MatSetUpMultiply_MPIAIJ()
140 PetscCall(VecDestroy(&aij->lvec)); in MatDisAssemble_MPIAIJ()
/petsc/src/dm/impls/composite/
H A Dpack.c837 static PetscErrorCode DMCreateLocalVector_Composite(DM dm, Vec *lvec) in DMCreateLocalVector_Composite() argument
847 PetscCall(VecCreate(PETSC_COMM_SELF, lvec)); in DMCreateLocalVector_Composite()
848 PetscCall(VecSetType(*lvec, dm->vectype)); in DMCreateLocalVector_Composite()
849 PetscCall(VecSetSizes(*lvec, com->nghost, PETSC_DECIDE)); in DMCreateLocalVector_Composite()
850 PetscCall(VecSetDM(*lvec, dm)); in DMCreateLocalVector_Composite()
1523 static PetscErrorCode DMGlobalToLocalBegin_Composite(DM dm, Vec gvec, InsertMode mode, Vec lvec) in DMGlobalToLocalBegin_Composite() argument
1538 PetscCall(VecGetArray(lvec, &larray)); in DMGlobalToLocalBegin_Composite()
1564 PetscCall(VecRestoreArray(lvec, &larray)); in DMGlobalToLocalBegin_Composite()
1568 static PetscErrorCode DMGlobalToLocalEnd_Composite(DM dm, Vec gvec, InsertMode mode, Vec lvec) in DMGlobalToLocalEnd_Composite() argument
1573 PetscValidHeaderSpecific(lvec, VEC_CLASSID, 4); in DMGlobalToLocalEnd_Composite()
[all …]
/petsc/src/ksp/ksp/tutorials/
H A Dex59.c441 Vec lvec, rvec; in ComputeSubdomainMatrix() local
443 PetscCall(MatCreateVecs(temp_local_mat, &lvec, &rvec)); in ComputeSubdomainMatrix()
444 PetscCall(VecSet(lvec, 1.0)); in ComputeSubdomainMatrix()
445 PetscCall(MatMult(temp_local_mat, lvec, rvec)); in ComputeSubdomainMatrix()
447 PetscCall(VecDestroy(&lvec)); in ComputeSubdomainMatrix()
622 Vec lvec, rvec; in GLLStuffs() local
624 PetscCall(MatCreateVecs(glldata->elem_mat, &lvec, &rvec)); in GLLStuffs()
625 PetscCall(VecSet(lvec, 1.0)); in GLLStuffs()
626 PetscCall(MatMult(glldata->elem_mat, lvec, rvec)); in GLLStuffs()
628 PetscCall(VecDestroy(&lvec)); in GLLStuffs()
[all …]
/petsc/src/mat/graphops/coarsen/impls/hem/
H A Dhem.c499 PetscCall(VecGetLocalSize(mpimat->lvec, &num_ghosts)); in MatCoarsenApply_HEM_private()
508 PetscCall(VecScatterBegin(mpimat->Mvctx, vec, mpimat->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatCoarsenApply_HEM_private()
509 PetscCall(VecScatterEnd(mpimat->Mvctx, vec, mpimat->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatCoarsenApply_HEM_private()
510 PetscCall(VecGetArrayRead(mpimat->lvec, &buf)); /* get proc ID in 'buf' */ in MatCoarsenApply_HEM_private()
514 PetscCall(VecRestoreArrayRead(mpimat->lvec, &buf)); in MatCoarsenApply_HEM_private()
520 PetscCall(VecScatterBegin(mpimat->Mvctx, vec, mpimat->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatCoarsenApply_HEM_private()
521 PetscCall(VecScatterEnd(mpimat->Mvctx, vec, mpimat->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatCoarsenApply_HEM_private()
522 …PetscCall(VecGetArrayRead(mpimat->lvec, &buf)); … in MatCoarsenApply_HEM_private()
524 PetscCall(VecRestoreArrayRead(mpimat->lvec, &buf)); in MatCoarsenApply_HEM_private()
533 PetscCall(VecScatterBegin(mpimat->Mvctx, vec, mpimat->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatCoarsenApply_HEM_private()
[all …]
/petsc/src/mat/impls/dense/mpi/
H A Dmpidense.c483 PetscCall(VecGetArrayWriteAndMemType(mdn->lvec, &ay, &aymtype)); in MatMultColumnRange_MPIDense()
486 PetscCall(VecRestoreArrayWriteAndMemType(mdn->lvec, &ay)); in MatMultColumnRange_MPIDense()
488 …MatMultColumnRange_C", (Mat, Vec, Vec, PetscInt, PetscInt), (mdn->A, mdn->lvec, yy, c_start, c_end… in MatMultColumnRange_MPIDense()
502 PetscCall(VecGetArrayWriteAndMemType(mdn->lvec, &ay, &aymtype)); in MatMult_MPIDense()
505 PetscCall(VecRestoreArrayWriteAndMemType(mdn->lvec, &ay)); in MatMult_MPIDense()
507 PetscCall((*mdn->A->ops->mult)(mdn->A, mdn->lvec, yy)); in MatMult_MPIDense()
521 PetscCall(VecGetArrayAndMemType(mdn->lvec, &ay, &aymtype)); in MatMultAddColumnRange_MPIDense()
524 PetscCall(VecRestoreArrayAndMemType(mdn->lvec, &ay)); in MatMultAddColumnRange_MPIDense()
526 …ddColumnRange_C", (Mat, Vec, Vec, Vec, PetscInt, PetscInt), (mdn->A, mdn->lvec, yy, zz, c_start, c… in MatMultAddColumnRange_MPIDense()
540 PetscCall(VecGetArrayAndMemType(mdn->lvec, &ay, &aymtype)); in MatMultAdd_MPIDense()
[all …]
/petsc/src/mat/impls/baij/mpi/
H A Dmpibaij.c23 PetscCall(VecDestroy(&baij->lvec)); in MatDestroy_MPIBAIJ()
1228 PetscCall(VecScatterBegin(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMult_MPIBAIJ()
1230 PetscCall(VecScatterEnd(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMult_MPIBAIJ()
1231 PetscCall((*a->B->ops->multadd)(a->B, a->lvec, yy, yy)); in MatMult_MPIBAIJ()
1240 PetscCall(VecScatterBegin(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMultAdd_MPIBAIJ()
1242 PetscCall(VecScatterEnd(a->Mvctx, xx, a->lvec, INSERT_VALUES, SCATTER_FORWARD)); in MatMultAdd_MPIBAIJ()
1243 PetscCall((*a->B->ops->multadd)(a->B, a->lvec, zz, zz)); in MatMultAdd_MPIBAIJ()
1253 PetscCall((*a->B->ops->multtranspose)(a->B, xx, a->lvec)); in MatMultTranspose_MPIBAIJ()
1257 PetscCall(VecScatterBegin(a->Mvctx, a->lvec, yy, ADD_VALUES, SCATTER_REVERSE)); in MatMultTranspose_MPIBAIJ()
1258 PetscCall(VecScatterEnd(a->Mvctx, a->lvec, yy, ADD_VALUES, SCATTER_REVERSE)); in MatMultTranspose_MPIBAIJ()
[all …]
/petsc/src/dm/impls/plex/
H A Dplexdd.c257 Vec gvec, svec, lvec; in DMCreateDomainDecompositionScatters_Plex() local
268 PetscCall(DMGetLocalVector(subdms[i], &lvec)); in DMCreateDomainDecompositionScatters_Plex()
281 if (lscat) PetscCall(VecScatterCreate(gvec, gl_is, lvec, ll_is, &(*lscat)[i])); in DMCreateDomainDecompositionScatters_Plex()
283 PetscCall(DMRestoreLocalVector(subdms[i], &lvec)); in DMCreateDomainDecompositionScatters_Plex()
/petsc/src/mat/impls/sbaij/mpi/
H A Dmmsbaij.c107 PetscCall(VecCreateSeq(PETSC_COMM_SELF, ec * bs, &sbaij->lvec)); in MatSetUpMultiply_MPISBAIJ()
118 PetscCall(VecScatterCreate(gvec, from, sbaij->lvec, to, &sbaij->Mvctx)); in MatSetUpMultiply_MPISBAIJ()
199 PetscCall(VecDestroy(&baij->lvec)); in MatDisAssemble_MPISBAIJ()

12