Fix tests

google · Jun 19, 2024 · f282493 · f282493
1 parent 0f75875
commit f282493
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/jax/_src/compilation_cache.py b/jax/_src/compilation_cache.py
@@ -113,7 +113,8 @@ def count_cache_items() -> int:
 
 
 def clear_cache() -> None:
-  _cache.clear()
+  if _cache is not None:
+    _cache.clear()
 
 
 def default_min_cache_entry_size() -> int: