Peter Hawkins
1db4af1c3c
Use scipy 1.13.0 instead of scipy 1.13.0rc1 in CI.
...
scipy 1.13.0 was just released.
PiperOrigin-RevId: 621552425
2024-04-03 09:45:09 -07:00
Peter Hawkins
1159691fc6
[JAX] Update JAX CI dockerfiles to use NumPy 2.0.0rc1, SciPy 1.13.0rc1, and ml_dtypes 0.4.0.
...
Change in preparation for releasing JAX with NumPy 2.0 support.
PiperOrigin-RevId: 621354875
2024-04-02 18:26:30 -07:00
dependabot[bot]
7a57b39a35
Bump actions/setup-python from 5.0.0 to 5.1.0
...
Bumps [actions/setup-python](https://github.com/actions/setup-python ) from 5.0.0 to 5.1.0.
- [Release notes](https://github.com/actions/setup-python/releases )
- [Commits](0a5c615913...82c7e631bb
2024-04-01 17:31:33 +00:00
dependabot[bot]
5ddc65ff56
Bump actions/cache from 4.0.1 to 4.0.2
...
Bumps [actions/cache](https://github.com/actions/cache ) from 4.0.1 to 4.0.2.
- [Release notes](https://github.com/actions/cache/releases )
- [Changelog](https://github.com/actions/cache/blob/main/RELEASES.md )
- [Commits](ab5e6d0c87...0c45773b62
2024-03-25 17:51:20 +00:00
Jake VanderPlas
dad31cf8ce
[array-api] update to latest array_api_tests commit
2024-03-18 17:37:46 -07:00
Shuhan Ding
9bdad4debe
setup ci for jax-metal plugin
2024-03-06 10:43:13 -08:00
dependabot[bot]
8a62918910
Bump actions/cache from 4.0.0 to 4.0.1
...
Bumps [actions/cache](https://github.com/actions/cache ) from 4.0.0 to 4.0.1.
- [Release notes](https://github.com/actions/cache/releases )
- [Changelog](https://github.com/actions/cache/blob/main/RELEASES.md )
- [Commits](13aacd865c...ab5e6d0c87
2024-03-04 17:06:05 +00:00
Jieying Luo
4c57d09590
[PJRT C API] Bump the minimum support libtpu version because there is a breaking change ( 075d25e0c1).
...
Also remove skip condition that are no longer needed because of this bump.
PiperOrigin-RevId: 611288492
2024-02-28 17:51:24 -08:00
Jieying Luo
ca1844dd60
[PJRT C API] Bump the minimum support libtpu version and clean up version check.
...
PiperOrigin-RevId: 610508159
2024-02-26 13:26:01 -08:00
Yash Katariya
d6a4fb5919
Create a unique name for each windows wheel that's uploaded to artifacts. This is required after a84f3e1207
...
PiperOrigin-RevId: 610029261
2024-02-24 11:34:48 -08:00
jax authors
1f06e34567
Merge pull request #19859 from jakevdp:update-array-api
...
PiperOrigin-RevId: 608713922
2024-02-20 13:15:34 -08:00
Sergei Lebedev
57e59eb6c3
Removed deprecated jax.config methods and jax.config.config
...
Reverts dcc65e621ea3a68fdc79fa9f2c995743a7b3faf7
PiperOrigin-RevId: 608676645
2024-02-20 11:25:16 -08:00
Jake VanderPlas
fafb5bba9c
[array api] update to latest test suite version
2024-02-20 08:56:04 -08:00
Thomas Köppe
dcc65e621e
Reverts b506fee9e389391efb1336bc7575dba913e75cdf
...
PiperOrigin-RevId: 608319964
2024-02-19 06:23:00 -08:00
Sergei Lebedev
b506fee9e3
Removed deprecated jax.config methods and jax.config.config
...
Reverts eb0343683547b6e2d29245f3ab6c91037c0cff81
PiperOrigin-RevId: 607803834
2024-02-19 06:21:15 -08:00
Sergei Lebedev
881436240e
Inlined triton.compat
...
We no longer need a compatibility layer, since Pallas does not use any Triton
IR building APIs.
PiperOrigin-RevId: 606948415
2024-02-14 05:23:15 -08:00
Jake VanderPlas
60bc415dd5
Streamline github issue template
2024-02-13 11:18:48 -08:00
jax authors
3a51de8140
Merge pull request #19757 from google:dependabot/github_actions/actions/upload-artifact-4.3.1
...
PiperOrigin-RevId: 606285111
2024-02-12 10:22:21 -08:00
dependabot[bot]
d44513d612
Bump pre-commit/action from 3.0.0 to 3.0.1
...
Bumps [pre-commit/action](https://github.com/pre-commit/action ) from 3.0.0 to 3.0.1.
- [Release notes](https://github.com/pre-commit/action/releases )
- [Commits](646c83fcd0...2c7b3805fd
2024-02-12 17:18:20 +00:00
dependabot[bot]
a84f3e1207
Bump actions/upload-artifact from 4.3.0 to 4.3.1
...
Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact ) from 4.3.0 to 4.3.1.
- [Release notes](https://github.com/actions/upload-artifact/releases )
- [Commits](26f96dfa69...5d5d22a312
2024-02-12 17:18:16 +00:00
Jake VanderPlas
6b713071e4
[array API] update to recent array API test version
2024-02-05 15:15:57 -08:00
Jake VanderPlas
2878567d43
api_test: install ci jaxlib version
2024-01-31 14:31:12 -08:00
jax authors
c7a1a095cf
Merge pull request #19556 from google:dependabot/github_actions/actions/upload-artifact-4.3.0
...
PiperOrigin-RevId: 602436797
2024-01-29 11:00:10 -08:00
dependabot[bot]
39422ca9cb
Bump actions/upload-artifact from 4.2.0 to 4.3.0
...
Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact ) from 4.2.0 to 4.3.0.
- [Release notes](https://github.com/actions/upload-artifact/releases )
- [Commits](694cdabd8b...26f96dfa69
2024-01-29 17:51:10 +00:00
dependabot[bot]
2b075fab75
Bump styfle/cancel-workflow-action from 0.12.0 to 0.12.1
...
Bumps [styfle/cancel-workflow-action](https://github.com/styfle/cancel-workflow-action ) from 0.12.0 to 0.12.1.
- [Release notes](https://github.com/styfle/cancel-workflow-action/releases )
- [Commits](01ce38bf96...85880fa030
2024-01-29 17:51:05 +00:00
jax authors
a415b567d3
Merge pull request #19464 from jakevdp:actions-cache
...
PiperOrigin-RevId: 600512505
2024-01-22 11:21:30 -08:00
Jake VanderPlas
3392b642f0
Bump actions/cache from 3.3.3 to 4.0.0
2024-01-22 10:54:32 -08:00
dependabot[bot]
577230e2b4
Bump actions/upload-artifact from 4.1.0 to 4.2.0
...
Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact ) from 4.1.0 to 4.2.0.
- [Release notes](https://github.com/actions/upload-artifact/releases )
- [Commits](1eb3cb2b3e...694cdabd8b
2024-01-22 17:51:38 +00:00
Jake VanderPlas
ad54a50c8b
CI: avoid parallelizing sphinx build to fix pydata-sphinx-theme warning
2024-01-18 15:46:38 -08:00
jax authors
c21602ef88
Merge pull request #19367 from google:dependabot/github_actions/actions/cache-3.3.3
...
PiperOrigin-RevId: 598876775
2024-01-16 10:17:45 -08:00
dependabot[bot]
2d8760f002
Bump actions/cache from 3.3.2 to 3.3.3
...
Bumps [actions/cache](https://github.com/actions/cache ) from 3.3.2 to 3.3.3.
- [Release notes](https://github.com/actions/cache/releases )
- [Changelog](https://github.com/actions/cache/blob/main/RELEASES.md )
- [Commits](704facf57e...e12d46a63a
2024-01-15 18:02:31 +00:00
dependabot[bot]
3facc199ed
Bump actions/upload-artifact from 4.0.0 to 4.1.0
...
Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact ) from 4.0.0 to 4.1.0.
- [Release notes](https://github.com/actions/upload-artifact/releases )
- [Commits](c7d193f32e...1eb3cb2b3e
2024-01-15 18:02:24 +00:00
Jake VanderPlas
635b9b6029
CI: update array API tests pin
2024-01-10 09:27:57 -08:00
Jake VanderPlas
873b33d7a5
CI: bump actions/upload-artifact from 3.1.3 to 4.0.0
2023-12-18 09:36:01 -08:00
Jieying Luo
717059edcf
Bump oldest support libtpu version to 20230927 which includes CopyToMemorySpace.
...
batch_device_put in libtpu will go through memory space path, and requires CopyToMemorySpace to be implemented if the backend uses memory space.
PiperOrigin-RevId: 591373300
2023-12-15 15:19:06 -08:00
Jake VanderPlas
ad67726c6e
array-api: update CI to use latest array-api-tests commit
2023-12-15 10:49:56 -08:00
Jieying Luo
8128b54e15
Increase test timeout for Cloud TPU nightly CI.
...
Recent runs got timeout when it is close to finish. https://github.com/google/jax/actions/runs/7182549669/job/19559323926
PiperOrigin-RevId: 590253735
2023-12-12 10:30:09 -08:00
Jake VanderPlas
b1e9afcaa6
CI: use ratchet to pin actions commits
2023-12-11 10:56:16 -08:00
Jake VanderPlas
bdd8318e36
CI: update actions/setup-python to v5
2023-12-11 09:23:30 -08:00
jax authors
e686ed7e93
Merge pull request #18870 from jakevdp:array-api-tests
...
PiperOrigin-RevId: 589184048
2023-12-08 10:58:23 -08:00
Jake VanderPlas
4b1077da09
array-api: update test suite & fix nonzero
2023-12-08 08:55:57 -08:00
Jake VanderPlas
eea2eabe0e
remove upstream-nightly CI job
...
Because of C-API changes, this no longer works with upstream numpy without a jaxlib
build, which is too expensive to run as part of a github action. We should replace
with a nightly kokoro job, but in the meantime there's no reason to use these CI
resources for a failing nightly run.
2023-12-08 08:46:53 -08:00
Peter Hawkins
7274960a6d
Remove nightly NVIDIA GPU multiprocess CI.
...
This CI seems to be dead.
2023-12-01 17:13:24 -05:00
Jake VanderPlas
ada5fe5dc9
Remove numpy-dispatch CI job & simplify build specification
...
The numpy-dispatch approach has been superseded by the Python Array API (Tracked for JAX in https://github.com/google/jax/issues/18353 ).
While we're here, we'll reduce the github CI to only two jobs: the oldest and newest supported Python versions. Other versions can be covered by Kokoro.
PiperOrigin-RevId: 587041291
2023-12-01 10:03:15 -08:00
Jake VanderPlas
053e2cff11
Update array-api-tests to most recent commit
2023-11-28 08:17:45 -08:00
Peter Hawkins
6a1dde605b
Request specific version information in the bug template.
2023-11-21 14:42:38 -05:00
Jake VanderPlas
c235045422
CI: pin actions/setup-python & actions/github-script to recent release
2023-11-20 09:58:14 -08:00
George Necula
3601b25899
Move multi_platform_export_test.py out of jax2tf.
...
This test is now independent of jax2tf. Move it out and rename it export_harnesses_multi_platform_test.py.
We disable the test in GitHub CI, because it is very large, pending
some changes to ensure it parallelizes well. The test is still
running in internal CI. This is matching the current behavior, since
jax2tf tests are only run internally.
PiperOrigin-RevId: 583603863
2023-11-18 02:52:44 -08:00
Jake VanderPlas
78b889632b
[array api] update array-api-tests
2023-11-17 08:59:53 -08:00
Jake VanderPlas
271d31c1c8
Add jax.experimental.array_api interface
2023-11-16 14:21:04 -08:00
