jax
98ee69ba - Emit a warning on non-hashble static arguments in jax.jit.

Commit
5 years ago
Emit a warning on non-hashble static arguments in jax.jit. The message looks like, e.g.: Static argument (index 1) of type <class 'numpy.ndarray'> for function f is non-hashable. As this can lead to unexpected cache-misses, it will raise an error in a near future.
Author
Committer
Parents
Loading