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

Fix Windows CI to install the jaxlib wheel it builds.

Browse Source 
Currently we install jax first and then try to install jaxlib. But pip won't overwrite the jaxlib installed as a jax dependency, so our CI builds were always using the released jaxlib.
This commit is contained in:
Peter Hawkins 2024-10-02 08:47:31 -04:00
parent e2d3bd866a
commit 9400ef57a1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/windows_ci.yml vendored
View File

@ -66,7 +66,7 @@ jobs:
PY_COLORS: 1
run: |
cd jax
python -m pip install --pre --find-links ${{ github.workspace }}\jax\dist jaxlib
python -m pip install -e ${{ github.workspace }}\jax
python -m pip install --no-index --find-links ${{ github.workspace }}\jax\dist jaxlib
echo "JAX_ENABLE_CHECKS=$JAX_ENABLE_CHECKS"
pytest -n auto --tb=short tests examples