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

Merge pull request #13382 from levskaya:fasterci

Browse Source 
PiperOrigin-RevId: 490600867
This commit is contained in:
jax authors 2022-11-23 15:54:37 -08:00
commit 8788a9438f
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -22,7 +22,7 @@ permissions:
jobs:
lint_and_typecheck:
runs-on: ubuntu-latest
runs-on: ubuntu-20.04-16core
timeout-minutes: 5
steps:
- name: Cancel previous
@ -46,7 +46,7 @@ jobs:
include:
- name-prefix: "with 3.8"
python-version: "3.8"
os: ubuntu-latest
os: ubuntu-20.04-16core
enable-x64: 0
prng-upgrade: 0
package-overrides: "none"
@ -54,7 +54,7 @@ jobs:
use-latest-jaxlib: false
- name-prefix: "with numpy-dispatch"
python-version: "3.9"
os: ubuntu-latest
os: ubuntu-20.04-16core
enable-x64: 1
prng-upgrade: 1
# Test experimental NumPy dispatch
@ -63,7 +63,7 @@ jobs:
use-latest-jaxlib: false
- name-prefix: "with 3.10"
python-version: "3.10"
os: ubuntu-latest
os: ubuntu-20.04-16core
enable-x64: 0
prng-upgrade: 0
package-overrides: "none"
@ -123,7 +123,7 @@ jobs:
documentation:
runs-on: ubuntu-latest
runs-on: ubuntu-20.04-16core
timeout-minutes: 5
strategy:
matrix:
@ -159,5 +159,5 @@ jobs:
XLA_FLAGS: "--xla_force_host_platform_device_count=8"
JAX_ARRAY: 1
run: |
pytest -n 1 --tb=short docs
pytest -n 1 --tb=short --doctest-modules jax --ignore=jax/experimental/jax2tf --ignore=jax/_src/lib/mlir --ignore=jax/interpreters/mlir.py --ignore=jax/_src/iree.py --ignore=jax/experimental/gda_serialization --ignore=jax/collect_profile.py
pytest -n auto --tb=short docs
pytest -n auto --tb=short --doctest-modules jax --ignore=jax/experimental/jax2tf --ignore=jax/_src/lib/mlir --ignore=jax/interpreters/mlir.py --ignore=jax/_src/iree.py --ignore=jax/experimental/gda_serialization --ignore=jax/collect_profile.py