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

[Rollback] Add a github presubmit build which runs with jax.Array flag enabled for OSS coverage.

Browse Source 
PiperOrigin-RevId: 473161716
This commit is contained in:
Yash Katariya 2022-09-08 22:09:17 -07:00 committed by jax authors
parent 9050f0d3a9
commit 160e14308c
1 changed files with 0 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
.github/workflows/ci-build.yaml vendored
View File

@ -45,7 +45,6 @@ jobs:
os: ubuntu-latest
enable-x64: 0
prng-upgrade: 0
jax_array: 0
package-overrides: "none"
num_generated_cases: 1
use-latest-jaxlib: false
@ -54,7 +53,6 @@ jobs:
os: ubuntu-latest
enable-x64: 1
prng-upgrade: 1
jax_array: 0
# Test experimental NumPy dispatch
package-overrides: "git+https://github.com/seberg/numpy-dispatch.git"
num_generated_cases: 1
@ -64,16 +62,6 @@ jobs:
os: ubuntu-latest
enable-x64: 0
prng-upgrade: 0
jax_array: 0
package-overrides: "none"
num_generated_cases: 1
use-latest-jaxlib: false
- name-prefix: "with 3.10 and jax.Array"
python-version: "3.10"
os: ubuntu-latest
enable-x64: 0
prng-upgrade: 0
jax_array: 1
package-overrides: "none"
num_generated_cases: 1
use-latest-jaxlib: false
@ -119,14 +107,12 @@ jobs:
JAX_ENABLE_CUSTOM_PRNG: ${{ matrix.prng-upgrade }}
JAX_ENABLE_CHECKS: true
JAX_SKIP_SLOW_TESTS: true
JAX_ARRAY: ${{ matrix.jax_array }}
run: |
pip install -e .
echo "JAX_NUM_GENERATED_CASES=$JAX_NUM_GENERATED_CASES"
echo "JAX_ENABLE_X64=$JAX_ENABLE_X64"
echo "JAX_ENABLE_CUSTOM_PRNG=$JAX_ENABLE_CUSTOM_PRNG"
echo "JAX_ENABLE_CHECKS=$JAX_ENABLE_CHECKS"
echo "JAX_ARRAY=$JAX_ARRAY"
pytest -n auto --tb=short tests examples