Skip to content

Conversation

hawkinsp
Copy link
Collaborator

…ache.

Some users were trying to use the internal method f._cache_clear() to clear the function cache associated with a jit. Unfortunately that method only clears the C++ cache, whereas to actually evict functions one must clear a Python-level cache as well. Add a new f.clear_cache() method that clears both caches.

Fixes #10878

Somewhat related to issue #10828 as well.

@google-ml-butler google-ml-butler bot added kokoro:force-run pull ready Ready for copybara import and testing labels Jun 11, 2022
@copybara-service copybara-service bot merged commit 691a1f8 into jax-ml:main Jun 14, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

pull ready Ready for copybara import and testing

Projects

None yet

Development

Successfully merging this pull request may close these issues.

GPU Memory Leak when I use numpy under a jitted fucntion.

3 participants