mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-21 22:36:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
12,059 Commits 508 Branches 263 Tags
Commit Graph

24 Commits

Author SHA1 Message Date
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