warp.jax_experimental.ffi.set_jax_callable_default_graph_cache_max#

warp.jax_experimental.ffi.set_jax_callable_default_graph_cache_max(cache_max)[source]#

Set the maximum size of the graph cache for graphs captured using GraphMode.WARP, unlimited if None.

Parameters:

cache_max (int | None)