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

Add upgrade=True to jax_array flag so that its marked as transient flag which will eventually be set to True.

Browse Source
This commit is contained in:
Yash Katariya 2022-08-22 13:07:53 -07:00 committed by GitHub
parent 384776f0c9
commit 37089ec1b8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
jax/_src/config.py
View File

@ -655,6 +655,7 @@ parallel_functions_output_gda = config.define_bool_state(
jax_array = config.define_bool_state(
name='jax_array',
default=False,
upgrade=True,
help=('If True, new pjit behavior will be enabled and `jax.Array` will be '
'used.'))