mirror of
https://github.com/ROCm/jax.git
synced 2025-04-17 20:36:05 +00:00

This commit adds the CI script needed for running Bazel GPU (non-RBE) tests. These run two Bazel commands: Single accelerator tests with one GPU a piece and multi-accelerator tests with all GPUs PiperOrigin-RevId: 700523594