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 |
|