mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 04:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,443 Commits 503 Branches 263 Tags
Commit Graph

2 Commits

Author SHA1 Message Date
jax authors
b8b119d9b9 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:48 -08:00
Frederic Bastien
72b6c9cf0b Document the compilation cache 2023-11-20 07:03:30 -08:00