mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
14,751 Commits 504 Branches 263 Tags
Commit Graph

40 Commits

Author SHA1 Message Date
Peter Hawkins
ec56d71d01 Drop support for NVIDIA Kepler series GPUs in jaxlib builds. 2023-02-10 14:15:15 -05:00
Yash Katariya
71569e1d16 Remove the specialized sm versions for testing. It caused release wheels to ship without correct sm versions compiled in. 
PiperOrigin-RevId: 494854373
 2022-12-12 16:02:46 -08:00
Yash Katariya
ea63baf1ad Remove python 3.7 from testing and release builds 
PiperOrigin-RevId: 491752393
 2022-11-29 15:26:46 -08:00
Yash Katariya
03aa266e45 Build GPU wheel only for Tesla (t4) when running on RBE. THis should in theory speed up the builds 
PiperOrigin-RevId: 490553589
 2022-11-23 11:57:24 -08:00
Yash Katariya
51e4b017bb Use the cuda 11.8 image for CPU builds to reduce the docker image churn from the JAX side in TF's RBE cluster 
PiperOrigin-RevId: 490522647
 2022-11-23 09:33:17 -08:00
Yash Katariya
6d11567142 Add cuda 11.8 configs to .bazelrc 
PiperOrigin-RevId: 489591457
 2022-11-18 16:55:28 -08:00
Yash Katariya
d77cccff43 Update docker images to add python 3.11 support. 
PiperOrigin-RevId: 485488215
 2022-11-01 20:35:38 -07:00
Rohit Santhanam
663441007c [ROCm] Added gfx90a and gfx1030. 2022-10-27 16:57:06 +00:00
Peter Hawkins
aa6445e6f5 Set --tsl_link_protobuf=true in .bazelrc to unbreak jaxlib build. 2022-10-13 13:50:09 +00:00
jax authors
254dc24a8b Merge pull request #11961 from jakeh-gc:plugin_device 
PiperOrigin-RevId: 476363760
 2022-09-23 07:29:17 -07:00
jax authors
fd90f40c45 Merge pull request #12443 from cloudhan:fix-mlir-chlo-stablehlo-symbols 
PiperOrigin-RevId: 475808753
 2022-09-21 06:12:44 -07:00
Cloud Han
dc414c8635 Switch from experimental_strict_action_env to incompatible_strict_action_env to avoid deprecation warning 2022-09-21 17:26:21 +08:00
Yash Katariya
a2e05b0df5 Try Cuda 11.2 build in presubmit for GPU. Presubmit uses 11.2 and CI uses 11.1 
PiperOrigin-RevId: 473071603
 2022-09-08 13:34:03 -07:00
Jake
21f82c6c0d Use the pjrt plugin device client. 2022-08-17 14:34:07 +01:00
Peter Hawkins
184f8ab723 Import .bazelrc.user to allow users to override bazel options easily. 2022-08-14 15:46:12 -04:00
Rohit Santhanam
080cf47002 [ROCm] Fixes for compilation failures caused by compiler changes in ROCm Tensorflow fork. 2022-06-29 14:34:08 +00:00
Peter Hawkins
69bda69fb6 Bump minimum Mac OS version to 10.14 (Mojave). 
It turns out that the support for C++17 is partial in 10.12, and in particular absl::optional and std::optional are not the same thing under 10.12. Increment to 10.14 which is the lowest version that builds successfully with absl::optional == std::optional.

See: 89cdaed655/absl/base/config.h (L528)
Strictly speaking, we could allow 10.13, but not without updating ABSL in the TF repository to incorporate c86347d4ce which fixes the version detection test to permit 10.13 as well.
 2022-06-01 20:32:22 -04:00
Nicholas Junge
7f7358cb58 Fix macOS builds by updating the minimum OSX version target 
This commit bumps the minimum macOS version target to 10.12, from 10.9
previously. The reason is that a newer LLVM version, which the XLA
compiler depended on, used `std::shared_mutex`, which is only available
starting at macOS 10.12.

The fix here consists of setting the minimum version target flag that
bazel consumes for the build to 10.12.
 2022-05-31 15:14:48 +02:00
Yash Katariya
dd261bde59 Switch to using C++17 
PiperOrigin-RevId: 450559689
 2022-05-23 17:15:00 -07:00
Peter Hawkins
ea868b4ffe Update .bazelrc to reflect GPU/TPU build changes. 
PiperOrigin-RevId: 449306362
 2022-05-17 13:48:55 -07:00
Peter Hawkins
9fb9e12169 Don't include PTX for older GPU generations. 
See: https://github.com/tensorflow/tensorflow/pull/55613

For a CUDA build at head with the default compute capabilities, reduces wheel size from 141MB to 112MB.

Don't redundantly specify default compute capabilities in .bazelrc and in
build.py.
 2022-05-02 20:27:37 +00:00
Yash Katariya
5fdad0ebf5 Roll forward manylinux2014 builds after fixes. 
PiperOrigin-RevId: 440589273
 2022-04-09 10:19:44 -07:00
Yash Katariya
e9f95fa5fa Make jaxlib builds manylinux2014 compliant. 
PiperOrigin-RevId: 440497401
 2022-04-08 18:51:46 -07:00
Yash Katariya
506a85b7ff Make jaxlib builds manylinux2014 compliant. 
PiperOrigin-RevId: 440476417
 2022-04-08 16:21:56 -07:00
Yash Katariya
aa5d6b4a58 Fix the breakage by including --experimental_cc_shared_library as done by TF. 
PiperOrigin-RevId: 438746867
 2022-03-31 23:07:42 -07:00
Peter Hawkins
2e0cfe8e42 Update the list of default CUDA capabilities used for wheel builds to match build.py. 2022-02-15 09:23:28 -05:00
Yash Katariya
5efa28566e Switch Cuda 11.4 cudnn 8.0.5 to build with cuda 11.1 cudnn 8.0.5 instead. 
PiperOrigin-RevId: 425681466
 2022-02-01 12:01:31 -08:00
Peter Hawkins
66823d1392 Include compute capability 8.0 SASS in jaxlib wheels. 
Drop compute capability 6.1 to avoid growing the wheel size.

Also fix an unrelated build error due to a gcc warning in boringssl.
 2021-12-14 14:27:19 -05:00
Yash Katariya
271bd3237b Add python 3.10 bazel build configs for jaxlib. 
PiperOrigin-RevId: 413755904
 2021-12-02 13:27:28 -08:00
Peter Hawkins
ce7ae6bd76 Make MLIR bindings build work under Bazel. 
Tested on Linux and Mac, but not Windows.
 2021-11-12 12:16:32 -05:00
Yash Katariya
e6e81ba885 Add Cuda 11.4 with cudnn 8.2 and cudnn 8.0.5 release builds 
PiperOrigin-RevId: 403661187
 2021-10-16 16:13:43 -07:00
Yash Katariya
0578ba68f4 Docker file for Cuda 11.4 built with Cudnn 8.0.5 
PiperOrigin-RevId: 403283584
 2021-10-14 22:51:23 -07:00
Yash Katariya
960d19e54b bazelrc changes for building Cuda 11.4 wheels with cudnn 8.2 
PiperOrigin-RevId: 397088624
 2021-09-16 08:37:26 -07:00
Yash Katariya
c69eedd6f8 Delete Cuda 10.2 release and continuous builds. 
PiperOrigin-RevId: 394772236
 2021-09-03 16:00:42 -07:00
Yash Katariya
d7b9adce1d Replace cuda 11.2 configs with cuda 11.1 since that's the earliest version where backwards compatibility is supported. 
PiperOrigin-RevId: 392714272
 2021-08-24 11:58:00 -07:00
Yash Katariya
c39c0933ae Add cuda 10.2 configs in bazelrc for RBE builds. 
PiperOrigin-RevId: 391899496
 2021-08-19 19:47:28 -07:00
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
476642578b Make .bazelrc more configurable so that cuda 10.2 configs and any other configs for RBE can be added. 
PiperOrigin-RevId: 391343779
 2021-08-17 11:58:04 -07:00
Yash Katariya
dca68f2dd4 Fix copybara so that .bazelrc will go out to github 
PiperOrigin-RevId: 390020496
 2021-08-10 17:51:15 -07:00