cudolfinx.jit

Routines for manipulating generated FFCX code.

Functions

get_tabulate_tensor_sources(form)

Given a compiled fem.Form, extract the C source code of the tabulate tensors.

get_wrapped_tabulate_tensors(form[, backend])

Given a fem.Form, wrap the tabulate tensors for use on a GPU.

cudolfinx.jit.get_tabulate_tensor_sources(form: Form)[source]

Given a compiled fem.Form, extract the C source code of the tabulate tensors.

cudolfinx.jit.get_wrapped_tabulate_tensors(form: Form, backend='cuda')[source]

Given a fem.Form, wrap the tabulate tensors for use on a GPU.