We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 622f4f9 commit fb2d4e1Copy full SHA for fb2d4e1
experimental/torch_xla2/torch_xla2/__init__.py
@@ -22,7 +22,8 @@
22
f'ml_framework_name:PyTorch/XLA2;ml_framework_version:{"v0.0.1"}'
23
)
24
xla_bridge._clear_backends()
25
- jax.devices() # open PJRT to see if it opens
+ if os.environ.get("DISABLE_XLA2_PJRT_TEST") != "true":
26
+ jax.devices() # open PJRT to see if it opens
27
except RuntimeError:
28
jax.config.update(
29
'jax_pjrt_client_create_options', old_pjrt_options
0 commit comments