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

Internal Code Change

Browse Source 
PiperOrigin-RevId: 520341781
This commit is contained in:
jax authors 2023-03-29 08:23:18 -07:00
parent 7200d07db5
commit a964ae7fac
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
jax/BUILD
View File

@ -88,6 +88,7 @@ py_library(
py_library_providing_imports_info( py_library_providing_imports_info(
name = "jax", name = "jax",
srcs = [ srcs = [
"_src/__init__.py",
"_src/abstract_arrays.py", "_src/abstract_arrays.py",
"_src/ad_checkpoint.py", "_src/ad_checkpoint.py",
"_src/ad_util.py", "_src/ad_util.py",
@ -104,7 +105,6 @@ py_library_providing_imports_info(
"_src/dispatch.py", "_src/dispatch.py",
"_src/dlpack.py", "_src/dlpack.py",
"_src/flatten_util.py", "_src/flatten_util.py",
"_src/__init__.py",
"_src/lax_reference.py", "_src/lax_reference.py",
"_src/maps.py", "_src/maps.py",
"_src/pjit.py", "_src/pjit.py",
@ -165,8 +165,8 @@ py_library_providing_imports_info(
deps = [ deps = [
":basearray", ":basearray",
":cloud_tpu_init", ":cloud_tpu_init",
":custom_api_util",
":config", ":config",
":custom_api_util",
":deprecations", ":deprecations",
":core", ":core",
":effects", ":effects",
@ -252,8 +252,8 @@ pytype_strict_library(
name = "environment_info", name = "environment_info",
srcs = ["_src/environment_info.py"], srcs = ["_src/environment_info.py"],
deps = [ deps = [
":xla_bridge",
":version", ":version",
":xla_bridge",
"//jax/_src/lib", "//jax/_src/lib",
] + py_deps("numpy"), ] + py_deps("numpy"),
) )
@ -391,8 +391,8 @@ pytype_strict_library(
":cloud_tpu_init", ":cloud_tpu_init",
":config", ":config",
":iree", ":iree",
":util",
":traceback_util", ":traceback_util",
":util",
"//jax/_src/lib", "//jax/_src/lib",
] + py_deps("numpy"), ] + py_deps("numpy"),
) )