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

Switch to zstd for numpy array serialization (jax.Array serialization is handled by JAX library).

Browse Source 
PiperOrigin-RevId: 522616067
This commit is contained in:
Colin Gaffney 2023-04-07 09:35:24 -07:00 committed by jax authors
parent a556074541
commit 38f6338299
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/experimental/array_serialization/serialization.py
View File

@ -70,7 +70,7 @@ def _get_metadata(arr):
local_shape = arr.addressable_data(0).shape
return {
'compressor': {
'id': 'gzip'
'id': 'zstd'
},
'shape': arr.shape,
'chunks': np.array(np.maximum(1, local_shape)),