rocm_jax/docs/jax.experimental.compilation_cache.rst

``jax.experimental.compilation_cache`` module
===============================================

JAX disk compilation cache.

.. automodule:: jax.experimental.compilation_cache.compilation_cache

API
---

.. autofunction:: is_initialized
.. autofunction:: initialize_cache
.. autofunction:: set_cache_dir
.. autofunction:: reset_cache
Document the compilation cache 2023-11-19 11:18:43 -08:00			``jax.experimental.compilation_cache`` module
			`===============================================`

			`JAX disk compilation cache.`

			`.. automodule:: jax.experimental.compilation_cache.compilation_cache`

			`API`
			`---`

			`.. autofunction:: is_initialized`
			`.. autofunction:: initialize_cache`
Cleanup deprecated compilation cache APIs. Since the compilation cache is now initialized lazily, existing APIs initialize_cache() and is_initialized() are confusing. Deprecate these APIs. Introduce a new API set_cache_dir() to explicitly set the cache directory path in code. Testing: revised unit tests, test workload. PiperOrigin-RevId: 598073423 2024-01-12 22:44:03 -08:00			`.. autofunction:: set_cache_dir`
Document the compilation cache 2023-11-19 11:18:43 -08:00			`.. autofunction:: reset_cache`