JIT compile a function — jit • anvil

Convert a function to a JIT compiled function.

Usage

jit(
  f,
  static = character(),
  cache_size = 100L,
  donate = character(),
  device = NULL
)

f: (function)
Function to compile.
static: (character())
Which parameters of f are static.
cache_size: (integer(1))
The size of the cache for the jit-compiled functions.
donate: (character())
Names of the arguments whose buffers should be donated. Donated buffers can be aliased with outputs of the same type, allowing in-place operations and reducing memory usage.
device: (NULL | character(1) | PJRTDevice)
The device to use if no input tensors are provided to infer the platform.

(function)