Skip to contents

JIT compile a function

Source: R/interpreter-jit.R
jit.Rd

Convert a function to a JIT compiled function.

Usage

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

Arguments

f

(function)
Function to compile.

static

(character())
Which parameters of f are static.

device

(NULL | character(1) | PJRTDevice)
The device to use for the compiled function. The default (NULL) uses the PJRT_PLATFORM environment variable or defaults to "cpu".

cache_size

(integer(1))
The size of the cache for the jit-compiled functions.

Value

(function)