mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-22 08:06:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
13,235 Commits 508 Branches 263 Tags
Commit Graph

261 Commits

Author SHA1 Message Date
Peter Hawkins
8d8643664c jax/jaxlib 0.3.20 release candidate. 2022-09-28 13:33:52 +00:00
Peter Hawkins
bcd36d8eb2 Jax and jaxlib 0.3.18 release candidate. 2022-09-26 14:10:57 +00:00
Peter Hawkins
bec9d2e168 Bump XLA version. 2022-09-19 13:17:46 +00:00
jax authors
45764ea9a8 Merge pull request #12125 from hawkinsp:xla 
PiperOrigin-RevId: 470295638
 2022-08-26 11:38:03 -07:00
Peter Hawkins
12cb55d2f0 Bump XLA version. 2022-08-26 18:27:59 +00:00
Peter Hawkins
b63801b4db Fixes for PocketFFT->ducc migration. 
* Rename modules from pocketfft to ducc.
* Fix up strides at their generation point rather than where they are
  consumed.
 2022-08-26 14:30:03 +00:00
Peter Hawkins
6128cb7aa0 Update XLA commit. 2022-08-18 20:17:37 +00:00
Peter Hawkins
87db8fc5f6 Update XLA commit. 
Fixes build error:
Label
 '@org_tensorflow//tensorflow/tsl/platform/default:build_config.bzl' is
 invalid because 'tensorflow/tsl/platform/default' is not a package.
 2022-08-17 15:01:43 +00:00
Adam Paszke
6ed2d22566 Update WORKSPACE 
to include the addition of use_global_device_ids in AllReduceOp.
 2022-08-17 07:30:54 +00:00
Peter Hawkins
5603cb31d4 Bump XLA version. 
Fixes https://github.com/google/jax/issues/9771
 2022-08-15 21:12:52 +00:00
Yash Katariya
e982dd105f
Update the TF commit hash 2022-08-05 10:51:21 -07:00
Peter Hawkins
96cb47df32 Update XLA. 2022-08-01 12:10:27 +00:00
Skye Wanderman-Milne
26fbeb6e2a Update WORKSPACE and libtpu version for jaxlib 0.3.15, take 3 2022-07-22 11:41:39 -07:00
Skye Wanderman-Milne
568cedba8d
Update WORKSPACE for 0.3.15 release, take 2 2022-07-20 17:23:52 -07:00
Skye Wanderman-Milne
9149c38e1e Update WORKSPACE and setup.py in preparation for 0.3.15 jax/jaxlib release 2022-07-14 10:12:59 -07:00
Peter Hawkins
7c49864fdf Symlink xla_client and xla_extension into jaxlib rather than copying them into place in the wheel build. 
Change in preparation for allowing JAX tests to run under Bazel.

Remove code to patch paths in xla_client.py in the wheel build script; the patch is no longer used.

PiperOrigin-RevId: 458522398
 2022-07-01 12:31:42 -07:00
Peter Hawkins
f4ddd3ef88 Update XLA. 2022-06-27 09:14:28 -04:00
Sharad Vikram
217d898124 Update TF version for jaxlib build 2022-06-21 14:34:22 -07:00
Sharad Vikram
9bd1bd67e0 Update versions for jax/jaxlib release 2022-06-21 12:57:28 -07:00
Peter Hawkins
5b7c485ade Update XLA pin. 2022-06-15 09:24:57 -04:00
Peter Hawkins
1e504b36e2 Improve the comment in WORKSPACE describing how to override the TF repository. 2022-06-14 22:25:39 -04:00
Ludger Paehler
e8903d4b47 Bump version of Tensorflow repository to fix build errors in the building of the MLIR compiler where cpp files have been renamed to cc files. 2022-06-12 00:51:10 +00:00
Parker Schuh
c9ba08be71 Bump XLA version to include fix to libtpu. 
PiperOrigin-RevId: 451475416
 2022-05-27 14:07:57 -07:00
Peter Hawkins
ec41cbe3e9 Bump XLA version. 
PiperOrigin-RevId: 450688054
 2022-05-24 08:32:28 -07: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
Yash Katariya
6b091b2c47
update workspace TF commit 2022-05-17 09:12:18 -07:00
Yash Katariya
40c61cfe62
Update workspace for jaxlib release for testing 2022-05-06 10:54:38 -07:00
Yash Katariya
38ce6d027b Update TF commit for release 
PiperOrigin-RevId: 446555288
 2022-05-04 14:42:50 -07:00
Yash Katariya
ff1a3c40ba jax and jaxlib release 
PiperOrigin-RevId: 446295827
 2022-05-03 14:52:40 -07:00
Peter Hawkins
6b33c55450 2nd attempt at jax/jaxlib 0.3.7 release. 2022-04-15 15:20:02 -04:00
Peter Hawkins
52a97f2e06 Jax 0.3.7 and jaxlib 0.3.7 release. 2022-04-15 12:02:05 -04: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
Peter Hawkins
8b6b736ef3 Revert: Pin a newer @platforms in the Bazel workspace to fix Mac ARM builds. 
The @platforms repository has been updated in the @tf_runtime repository, which was pulling in the old version of @platforms. We no longer need to override @platforms in the JAX WORKSPACE.

PiperOrigin-RevId: 440375016
 2022-04-08 08:49:00 -07:00
Peter Hawkins
96ba290faf Jax 0.3.5 and jaxlib 0.3.5 release. 2022-04-06 23:56:41 +00:00
Peter Hawkins
d073a0f264 Pin a newer @platforms in the Bazel workspace to fix Mac ARM builds. 2022-04-06 14:43:56 +00:00
Robert Suderman
c8fbe7563e Update tensorflow version to 0dfdb8 2022-03-30 18:44:58 +00:00
Skye Wanderman-Milne
057905bddb Update WORKSPACE for jaxlib 0.3.2 release, take 3 
PiperOrigin-RevId: 434789818
 2022-03-15 10:30:36 -07:00
Skye Wanderman-Milne
1ba51786c0 Update WORKSPACE for jaxlib 0.3.2, take 2 2022-03-14 15:00:07 -07:00
Skye Wanderman-Milne
ddd7477d1f Update WORKSPACE for jaxlib 0.3.1 2022-03-11 00:51:30 +00:00
Peter Hawkins
81bb526829 Update XLA. 2022-03-08 14:45:35 +00:00
Yash Katariya
ef11d53ff9
Update TF commit for release 2022-02-10 11:33:20 -08:00
Yash Katariya
cabc98c047
update TF commit for release 2022-02-09 16:34:44 -08:00
Yash Katariya
303eec4e02
Cut a new jaxlib release 2022-02-03 10:06:27 -08:00
Peter Hawkins
be2f6a91ec Update XLA for jaxlib 0.1.76 release. 2022-01-27 14:26:16 +00:00
Stella Laurenzo
a3a6afcd5b Internal change 
PiperOrigin-RevId: 418960726
 2021-12-30 08:11:15 -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
b0b853c503
Update WORKSPACE 2021-12-07 07:04:44 -08:00
Yash Katariya
87e49471c0
Update TF commit for jaxlib release 2021-12-03 04:38:22 -08:00