mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 11:46:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
8,864 Commits 508 Branches 263 Tags
Commit Graph

78 Commits

Author SHA1 Message Date
Yash Katariya
7dca05066a Remove build --strategy=Genrule=standalone since this makes genrule run locally instead of on remote RBE clusters leading to errors that are very hard to debug. 
Add this back in build.py because that script is used for building jaxlib locally.

PiperOrigin-RevId: 391603481
 2021-08-18 13:56:43 -07:00
Reza Rahimi
f454f6b7b8 fix rocm_amdgpu_targets for rocm 2021-08-17 22:13:29 +00:00
Yash Katariya
2d1854a8ba Create a .bazelrc file that is the base for all the builds. The current build.py workflow will not be affected since this .bazelrc file will be overridden. I am going to change that workflow in the coming CLs. 
PiperOrigin-RevId: 390003558
 2021-08-10 16:17:58 -07:00
jax authors
1a2ddc055d Merge pull request #7419 from cloudhan:win-builder-prepare 
PiperOrigin-RevId: 389938513
 2021-08-10 11:42:48 -07:00
yashkatariya
b24574f743 Fix the regex 2021-08-05 13:25:06 -07:00
yashkatariya
86aaf80dce Use bazel --version 2021-08-05 09:45:39 -07:00
Cloud Han
be1705306c fix bazel auto download 2021-07-31 19:13:16 +08:00
Cloud Han
d4349a42b5 workaround weird issue due to non-standard cuda path 2021-07-30 19:25:33 +08:00
Cloud Han
86e37d996a some fs support now can be disabled without compile error 2021-07-30 19:23:18 +08:00
Peter Hawkins
6e9169d100 Drop support for NumPy 1.17. 2021-07-29 09:18:01 -04:00
Peter Hawkins
6c08702489 Add support for ppc64le cross-compilation on Linux. 
Use Bazel 4.1.0 unconditionally on all platforms.
 2021-07-23 10:39:02 -04:00
Cloud Han
2d321c26e6 Use TF_CUDA_PATHS 
CUDA_TOOLKIT_PATH and CUDNN_INSTALL_PATH are deprecated, see TF 2.0
release notes for more information
 2021-07-18 22:55:34 +08:00
jax authors
6aa20d8f8f Merge pull request #7294 from hawkinsp:py36 
PiperOrigin-RevId: 384994957
 2021-07-15 13:19:23 -07: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
Peter Hawkins
f5c61a892a Add support for cross-compiling jaxlib for Mac ARM. 2021-07-15 10:37:53 -04:00
Peter Hawkins
7d2aec105f Add an option to disable NCCL. 2021-07-13 09:10:29 -04:00
Qiao Zhang
18c7610e96 Catch CalledProcessError in build script. 2021-06-22 13:36:14 -07:00
Peter Hawkins
b130257ee1 Drop support for NumPy 1.16. 2021-06-11 09:03:09 -04:00
akbir khan
dc81610bb6 updated to official bazel.4.1.0 2021-05-24 21:40:08 +01:00
Peter Hawkins
dacf31f202 Check for NumPy and SciPy versions during jaxlib builds. 2021-05-24 12:39:37 -04:00
Peter Hawkins
86887a21e2 Add Mac M1 support to build.py. 2021-05-18 21:53:31 -04:00
Peter Hawkins
7d4551dd45 Add support for arm64 builds to build.py and build_wheel scripts. 2021-05-10 11:08:09 -04:00
Skye Wanderman-Milne
7b7a2a1ba8 Don't build with Cloud TPU support for GPU or Mac wheels. 
This should make our builds simpler and less failure-prone.
 2021-03-08 16:19:19 -08:00
Christian Sigg
c0e42e59eb
Switch to new flag to enable CUDA support in TF 2021-03-03 08:55:08 +01:00
Situ Yi
77905ef5fd
Update build.py with Cloud TPU support 
Update build.py to have Cloud TPU support by default.
b/180656002
 2021-02-19 12:57:57 -08:00
jax authors
15359e018d Merge pull request #5738 from hawkinsp:avx 
PiperOrigin-RevId: 357785312
 2021-02-16 12:53:41 -08:00
Skye Wanderman-Milne
3caf52d54a Fix build.py version comparison 2021-02-16 11:32:50 -08:00
Peter Hawkins
a27545497b Enable AVX by default in release builds. 
TensorFlow also enables AVX by default, so this seems like a reasonable baseline CPU feature to expect.
 2021-02-16 14:12:01 -05:00
jax authors
5fd3deb0be Merge pull request #5580 from daskol:search-bazel-in-path 
PiperOrigin-RevId: 357755577
 2021-02-16 10:49:33 -08:00
Skye Wanderman-Milne
dd513c1585
Small simplification 2021-02-16 08:39:32 -08:00
Daniel Bershatsky
96193cef12 Simplify bazel version check 2021-02-13 16:32:21 +03:00
Skye Wanderman-Milne
74db363830 Bump required bazel version to 3.7.2 
Untested, but I'm about to build a new jaxlib and will make any fixes then.
 2021-02-11 09:07:04 -08:00
Daniel Bershatsky
c71a4f5871 Try to download bazel as the last resort 
Resolution order of paths to bazel binary is as follows.

1. Use --bazel_path command line option.
2. Search bazel binary in PATH environment variable.
3. Download required bazel release.
 2021-02-01 15:15:38 +03:00
Clemens Giuliani
330e69dd30 Add support for ROCm 2020-12-06 22:55:53 +01:00
Cloud Han
ea340eed93 use strict action env on windows to avoid constant full rebuilding 2020-11-24 23:44:26 +08:00
Peter Hawkins
c06ead6b04 Change jaxlib build rules to build a wheel, rather than writing output to the source directory. 2020-11-20 11:47:00 -05:00
Peter Hawkins
1fcbd2f083 Add -Wno-stringop-truncation to build flags on Linux. 
Works around https://github.com/tensorflow/tensorflow/issues/39467 for gcc 10+ builds.
 2020-11-19 22:50:08 -05:00
Cloud Han
a6acce58e0 Build on Windows 
1. Build on Windows

2. Fix OverflowError

    When calling `key = random.PRNGKey(0)` OverflowError: Python int too
    large to convert to C long for casting value 4294967295 (0xFFFFFFFF)
    from python int to int32.

3. fix file path in regex of errors_test

4. handle ValueError of os.path.commonpath
 2020-11-19 23:33:06 +08:00
Peter Hawkins
b222e5e05f Update bazel version to 3.1.0. 2020-11-17 15:50:30 -05:00
Peter Hawkins
2c6f932e0e Add Python 3.9 support to jaxlib build. 2020-11-09 16:25:58 -05:00
Srijan Saurav
40e20242db
Fix code quality issues (#4302) 
Changes:
- Fix unnecessary generator
- Iterate dictionary directly instead of calling .keys()
- Remove global statement at the module level
- Use list() instead of a list comprehension
- Use with statement to open the file
- Merge isinstance calls
 2020-09-17 09:21:18 -07:00
Peter Hawkins
a141cc6e8d
Make CUDA wheels manylinux2010 compliant, add CUDA 11, drop CUDA 9.2 (#3555) 
* Use dynamic loading to locate CUDA libraries in jaxlib.

This should allow jaxlib CUDA wheels to be manylinux2010 compliant.

* Tag CUDA jaxlib wheels as manylinux2010.

Drop support for CUDA 9.2, add support for CUDA 11.0.

* Reorder CUDA imports.
 2020-06-25 14:37:14 -04:00
Peter Hawkins
821193b2b7
Explicitly build specific CUDA capabilities. (#2722) 
We choose the same set as TensorFlow (minus 3.7, which TF is apparently considering dropping anyway).

This avoids a slow PTX -> SASS compilation on first time startup.
 2020-04-15 10:57:53 -04:00
Peter Hawkins
cbdf9a5a43
Drop support for Python 3.5. (#2445) 2020-03-18 10:54:28 -04:00
Peter Hawkins
219d503e71
Don't show progress bar in build script if output is not a terminal. (#2429) 2020-03-16 11:01:08 -04:00
Peter Hawkins
80abdf0c53
Unbreak build and update XLA. (#2289) 
* raise minimum Bazel version to 2.0.0 to match TensorFlow.
* set --experimental_repo_remote_exec since it is required by the TF build.
* bump TF/XLA version.
* use the --config=short_logs trick from TF to suppress build warnings.
 2020-02-22 09:45:24 -08:00
Peter Hawkins
b6e8341176
Improve developer documentation. (#2247) 
Add Python version test to build.py.
 2020-02-17 11:24:03 -08:00
Skye Wanderman-Milne
bf91ebf67a
Return error number in build.py on bad bazel version. (#2218) 
This prevents our build scripts from continuing on error.
 2020-02-12 09:57:54 -08:00
Skye Wanderman-Milne
96a65de6c8
Try downloading bazel before using pre-installed bazel. (#2217) 
This ensures we're using the right bazel version.
 2020-02-12 09:49:33 -08:00
Peter Hawkins
fe041c7590 Set minimum Bazel version to 1.2.1. 2020-02-03 10:13:51 -05:00