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

108 Commits

Author SHA1 Message Date
Roy Frostig
2a7b3197e0 add nvidia-smi question to bug template 2022-09-26 11:06:29 -07:00
Peter Hawkins
ba557d5e1b Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.". 
See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.

PiperOrigin-RevId: 476167538
 2022-09-22 12:27:19 -07:00
Yash Katariya
160e14308c [Rollback] Add a github presubmit build which runs with jax.Array flag enabled for OSS coverage. 
PiperOrigin-RevId: 473161716
 2022-09-08 22:09:44 -07:00
Yash Katariya
49672cd2bc Add a github presubmit build which runs with jax.Array flag enabled for OSS coverage. 
PiperOrigin-RevId: 473100614
 2022-09-08 15:32:31 -07:00
Sudhakar
5f1858f533 Add pytest marker inside the test only if pytest is present in the env 2022-09-06 11:45:59 -07:00
Roy Frostig
fe2de26b2c test the RNG key upgrade in one of our github CI runs 
Choosing the "numpy-dispatch" configuration since it tends to be our
frequent pick for future-proofing.
 2022-08-30 21:43:44 -07:00
Sudhakar
a571db18db Enable one gpu per process in multinode GPU CI 2022-08-29 09:00:19 -07:00
Sudhakar
4b1a2eaaec combine gpu tests 2022-08-25 15:27:07 -07:00
Roy Frostig
655ecb4aaf
fix bug issue template yaml parsing error 2022-08-23 19:56:41 -07:00
Roy Frostig
825cbace65
remove 'BUG' title prefix from bug report template 
We auto-label with "bug" to this end.
 2022-08-22 15:09:38 -07:00
Sudhakar
c2e521807c Add support to test gpu jaxlib nightly in CI instead of prebuilt jax/jaxlib 2022-08-19 11:08:11 -07:00
Jake VanderPlas
f00dfb434e issue template: avoid checkboxes because they're interpreted as tasks 2022-08-11 14:05:23 -07:00
Jake VanderPlas
7ec6acd981 nightly multiprocess test: create issue on failure 2022-08-09 19:12:32 -07:00
Sudhakar Singh
efb37ff784 Bump EnricoMi/publish-unit-test-result-action from 1 to 2 2022-08-08 10:55:53 -07:00
jax authors
0a8ca1982c Merge pull request #11721 from sudhakarsingh27:main 
PiperOrigin-RevId: 465381834
 2022-08-04 12:52:16 -07:00
Jake VanderPlas
03e5f9a24e bug-report.yml: fix markdown formatting 2022-08-04 10:48:30 -07:00
Jake VanderPlas
f0c5747d9a bug-report.yaml: fix yaml syntax error 2022-08-04 10:30:25 -07:00
Sudhakar Singh
1565fd2525 Squashed commit of the following: 
commit 0b4c3f05a49037be93eb0612113e193f3a8d61c5
Author: Sudhakar Singh <sudhakars@nvidia.com>
Date:   Thu Aug 4 09:53:04 2022 -0700

    change the path

commit 2c629739c1cfa45d848a2cf7109d329c1262e6ac
Author: Sudhakar Singh <sudhakars@nvidia.com>
Date:   Wed Aug 3 16:37:46 2022 -0700

    rename file to reflect current objective

commit ef46bcae6cd66d6fe7b04bd6d8aeed42c4f3ddfa
Author: Sudhakar Singh <sudhakars@nvidia.com>
Date:   Wed Aug 3 15:56:32 2022 -0700

    correct formatting

commit e5da60ad855592d5f150612f65ad679872160132
Author: Sudhakar Singh <sudhakars@nvidia.com>
Date:   Wed Aug 3 15:26:32 2022 -0700

    Add multi-node multi-GPU JAX tests

    This adds multi-node multi-GPU test for `jax.distributed.initialize`.
    Presently, this is expected to run on a nightly basis. Under the hood,
    SLURM is used to launch the `pytest <test_name>` commands on multiple
    nodes.

    Resolves: #11648
 2022-08-04 10:13:50 -07:00
Jake VanderPlas
945adfdd98 Add enhanced bug report template 2022-08-04 09:31:06 -07:00
jax authors
9f96a0474e Merge pull request #11624 from alexalemi:notifications 
PiperOrigin-RevId: 463390452
 2022-07-26 11:43:51 -07:00
Alex Alemi
31d2a74aeb remove broken-main-notify 2022-07-26 14:29:31 -04:00
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