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 #8054 from skye:fix

Browse Source 
PiperOrigin-RevId: 399996116
This commit is contained in:
jax authors 2021-09-30 12:34:37 -07:00
commit 372839863d
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
jax/_src/lib/xla_bridge.py
View File

@ -48,8 +48,7 @@ FLAGS = flags.FLAGS
# TODO(phawkins): Remove jax_xla_backend.
flags.DEFINE_string(
'jax_xla_backend', '',
'jax_xla_backend is an alias for jax_platform_name. If both are '
'provided, --jax_xla_backend takes priority. Prefer --jax_platform_name.')
'Deprecated, please use --jax_platforms instead.')
flags.DEFINE_string(
'jax_backend_target', '',
'Either "local" or "rpc:address" to connect to a remote service target.')