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

No changes.

Browse Source 
PiperOrigin-RevId: 515737909
This commit is contained in:
Peter Hawkins 2023-03-10 14:22:55 -08:00 committed by jax authors
parent 054c07b025
commit a722ec08f9
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/BUILD
View File

@ -315,6 +315,7 @@ pytype_library(
"_src/distributed.py",
"_src/xla_bridge.py",
],
visibility = [":internal"] + jax_visibility("xla_bridge"),
deps = [
":cloud_tpu_init",
":config",