Skip to content

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu"). #1116

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu").

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu"). #1116

Triggered via pull request November 14, 2023 22:00
Status Success
Total duration 5m 17s
Artifacts

ci.yml

on: pull_request
Matrix: build-and-test
Fit to window
Zoom out
Zoom in

Annotations

2 warnings
Python 3.10 on ubuntu-latest
The following actions uses node12 which is deprecated and will be forced to run on node16: actions/checkout@v2, actions/setup-python@v1. For more info: https://github.blog/changelog/2023-06-13-github-actions-all-actions-will-run-on-node16-instead-of-node12-by-default/
Python 3.9 on ubuntu-latest
The following actions uses node12 which is deprecated and will be forced to run on node16: actions/checkout@v2, actions/setup-python@v1. For more info: https://github.blog/changelog/2023-06-13-github-actions-all-actions-will-run-on-node16-instead-of-node12-by-default/