dolfinx_mpc.numba

dolfinx_mpc.numba#

Numba extension for dolfinx_mpc

dolfinx_mpc.numba.assemble_matrix(form: Form_float32 | Form_float64 | Form_complex128, constraint: MultiPointConstraint, bcs: List[DirichletBC_float32 | DirichletBC_float64 | DirichletBC_complex64 | DirichletBC_complex128] | None = None, diagval: float64 = 1.0, A: Mat | None = None)[source]#

Assembles a compiled DOLFINx form with given a multi point constraint and possible Dirichlet boundary conditions. NOTE: Strong Dirichlet conditions cannot be on master dofs.

Parameters:
  • form – The compiled bilinear form

  • constraint – The multi point constraint

  • bcs – List of Dirichlet boundary conditions

  • diagval – Value to set on the diagonal of the matrix

  • A – PETSc matrix to assemble into (optional)

dolfinx_mpc.numba.assemble_vector(form: Form_float32 | Form_float64 | Form_complex128, constraint: MultiPointConstraint, b: Vec | None = None) Vec[source]#

Assemble a compiled DOLFINx form into vector b.

Parameters:
  • form – The complied linear form

  • constraint – The multi point constraint

  • b – PETSc vector to assemble into (optional)