mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00

History

Skye Wanderman-Milne 120125f3dd Make pytest-xdist work on TPU and update Cloud TPU CI.

This change also marks multiaccelerator test files in a way pytest can
understand (if pytest is installed).

By running single-device tests on a single TPU chip, running the test
suite goes from 1hr 45m to 35m (both timings are running slow tests).

I tried using bazel at first, which already supported parallel
execution across TPU cores, but somehow it still takes 2h 20m! I'm not
sure why it's so slow. It appears that bazel creates many new test
processes over time, vs. pytest reuses the number of processes
initially specified, and starting and stopping the TPU runtime takes a
few seconds so that may be adding up. It also appears that
single-process bazel is slower than single-process pytest, which I
haven't looked into yet.

2022-11-18 22:05:13 +00:00

self_hosted_runner_utils

Add .github/workflows/self_hosted_runner_utils/README.md

2022-11-04 17:12:54 +00:00

slurm_job_scripts

Add multihost GPU CI run with last public jaxlib release

2022-09-29 17:06:56 -07:00

cat_slurm_logs.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

ci-build.yaml

Use jax.Array by default for doctests

2022-11-15 11:52:22 -08:00

cloud-tpu-ci-nightly.yml

Make pytest-xdist work on TPU and update Cloud TPU CI.

2022-11-18 22:05:13 +00:00

nightly-ci-multiprocess-gpu.yml

Update Github Actions workflows using Ratchet

2022-11-16 18:45:59 +00:00

parse_logs.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

README.md

Update Github Actions workflows using Ratchet

2022-11-16 18:45:59 +00:00

release-notification.yml

Update Github Actions workflows using Ratchet

2022-11-16 18:45:59 +00:00

upstream-nightly.yaml

[CI] upstream-dev: run all tests

2022-07-18 11:06:56 -07:00

README.md

Github Actions workflows

See the Github documentation for more information on Github Actions in general.

Notes

https://opensource.google/documentation/reference/github/services#actions mandates using a specific commit for non-Google actions. We use Ratchet to pin specific versions. If you'd like to update an action, you can write something like uses: 'actions/checkout@v3', and then run ./ratchet pin workflow.yml to convert to a commit hash. See the Ratchet README for installation and more detailed instructions.