autopdex.utility.jit_with_docstring

autopdex.utility.jit_with_docstring(static_argnames=None, possibly_static_argnames=None, inline=False)[source]

JIT wrapper that preserves the original docstring of the function and additionally treats arguments from possibly_static_argnames as static if their value is callable.

Parameters:

  • static_argnames – Iterable of argument names that are always static.

  • possibly_static_argnames – Iterable of argument names that are treated as static only if their value is callable.