mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,216 Commits 506 Branches 263 Tags
Commit Graph

475 Commits

Author SHA1 Message Date
dependabot[bot]
802abfef92
Bump actions/upload-artifact from 4.3.5 to 4.3.6 
Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact) from 4.3.5 to 4.3.6.
- [Release notes](https://github.com/actions/upload-artifact/releases)
- [Commits](89ef406dd8...834a144ee9)

---
updated-dependencies:
- dependency-name: actions/upload-artifact
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
 2024-08-12 17:54:09 +00:00
Jake VanderPlas
d999208863 [array API] update test suite to most recent commit 2024-08-08 12:33:30 -07:00
dependabot[bot]
6d7cf3fcfd
Bump actions/upload-artifact from 4.3.3 to 4.3.5 
Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact) from 4.3.3 to 4.3.5.
- [Release notes](https://github.com/actions/upload-artifact/releases)
- [Commits](65462800fd...89ef406dd8)

---
updated-dependencies:
- dependency-name: actions/upload-artifact
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
 2024-08-05 17:09:11 +00:00
Jake VanderPlas
521c94c6c6 Tighten the public API for jax.lib.xla_client & xla_extension 2024-08-03 05:26:22 -07:00
Jake VanderPlas
48c5fab023 [array api] fix deprecation to support old import pattern 2024-08-01 14:38:59 -07:00
Jake VanderPlas
14fa06298e [array api] Finalize array API in jax.numpy & deprecate jax.experimental.array_api 2024-08-01 11:19:17 -07:00
jax authors
83f0c979fa Merge pull request #22456 from google:dependabot/github_actions/actions/setup-python-5.1.1 
PiperOrigin-RevId: 654787031
 2024-07-22 09:36:33 -07:00
Sergei Lebedev
7a62b8dd18 Re-enabled PallasCallPrintTest on Cloud TPUs 
PiperOrigin-RevId: 652823653
 2024-07-16 06:55:20 -07:00
dependabot[bot]
98f978f32b
Bump actions/setup-python from 5.1.0 to 5.1.1 
Bumps [actions/setup-python](https://github.com/actions/setup-python) from 5.1.0 to 5.1.1.
- [Release notes](https://github.com/actions/setup-python/releases)
- [Commits](82c7e631bb...39cd14951b)

---
updated-dependencies:
- dependency-name: actions/setup-python
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
 2024-07-15 18:01:34 +00:00
Peter Hawkins
3019467992 Update references to jaxlib_nightly_releases (which is a legacy index and no longer updated) to jax_nightly_releases. 
Fixes CI jobs using old jaxlib nightly wheels.
 2024-07-15 12:42:50 -04:00
dependabot[bot]
8bc4643627
Bump actions/download-artifact from 4.1.7 to 4.1.8 
Bumps [actions/download-artifact](https://github.com/actions/download-artifact) from 4.1.7 to 4.1.8.
- [Release notes](https://github.com/actions/download-artifact/releases)
- [Commits](65a9edc588...fa0a91b85d)

---
updated-dependencies:
- dependency-name: actions/download-artifact
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
 2024-07-08 17:27:27 +00:00
Jake VanderPlas
d03572eac5 CI: add jax2tf_test action 2024-06-27 10:23:58 -07:00
Peter Hawkins
945fde41e4 Update minimum Python version to 3.10. 2024-06-26 13:47:14 -04:00
Peter Hawkins
f5f9da34e6 Update Github actions builds to use Python 3.10 instead of Python 3.9. 
Change in preparation for dropping Python 3.9 support.

PiperOrigin-RevId: 646997432
 2024-06-26 10:33:11 -07:00
Dan Foreman-Mackey
5b5214b507 Update Windows Actions workflow name 
When skipped, the build matrix for the Windows workflow on GitHub
Actions doesn't get populated, so the run name ends up as
`${{ matrix.os }} CI build`. It doesn't seem necessary to include the
matrix value in the name since there is only one build anyways. This PR
replaces the name with a constant.
 2024-06-18 12:54:03 -04:00
Peter Hawkins
b0b02682d5 Add --allow-downgrade to Windows wheel builds. 
We want a specific LLVM version, downgrades are fine.
Also add --no-progress for a more compact log.
 2024-06-18 08:15:26 -04:00
Peter Hawkins
160e09e235 Use NumPy 2.0.0 and SciPy 1.13.1 in builds. 
Don't override the XLA repository in the nightly Windows CI builds,
which should be building JAX as it exists in the source repository.
 2024-06-17 19:35:08 +00:00
Peter Hawkins
02395a406a Add --allow-downgrade to LLVM install on Windows. 
We want to pin a specific version in CI, even if a newer version exists.
 2024-06-13 15:08:43 -04:00
Jake VanderPlas
d8f9709a53 Add option to force windows CI run by adding label 2024-06-13 11:30:01 -07:00
George Necula
105cc9a103 [export] Add documentation for jax.export 2024-06-12 19:44:47 +02:00
Peter Hawkins
ab6d361620 [windows] Add --no-progress to LLVM install command to avoid excessive logs in Github actions. 2024-06-05 10:52:43 -04:00
George Necula
0223e830c1 [docs] Enable doctest for .md documentation files 
The current invocation of doctest from GH actions picked up only .rst files.
We enable .md files also, and we make a few changes to ensure that the
doctest passes on the existing files.

The changes fall into several categories:
  * add a newline before the end of the code block, for doctest to
    pick up the expected output properly
  * update the expected values to match the current behavior
  * disable some doctests that raise expected exceptions, whenever
    I could not get doctest to match the exception details.
    Sometimes +IGNORE_EXCEPTION_DETAIL was enough, and other times
    I had to use +SKIP.
 2024-06-03 11:27:34 +03:00
Jake VanderPlas
6fb357d651 [array api] update test suite to most recent commit 2024-05-29 08:48:49 -07:00
Sergei Lebedev
2473ebf508 Removed mentions of iree from the test suite 2024-05-24 10:31:57 +01:00
Jake VanderPlas
d5d2fb087f Ignore deprecation warnings locally rather than globally 2024-05-20 20:28:25 -07:00
Michael Hudgins
76a7b19c10
Set V3 cores to 4 2024-05-15 14:40:57 -04:00
Michael Hudgins
0232cb9f8d
Run v3-8 tests with cores set at 8 2024-05-15 14:37:27 -04:00
Michael Hudgins
181da12809
Add missing bash symbol 2024-05-15 14:33:01 -04:00
Michael Hudgins
3015699966
Add core count to tpu nightly fix v5 job 
The current job assumes a 4 core TPU.   Modify the matrix to enable defining the core count for each tpu
 2024-05-15 13:46:40 -04:00
Jake VanderPlas
35a512dadf CI: update NumPy build version to 2.0.0rc2 
PiperOrigin-RevId: 633233231
 2024-05-13 09:18:40 -07:00
jax authors
0e57c79b0b Switch Windows jobs to use Clang. 
Remove the experimental/trial Clang job.

PiperOrigin-RevId: 631814321
 2024-05-08 08:47:34 -07:00
Jake VanderPlas
496795e95f CI: fix typo in workflow 2024-05-07 11:14:11 -07:00
jax authors
eee2783e85 Merge pull request #21070 from shuhand0:rel0.0.7 
PiperOrigin-RevId: 631218770
 2024-05-06 16:22:15 -07:00
Shuhan Ding
aac36799fd
fix jaxlib config name 2024-05-06 13:51:22 -07:00
jax authors
bb6aa12fc1 Merge pull request #21087 from jakevdp:upstream-print-version 
PiperOrigin-RevId: 631154080
 2024-05-06 12:48:29 -07:00
Jake VanderPlas
7d96e78a55 CI: print numpy/scipy version in upstream job 2024-05-06 11:14:38 -07:00
jax authors
a265e42192 Add an experimental, Clang version of the Windows CI job. 
Once proven to work, this job will be deleted, and the MSVC job changed to use Clang.

PiperOrigin-RevId: 631122967
 2024-05-06 11:10:09 -07:00
Shuhan Ding
28fe45d872
metal_plugin ci with jaxlib nightly 2024-05-02 18:40:01 -07:00
jax authors
5d9b5adb17 Return to the win-2019 image for the windows_ci workflow. 
Due to https://github.com/bazelbuild/bazel/issues/18592

PiperOrigin-RevId: 629459786
 2024-04-30 10:40:27 -07:00
jax authors
e680c495a2 Try fixing the MSVC path for windows_ci workflow once more. 
PiperOrigin-RevId: 629436690
 2024-04-30 09:26:27 -07:00
jax authors
1bb8b0fe8c Point the windows_ci workflow to the correct VC directory. 
PiperOrigin-RevId: 629418384
 2024-04-30 08:18:41 -07:00
jax authors
bc0fa9b0e3 Use the windows-2022 image for running the windows_ci workflow. 
This should help reduce the number of MSVC-related errors.

PiperOrigin-RevId: 629404922
 2024-04-30 07:24:43 -07:00
dependabot[bot]
bc354dfeaf
Bump actions/download-artifact from 4.1.6 to 4.1.7 
Bumps [actions/download-artifact](https://github.com/actions/download-artifact) from 4.1.6 to 4.1.7.
- [Release notes](https://github.com/actions/download-artifact/releases)
- [Commits](9c19ed7fe5...65a9edc588)

---
updated-dependencies:
- dependency-name: actions/download-artifact
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
 2024-04-29 17:16:47 +00:00
jax authors
461c5ec455 Merge pull request #20867 from google:dependabot/github_actions/actions/download-artifact-4.1.6 
PiperOrigin-RevId: 627137002
 2024-04-22 13:24:16 -07:00
dependabot[bot]
e93954c13c
Bump actions/upload-artifact from 4.3.1 to 4.3.3 
Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact) from 4.3.1 to 4.3.3.
- [Release notes](https://github.com/actions/upload-artifact/releases)
- [Commits](5d5d22a312...65462800fd)

---
updated-dependencies:
- dependency-name: actions/upload-artifact
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
 2024-04-22 18:06:45 +00:00
dependabot[bot]
4b79d362aa
Bump actions/download-artifact from 4.1.4 to 4.1.6 
Bumps [actions/download-artifact](https://github.com/actions/download-artifact) from 4.1.4 to 4.1.6.
- [Release notes](https://github.com/actions/download-artifact/releases)
- [Commits](c850b930e6...9c19ed7fe5)

---
updated-dependencies:
- dependency-name: actions/download-artifact
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
 2024-04-22 18:06:38 +00:00
Adam Paszke
8e3f5b1018 Initial commit for Mosaic GPU 
Moving this to JAX to make it easier to explore Pallas integration.

PiperOrigin-RevId: 625982382
 2024-04-18 04:04:10 -07:00
Jake VanderPlas
572c16284e [array api] update to latest test repo commit 2024-04-16 06:09:00 -07:00
Jake VanderPlas
3f6c4fc60d CI: run the metal CI nightly & on PRs touching the config file 2024-04-11 11:18:27 -07:00
Jake VanderPlas
f029532183 CI: pin actions dependencies to most recent release 2024-04-08 11:24:43 -07:00