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

chore: adopt new local wheel installation logic

Browse Source 
PiperOrigin-RevId: 641972325
This commit is contained in:
Vadym Matsishevskyi 2024-06-10 11:41:07 -07:00 committed by jax authors
parent 3fe7377719
commit a073476fa0
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
WORKSPACE
View File

@ -15,6 +15,9 @@ python_init_repositories(
"3.12": "//build:requirements_lock_3_12.txt",
"3.13": "//build:requirements_lock_3_13.txt",
},
local_wheel_workspaces = ["//jaxlib:jax.bzl"],
local_wheel_dist_folder = "../dist",
default_python_version = "system",
)
load("@xla//third_party/py:python_init_toolchains.bzl", "python_init_toolchains")