Internal.

PiperOrigin-RevId: 710787526
jax-ml · Jan 6, 2025 · fa52f25 · fa52f25
1 parent 18b193c
commit fa52f25
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/jax/BUILD b/jax/BUILD
@@ -19,6 +19,7 @@ load("@rules_python//python:defs.bzl", "py_library")
 load(
     "//jaxlib:jax.bzl",
     "if_building_jaxlib",
+    "jax_export_file_visibility",
     "jax_extend_internal_users",
     "jax_extra_deps",
     "jax_internal_export_back_compat_test_util_visibility",
@@ -62,6 +63,11 @@ exports_files([
     "version.py",
 ])
 
+exports_files(
+    ["_src/export/serialization.fbs"],
+    visibility = jax_export_file_visibility,
+)
+
 # Packages that have access to JAX-internal implementation details.
 package_group(
     name = "internal",

diff --git a/jaxlib/jax.bzl b/jaxlib/jax.bzl
@@ -399,6 +399,8 @@ def jax_wheel(name, wheel_binary, enable_cuda = False, platform_version = ""):
 
 jax_test_file_visibility = []
 
+jax_export_file_visibility = []
+
 def xla_py_proto_library(*args, **kw):  # buildifier: disable=unused-variable
     pass
-Original file line number
+Diff line change
@@ Expand Up @@
     jax_test_file_visibility = []
+    jax_export_file_visibility = []
     def xla_py_proto_library(*args, **kw):  # buildifier: disable=unused-variable
         pass
@@ Expand Down @@