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 #21832 from jakevdp:serialization-version-doc

Browse Source 
PiperOrigin-RevId: 642732593
This commit is contained in:
jax authors 2024-06-12 14:13:45 -07:00
commit b7a8f9d584
3 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/export/export.md
View File

@ -584,6 +584,8 @@ scalar operands corresponding to the format specifiers.
error_message=""Dimension variable 'h' must have integer value >= 1. Found {0}")
```
(export-serialization-version)=
### Serialization version numbers
We list here a history of the serialization version numbers:

11
docs/jax.export.rst
View File

@ -34,3 +34,14 @@ Functions related to shape polymorphism
symbolic_args_specs
is_symbolic_dim
SymbolicScope
Constants
---------
.. data:: jax.export.minimum_supported_serialization_version
The minimum supported serialization version; see :ref:`export-serialization-version`.
.. data:: jax.export.maximum_supported_serialization_version
The maximum supported serialization version; see :ref:`export-serialization-version`.

4
jax/_src/export/_export.py
View File

@ -65,13 +65,13 @@ HloSharding = xla_client.HloSharding
"""The minimum supported serialization version.
See https://jax.readthedocs.io/en/latest/export.html#module-calling-convention#module-serialization-versions
See https://jax.readthedocs.io/en/latest/export/export.html#serialization-version-numbers
"""
minimum_supported_serialization_version = 9
"""The maximum supported serialization version.
See https://jax.readthedocs.io/en/latest/export.html#module-calling-convention#module-serialization-versions
See https://jax.readthedocs.io/en/latest/export/export.html#serialization-version-numbers
"""
maximum_supported_serialization_version = 9