Mark a function for automatic JIT compilation, with operator fusion.
The function will be compiled the first time it is called with a set of
argument shapes.
You can call .dispose() on the returned, JIT-compiled function after all
calls to free memory associated with array constants.
Options:
staticArgnums: An array of argument indices to treat as static
(compile-time constant). These arguments must be hashable, won't be traced,
and different values will trigger recompilation.
device: The device to place the computation on. If not specified, the
computation will be placed on the default device.
Mark a function for automatic JIT compilation, with operator fusion.
The function will be compiled the first time it is called with a set of argument shapes.
You can call
.dispose()on the returned, JIT-compiled function after all calls to free memory associated with array constants.Options:
staticArgnums: An array of argument indices to treat as static (compile-time constant). These arguments must be hashable, won't be traced, and different values will trigger recompilation.device: The device to place the computation on. If not specified, the computation will be placed on the default device.