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
12,356 Commits 508 Branches 263 Tags
Commit Graph

87 Commits

Author SHA1 Message Date
Jake VanderPlas
993196c451 CI: make parse_logs more robust to errors 2022-07-20 10:33:32 -07:00
Jake VanderPlas
2eaa44e8d8 [CI] upstream-dev: run all tests 2022-07-18 11:06:56 -07:00
Yash Katariya
90433c0518
Update ci-build.yaml 2022-07-15 16:03:16 -07:00
jax authors
5a10c1af3c Merge pull request #11487 from google:upstream-dev-logs 
PiperOrigin-RevId: 461241333
 2022-07-15 14:43:37 -07:00
Jake VanderPlas
b0cd7de999 CI: use minimum jaxlib in upstream-ci build 2022-07-14 14:34:47 -07:00
Jake VanderPlas
b97f6f4819 upstream-dev: add failure information to CI issues 2022-07-13 14:05:11 -07:00
Jake VanderPlas
917c4e9dfc add upstream-nightly CI job 2022-07-12 17:04:53 -07:00
Arjun Sharda
cc8e302933 Update ci-build.yaml 
Update ci-build.yaml
 2022-06-28 12:45:59 -07:00
dependabot[bot]
19ae5d8581
Bump styfle/cancel-workflow-action from 0.9.1 to 0.10.0 
Bumps [styfle/cancel-workflow-action](https://github.com/styfle/cancel-workflow-action) from 0.9.1 to 0.10.0.
- [Release notes](https://github.com/styfle/cancel-workflow-action/releases)
- [Commits](https://github.com/styfle/cancel-workflow-action/compare/0.9.1...0.10.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>
 2022-06-27 17:17:21 +00:00
dependabot[bot]
11777fab8c
Bump actions/setup-python from 3 to 4 
Bumps [actions/setup-python](https://github.com/actions/setup-python) from 3 to 4.
- [Release notes](https://github.com/actions/setup-python/releases)
- [Commits](https://github.com/actions/setup-python/compare/v3...v4)

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

Signed-off-by: dependabot[bot] <support@github.com>
 2022-06-13 17:09:30 +00:00
Sharad Vikram
143ed40a78 Add collect_profile script 2022-06-03 17:56:17 -07:00
Alex Alemi
4ca92edaec Remove pull request notification. 
PiperOrigin-RevId: 452280419
 2022-06-01 06:20:09 -07:00
Alex Alemi
e8a92f33b0 Remove bodies from the notification messages as they tend to break bash syntax. 
PiperOrigin-RevId: 452068897
 2022-05-31 09:39:25 -07:00
Alex Alemi
57da8d941b Attempting again to fix the yaml syntax issues in the pull request notifier and trying to fix the trigger for the broken main. 2022-05-27 16:40:48 -04:00
Alex Alemi
8e34061739 Fixing the interpolation errors and yaml syntax errors. 2022-05-27 14:07:21 -04:00
Alex Alemi
6c7542e4a3 adding notification workflows. 2022-05-27 12:19:47 -04:00
Jeppe Klitgaard
a11f15e3ec feat: officially support Python 3.10 2022-05-07 13:43:12 +01:00
dependabot[bot]
bff0845794
Bump actions/cache from 2 to 3 
Bumps [actions/cache](https://github.com/actions/cache) from 2 to 3.
- [Release notes](https://github.com/actions/cache/releases)
- [Commits](https://github.com/actions/cache/compare/v2...v3)

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

Signed-off-by: dependabot[bot] <support@github.com>
 2022-03-21 17:13:18 +00:00
dependabot[bot]
16b147861d
Bump actions/checkout from 2 to 3 
Bumps [actions/checkout](https://github.com/actions/checkout) from 2 to 3.
- [Release notes](https://github.com/actions/checkout/releases)
- [Changelog](https://github.com/actions/checkout/blob/main/CHANGELOG.md)
- [Commits](https://github.com/actions/checkout/compare/v2...v3)

---
updated-dependencies:
- dependency-name: actions/checkout
  dependency-type: direct:production
  update-type: version-update:semver-major
...

Signed-off-by: dependabot[bot] <support@github.com>
 2022-03-07 17:12:34 +00:00
dependabot[bot]
680c06ddc4
Bump actions/setup-python from 2 to 3 
Bumps [actions/setup-python](https://github.com/actions/setup-python) from 2 to 3.
- [Release notes](https://github.com/actions/setup-python/releases)
- [Commits](https://github.com/actions/setup-python/compare/v2...v3)

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

Signed-off-by: dependabot[bot] <support@github.com>
 2022-02-28 17:11:41 +00:00
Jake VanderPlas
f2222bb1cf CI: error if docstring rewrite fails 2022-02-07 14:43:00 -08:00
Yash Katariya
0bb7d204ab Move serialization/de-serialization of GDA into jax. 
PiperOrigin-RevId: 414607092
 2021-12-06 20:05:02 -08:00
Peter Hawkins
70b8a6a806 Add a prototype IREE backend for JAX. 
This is to support experimentation with the combination of JAX/IREE. Many things do not work yet.

PiperOrigin-RevId: 409980064
 2021-11-15 07:57:04 -08:00
Peter Hawkins
8f6e077d9a Adds an initial prototype of an alternate JAX compilation path that emits the MLIR MHLO/CHLO dialects instead of classic XLA HLO. 
This lowering is missing a number of features, but it is complete enough that many tests pass, and that I would like to start checking it in.

PiperOrigin-RevId: 409134016
 2021-11-11 06:37:12 -08:00
Skye Wanderman-Milne
2fcf3f7270 Remove .[minimum-jaxlib] from test-requirements.txt 
This means that jax and its dependencies (e.g. jaxlib) must be
manually installed before running the tests. This is useful for
testing an existing jax install, e.g. a later version of jaxlib, GPU
jaxlib, etc.
 2021-09-23 12:24:24 -07:00
Yash Katariya
19d7710d35 Remove the last long running CI build from github actions which is being replaced by internal infra build. 
PiperOrigin-RevId: 394694532
 2021-09-03 09:04:10 -07:00
Yash Katariya
b31293f743 Replace github action test with test running on internal infra. Also replace JAX_NUM_GENERATED_CASES to 1 for test with 3.8 and test with numpy dispatch tests to reduce the build time. 
PiperOrigin-RevId: 394531539
 2021-09-02 13:35:19 -07:00
jax authors
d349086ca5 Merge pull request #7683 from google:dependabot/github_actions/pre-commit/action-2.0.3 
PiperOrigin-RevId: 392098389
 2021-08-20 17:17:02 -07:00
dependabot[bot]
60f025dd41
Bump pre-commit/action from 2.0.0 to 2.0.3 
Bumps [pre-commit/action](https://github.com/pre-commit/action) from 2.0.0 to 2.0.3.
- [Release notes](https://github.com/pre-commit/action/releases)
- [Commits](https://github.com/pre-commit/action/compare/v2.0.0...v2.0.3)

---
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>
 2021-08-20 20:59:50 +00:00
dependabot[bot]
050cce0358
Bump styfle/cancel-workflow-action from 0.8.0 to 0.9.1 
Bumps [styfle/cancel-workflow-action](https://github.com/styfle/cancel-workflow-action) from 0.8.0 to 0.9.1.
- [Release notes](https://github.com/styfle/cancel-workflow-action/releases)
- [Commits](https://github.com/styfle/cancel-workflow-action/compare/0.8.0...0.9.1)

---
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>
 2021-08-20 20:59:48 +00:00
Matthieu MOREL
6a360e9684
Update dependabot.yml 2021-08-19 20:21:16 +02:00
Matthieu MOREL
952bd03f20
chore(ci) : setup dependabot 
Signed-off-by: Matthieu MOREL <mmorel-35@users.noreply.github.com>
 2021-08-19 20:17:45 +02:00
Yash Katariya
f729f7ea1f Replace py3.8 build with JAX_NUM_GENERATED_CASES=25 with JAX_NUM_GENERATED_CASES=10. 
JAX_NUM_GENERATED_CASES=25 is now using kokoro with py3.9 `CPU` which is 4x faster than the github action.

Since py3.8 build with JAX_NUM_GENERATED_CASES=25 on github action is the slowest build, this should give an improvement of 10 minutes overall since the second slowest build is 10 mins faster than the above one.

But moving py3.8 build (Github actions) with JAX_NUM_GENERATED_CASES=25 to Kokoro (JAX_NUM_GENERATED_CASES=25 and py3.9 and `CPU`) gives a `4x` speedup (testing time).
Overall kokoro time is around 11 mins. So its a `2.5x` speedup overall.

Github actions:

```
=============== 23033 passed, 2033 skipped in 1727.43s (0:28:47) ===============
```

Kokoro:

```
=============== 22877 passed, 2065 skipped in 446.36s (0:07:26) ================

Total kokoro time = 11 mins. So 2.5x speedup overall.
```

PiperOrigin-RevId: 390656668
 2021-08-13 11:33:39 -07:00
Peter Hawkins
6e9169d100 Drop support for NumPy 1.17. 2021-07-29 09:18:01 -04:00
Peter Hawkins
94446ff757 Drop Python 3.6 support. 
Per the deprecation policy (https://jax.readthedocs.io/en/latest/deprecation.html),
Python 3.6 support has been due for removal since June 23, 2020.
 2021-07-15 14:20:29 -04:00
Qiao Zhang
0216a317a0 Change test matrix to use latest jaxlib release. 2021-07-12 15:04:10 -07:00
George Necula
6a48c60a72 Rename master to main in embedded links. 
Tried to avoid the change on external links to repos that
have not yet renamed master.
 2021-06-18 10:00:01 +03:00
George Necula
a6c16119c9 Renaming master to main: step 1. 
Here we just enable the GitHub actions to run on both master and main.
 2021-06-17 14:25:01 +03:00
Peter Hawkins
b130257ee1 Drop support for NumPy 1.16. 2021-06-11 09:03:09 -04:00
Peter Hawkins
f07ccf0074 Use short tracebacks in CI builds. 
Often useful information is hard to see in the GitHub UI with the default traceback verbosity of pytest.
 2021-05-26 13:43:38 -04:00
Jake VanderPlas
ed66dd4d57 CI: test Python 3.9 with numpy dispatch test 2021-05-17 10:10:52 -07:00
Peter Hawkins
aea068ed84 Reenable numpy dispatch tests. 2021-05-11 09:12:06 -04:00
Lukas Geiger
78ea235ac7 CI: Install wheel pkg to improve pip-cache 2021-04-16 02:55:41 +02:00
Peter Hawkins
3861da4ffc Remove omnistaging references in CI build config. 2021-04-09 15:44:18 -04:00
Jake VanderPlas
cee3af580b Testing: avoid global fixture for doctests 2021-04-08 12:11:14 -07:00
Jake VanderPlas
8e789c7380 Run doctest on all source files except jax2tf 2021-04-05 10:39:59 -07:00
George Necula
7c4e26af8c Rename GitHub workflow "without omnistaging" to "with 3.7" 
PiperOrigin-RevId: 366267935
 2021-04-01 09:34:42 -07:00
Jake VanderPlas
8f40b68cc6 CI: quick fix for disable omnistaging test 2021-03-25 14:30:22 -07:00
Jake VanderPlas
01ee29cf88 CI: fix broken package-overrides check 2021-03-16 16:48:49 -07:00
Jake VanderPlas
60dcd0da61 Set reasonable timeouts for github actions jobs 2021-03-12 09:31:43 -08:00