mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Limit visibility of Bazel target jax:global_device_array.

Browse Source 
PiperOrigin-RevId: 510521459
This commit is contained in:
Peter Hawkins 2023-02-17 14:23:10 -08:00 committed by jax authors
parent 7011ef0a5c
commit f7734fd6a4
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
jax/BUILD
View File

@ -17,6 +17,7 @@
load("@bazel_skylib//rules:common_settings.bzl", "bool_flag")
load(
"//jaxlib:jax.bzl",
"global_device_array_visibility",
"jax_extra_deps",
"jax_internal_packages",
"jax_test_util_visibility",
@ -269,6 +270,6 @@ pytype_library(
srcs = ["experimental/global_device_array.py"],
# TODO(phawkins): remove after removing global_device_array.py from :jax.
tags = ["keep_dep"],
visibility = ["//visibility:public"],
visibility = [":internal"] + global_device_array_visibility,
deps = [":jax"],
)

2
jaxlib/jax.bzl
View File

@ -40,6 +40,8 @@ jax_internal_packages = []
jax_test_util_visibility = []
loops_visibility = []
global_device_array_visibility = []
def py_deps(_package):
"""Returns the Bazel deps for Python package `package`."""