mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +00:00
b8b119d9b9
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
15 lines
333 B
ReStructuredText
15 lines
333 B
ReStructuredText
``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
|