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

293 Commits

Author SHA1 Message Date
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
Jake VanderPlas
c34b118972 Restore upstream-nightly github action 2024-04-04 19:00:55 -07:00
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)

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

Signed-off-by: dependabot[bot] <support@github.com>
 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)

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

Signed-off-by: dependabot[bot] <support@github.com>
 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)

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

Signed-off-by: dependabot[bot] <support@github.com>
 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)

---
updated-dependencies:
- dependency-name: pre-commit/action
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
 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)

---
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-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)

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

Signed-off-by: dependabot[bot] <support@github.com>
 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)

---
updated-dependencies:
- dependency-name: styfle/cancel-workflow-action
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
 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)

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

Signed-off-by: dependabot[bot] <support@github.com>
 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)

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

Signed-off-by: dependabot[bot] <support@github.com>
 2024-01-15 18:02:31 +00:00