dolfinx.jit

Just-in-time (JIT) compilation using FFCx

Functions

ffcx_jit(ufl_object[, form_compiler_params, ...])

Compile UFL object with FFCx and CFFI.

get_parameters([priority_parameters])

Return a copy of the merged JIT parameter values for DOLFINx.

mpi_jit_decorator(local_jit, *args, **kwargs)

A decorator for jit compilation.

dolfinx.jit.ffcx_jit(ufl_object, form_compiler_params={}, jit_params={})[source]

Compile UFL object with FFCx and CFFI.

Parameters:
  • ufl_object – Object to compile, e.g. ufl.Form

  • form_compiler_params – Parameters used in FFCx compilation of this form. Run ffcx –help at the commandline to see all available options. Takes priority over all other parameter values.

  • jit_params – Parameters used in CFFI JIT compilation of C code generated by FFCx. See python/dolfinx/jit.py for all available parameters. Takes priority over all other parameter values.

Returns:

(compiled object, module, (header code, implementation code))

Notes

Priority ordering of parameters controlling DOLFINx JIT compilation from highest to lowest is:

  • jit_params (API)

  • $PWD/dolfinx_jit_parameters.json (local parameters)

  • $XDG_CONFIG_HOME/dolfinx/dolfinx_jit_parameters.json (user parameters)

  • DOLFINX_DEFAULT_JIT_PARAMETERS in dolfinx.jit

Priority ordering of parameters controlling FFCx from highest to lowest is:

  • form_compiler_paramss (API)

  • $PWD/ffcx_parameters.json (local parameters)

  • $XDG_CONFIG_HOME/ffcx/ffcx_parameters.json (user parameters)

  • FFCX_DEFAULT_PARAMETERS in ffcx.parameters

$XDG_CONFIG_HOME is ~/.config/ if the environment variable is not set.

The contents of the dolfinx_parameters.json files are cached on the first call. Subsequent calls to this function use this cache.

Example dolfinx_jit_parameters.json file:

{ “cffi_extra_compile_args”: [“-O2”, “-march=native” ], “cffi_verbose”: True }

dolfinx.jit.get_parameters(priority_parameters: Optional[dict] = None) dict[source]

Return a copy of the merged JIT parameter values for DOLFINx.

Parameters:

priority_parameters – take priority over all other parameter values (see notes)

Returns:

merged parameter values

Return type:

dict

Notes

See ffcx_jit for user facing documentation.