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
14,859 Commits 508 Branches 263 Tags
Commit Graph

271 Commits

Author SHA1 Message Date
Yash Katariya
58e46b48e6 Prepare for jax and jaxlib 0.4.4 release 
PiperOrigin-RevId: 510152471
 2023-02-16 08:37:15 -08:00
Skye Wanderman-Milne
8ab158574d Update WORKSPACE and setup.py for jax/jaxlib 0.4.3 release 2023-02-07 15:45:28 -08:00
Skye Wanderman-Milne
9a7f29a9fb Update WORKSPACE for jaxlib 0.4.2 release (again) 
This fixes a segfault that was affecting the previous 0.4.2 candidate.
 2023-01-24 18:07:11 +00:00
Skye Wanderman-Milne
3f4bd5f449 Updates for jax + jaxlib 0.4.2 release 2023-01-20 19:04:46 +00:00
Yash Katariya
c4d590b1b6 Update values for release 0.4.1 
PiperOrigin-RevId: 494889744
 2022-12-12 19:04:38 -08:00
Yash Katariya
0118f8d568 Prepare for jax and jaxlib 0.4.0 release 
PiperOrigin-RevId: 493733609
 2022-12-07 16:02:24 -08:00
Jake VanderPlas
e66fe1dff4 Update XLA commit. 
Fixes build error: no such target '@org_tensorflow//tensorflow/compiler/xla/mlir_hlo:python/MlirHloModule.cc'
 2022-11-16 15:51:28 -08:00
Yash Katariya
50f7512a6e
Update WORKSPACE for jaxlib release 2022-11-14 16:37:09 -08:00
Yash Katariya
2e384ce58f Prepare for release of jax and jaxlib 0.3.24 
PiperOrigin-RevId: 485985460
 2022-11-03 15:13:23 -07:00
Skye Wanderman-Milne
63be2201aa Update WORKSPACE and setup.py for jaxlib 0.3.22 release 2022-10-11 17:55:05 +00:00
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