jax
deab6fbd - Remove _pjit_lower_cached cache. We can simplify the caching of jit as we have downstream caches and a cpp cache too.

Commit

1 year ago

Remove _pjit_lower_cached cache. We can simplify the caching of jit as we have downstream caches and a cpp cache too. If you drop out of cpp cache, things are going to be slow anyways. PiperOrigin-RevId: 700052522

References

#170 - CI: 12/02/24 upstream sync

Author

yashk2810

Committer

Google-ML-Automation

Parents

066859e6

jax deab6fbd - Remove _pjit_lower_cached cache. We can simplify the caching of jit as we have downstream caches and a cpp cache too.

jax
deab6fbd - Remove _pjit_lower_cached cache. We can simplify the caching of jit as we have downstream caches and a cpp cache too.