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
jax authors
7ffcbbbe5c
Merge pull request #18019 from google:dependabot/github_actions/styfle/cancel-workflow-action-0.12.0
...
PiperOrigin-RevId: 580675174
2023-11-08 15:13:53 -08:00
Jieying Luo
8ffc88c1e4
Bump LIBTPU_OLDEST_VERSION_DATE to 20230915 to include https://github.com/openxla/xla/pull/5617 .
...
PiperOrigin-RevId: 577285685
2023-10-27 13:12:36 -07:00
dependabot[bot]
65fed59cda
Bump actions/checkout from 4.1.0 to 4.1.1
...
Bumps [actions/checkout](https://github.com/actions/checkout ) from 4.1.0 to 4.1.1.
- [Release notes](https://github.com/actions/checkout/releases )
- [Changelog](https://github.com/actions/checkout/blob/main/CHANGELOG.md )
- [Commits](8ade135a41...b4ffde65f4
2023-10-23 17:39:31 +00:00
Skye Wanderman-Milne
f16ea4f180
Fix oldest supported libtpu install command
2023-10-20 09:45:05 -07:00
Samuel Agyakwa
0cd6ef012e
Merge remote-tracking branch 'origin/patch-1' into patch-1
2023-10-13 16:47:10 +00:00
Samuel Agyakwa
91f17459ed
making ENABLE_PJRT_COMPATIBILITY conditionally true
2023-10-12 23:42:39 +00:00
Samuel Agyakwa
d69e810e90
Merge branch 'google:main' into patch-1
2023-10-12 19:50:25 +00:00
Samuel Agyakwa
f59b157b6a
Setup compatibility testing to support the oldest supported libtpu version (12 weeks)
...
- A new matrix option was added to jaxlib-version matrix
- Expected to immediately fail due to strict jaxlib version check (requires libtpu to have the same API verison as jaxlib, fails otherwise)
- Due to expected failure, so chat will be sent when the jaxlib-version matrix is "nightly+oldest_supported_libtpu"
2023-10-11 15:02:15 -07:00
dependabot[bot]
c24c7da085
Bump styfle/cancel-workflow-action from 0.11.0 to 0.12.0
...
Bumps [styfle/cancel-workflow-action](https://github.com/styfle/cancel-workflow-action ) from 0.11.0 to 0.12.0.
- [Release notes](https://github.com/styfle/cancel-workflow-action/releases )
- [Commits](https://github.com/styfle/cancel-workflow-action/compare/0.11.0...0.12.0 )
---
updated-dependencies:
- dependency-name: styfle/cancel-workflow-action
dependency-type: direct:production
update-type: version-update:semver-minor
...
Signed-off-by: dependabot[bot] <support@github.com>
2023-10-09 17:51:44 +00:00
Peter Hawkins
58cb0d9ede
Use the released version of Python 3.12 in the Windows wheel builds.
2023-10-03 21:51:37 -04:00
Skye Wanderman-Milne
24c9e62910
Remove the LD_PRELOAD workaround introduced in ef241e506e
...
libtpu has been updated to hopefully avoid the deadlock.
2023-10-02 11:30:20 -07:00
Skye Wanderman-Milne
ef241e506e
Cloud TPU CI: don't use tcmalloc (temporary workaround for tcmalloc deadlock)
2023-09-28 16:55:11 -07:00
Skye Wanderman-Milne
b6bd74f331
[Cloud TPU CI] Update job names to be more descriptive and add description comment to yaml file.
2023-09-27 09:56:55 -07:00
Peter Hawkins
cf28e2c5fa
Small improvements to build/build.py
...
Add a --verbose option that logs all shell() commands run by the script.
Remove some Python 2 backward compatibility logic related to urllib and shutil.
Enable debug logging on Windows wheel builds.
Also include setuptools in the build requirements and test for its presence in build.py.
2023-09-26 21:55:45 -04:00
dependabot[bot]
867201a77e
Bump actions/checkout from 4.0.0 to 4.1.0
...
Bumps [actions/checkout](https://github.com/actions/checkout ) from 4.0.0 to 4.1.0.
- [Release notes](https://github.com/actions/checkout/releases )
- [Changelog](https://github.com/actions/checkout/blob/main/CHANGELOG.md )
- [Commits](3df4ab11eb...8ade135a41
2023-09-25 17:21:10 +00:00
jax authors
5db71b877f
Merge pull request #17674 from hawkinsp:py312
...
PiperOrigin-RevId: 566813669
2023-09-19 18:56:45 -07:00
