mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #14789 from patrick-kidger:patch-2

Browse Source 
PiperOrigin-RevId: 514490253
This commit is contained in:
jax authors 2023-03-06 12:27:53 -08:00
commit 7d154103e3
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/_src/numpy/lax_numpy.py
View File

@ -162,6 +162,8 @@ def _dtype(x: Any) -> DType:
# types return JAX arrays when instantiated.
class _ScalarMeta(type):
dtype: np.dtype
def __hash__(self) -> int:
return hash(self.dtype.type)