mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,443 Commits 500 Branches 263 Tags
Commit Graph

648 Commits

Author SHA1 Message Date
jax authors
86f5de855b Update XLA dependency to use revision 
a52ced2433.

PiperOrigin-RevId: 559407111
 2023-08-23 06:36:17 -07:00
jax authors
a77da34375 Update XLA dependency to use revision 
e3533af05f.

PiperOrigin-RevId: 559090025
 2023-08-22 06:15:25 -07:00
jax authors
a14d64b160 Update XLA dependency to use revision 
c1e30de7d3.

PiperOrigin-RevId: 558765656
 2023-08-21 06:04:36 -07:00
jax authors
b4a628400f Update XLA dependency to use revision 
51c0ad464e.

PiperOrigin-RevId: 558390333
 2023-08-19 05:26:07 -07:00
jax authors
332369225f Update XLA dependency to use revision 
465c945d59.

PiperOrigin-RevId: 558119555
 2023-08-18 05:34:56 -07:00
jax authors
ab9555e084 Update XLA dependency to use revision 
b55511f5fa.

PiperOrigin-RevId: 557800589
 2023-08-17 06:19:19 -07:00
jax authors
e2ebd9a7fb Update XLA dependency to use revision 
c7e022af03.

PiperOrigin-RevId: 557787256
 2023-08-17 05:02:31 -07:00
jax authors
a833652bb8 Update XLA dependency to use revision 
c782fe8356.

PiperOrigin-RevId: 557780339
 2023-08-17 04:35:07 -07:00
jax authors
5ab3e9f6f2 Update XLA dependency to use revision 
2a2bc3d943.

PiperOrigin-RevId: 557777322
 2023-08-17 04:07:20 -07:00
jax authors
4c0664ae3d Update XLA dependency to use revision 
f8dfd1ee60.

PiperOrigin-RevId: 557766683
 2023-08-17 03:11:51 -07:00
jax authors
4f3d9df212 Update XLA dependency to use revision 
08af1e149d.

PiperOrigin-RevId: 557763125
 2023-08-17 02:55:13 -07:00
jax authors
49667e557e Update XLA dependency to use revision 
4acd835d9c.

PiperOrigin-RevId: 557759271
 2023-08-17 02:38:38 -07:00
jax authors
7a5b9ff522 Update XLA dependency to use revision 
225f05fc6d.

PiperOrigin-RevId: 557700420
 2023-08-16 21:58:34 -07:00
jax authors
fcc44fb55e Update XLA dependency to use revision 
3bbf0a0449.

PiperOrigin-RevId: 557696920
 2023-08-16 21:41:49 -07:00
jax authors
1be198c205 Update XLA dependency to use revision 
02d59d5d9a.

PiperOrigin-RevId: 557693150
 2023-08-16 21:21:46 -07:00
jax authors
0045b03078 Update XLA dependency to use revision 
1e81d47ecb.

PiperOrigin-RevId: 557689198
 2023-08-16 20:59:47 -07:00
jax authors
86380b00da Update XLA dependency to use revision 
c3512249ed.

PiperOrigin-RevId: 557686293
 2023-08-16 20:42:01 -07:00
jax authors
82e840c7b2 Update XLA dependency to use revision 
ac00a88b0b.

PiperOrigin-RevId: 557671298
 2023-08-16 19:18:00 -07:00
jax authors
ebd09cc4db Update XLA dependency to use revision 
d79398ede0.

PiperOrigin-RevId: 557668919
 2023-08-16 19:03:24 -07:00
jax authors
9c4ba21a4a Update XLA dependency to use revision 
4ca5589bff.

PiperOrigin-RevId: 557666310
 2023-08-16 18:50:41 -07:00
jax authors
03a0d73bd7 Update XLA dependency to use revision 
28e7569406.

PiperOrigin-RevId: 557651441
 2023-08-16 17:28:43 -07:00
jax authors
124bdb7563 Update XLA dependency to use revision 
51f7513132.

PiperOrigin-RevId: 557647957
 2023-08-16 17:13:41 -07:00
jax authors
3bffa07fe1 Update XLA dependency to use revision 
13bb9d0812.

PiperOrigin-RevId: 557633286
 2023-08-16 16:15:51 -07:00
jax authors
c91f010e6f Update XLA dependency to use revision 
4c2bf94fbb.

PiperOrigin-RevId: 557628169
 2023-08-16 15:57:11 -07:00
jax authors
329791a1bc Update XLA dependency to use revision 
c05021a5bb.

PiperOrigin-RevId: 557623875
 2023-08-16 15:41:24 -07:00
jax authors
e959c045cd Update XLA dependency to use revision 
6785669435.

PiperOrigin-RevId: 557618963
 2023-08-16 15:23:11 -07:00
jax authors
42d5441934 Update XLA dependency to use revision 
b3f12d680d.

PiperOrigin-RevId: 557607869
 2023-08-16 14:46:11 -07:00
jax authors
4cf85b947d Update XLA dependency to use revision 
baaeccccf8.

PiperOrigin-RevId: 557563190
 2023-08-16 12:13:02 -07:00
jax authors
539a6d1e58 Update XLA dependency to use revision 
0cebcdd469.

PiperOrigin-RevId: 557551693
 2023-08-16 11:35:08 -07:00
jax authors
a3cd35183f Update XLA dependency to use revision 
af311aff5c.

PiperOrigin-RevId: 557535177
 2023-08-16 10:46:45 -07:00
jax authors
6eef465b30 Update XLA dependency to use revision 
cd23682502.

PiperOrigin-RevId: 557526553
 2023-08-16 10:29:39 -07:00
jax authors
a1d8787e15 Update XLA dependency to use revision 
2784a58880.

PiperOrigin-RevId: 557520885
 2023-08-16 10:10:45 -07:00
jax authors
14fa0672b6 Update XLA dependency to use revision 
2681564419.

PiperOrigin-RevId: 557515766
 2023-08-16 09:46:21 -07:00
Peter Hawkins
f1a2bbfb1e Update XLA dependency to use revision 
2c817aeb83.

PiperOrigin-RevId: 557489343
 2023-08-16 08:09:24 -07:00
Peter Hawkins
d6e06f4476 Move the XLA commit out of the top-level JAX WORKSPACE file and into a separate .bzl file. 
No functional changes intended.

PiperOrigin-RevId: 556906943
 2023-08-14 14:09:45 -07:00
Antonio Sanchez
a600020346 Update ducc to commit: 2b2cead005e08d2632478e831d7f45da754162dc 
NOTE: this version of DUCC has a breaking change, where the fft.h header
no longer contains the definitions of many fft functions - instead they exist
within fft1d_impl.h and fftnd_impl.h.
PiperOrigin-RevId: 554567641
 2023-08-07 13:06:43 -07:00
jax authors
42ef649e65 Merge pull request #14475 from hawkinsp:openxla 
PiperOrigin-RevId: 516316330
 2023-03-13 14:04:41 -07:00
Peter Hawkins
172a831219 Switch JAX to use the OpenXLA repository. 2023-03-13 18:38:26 +00:00
Peter Hawkins
ba557d5e1b Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.". 
See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.

PiperOrigin-RevId: 476167538
 2022-09-22 12:27:19 -07:00
Peter Hawkins
20475cc55c Add missing header to ducc build file. 2022-08-26 19:30:04 +00:00
jax authors
498fd2083e Merge pull request #12122 from hawkinsp:fft 
PiperOrigin-RevId: 470294824
 2022-08-26 11:32:07 -07: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
Gordian Edenhofer
024ae47e79 Switch from pocketfft to ducc 
All credit goes to Martin Reinecke <martin@mpa-garching.mpg.de>.
 2022-08-26 13:36:25 +00:00
Peter Hawkins
caad556a03 Update PocketFFT archive to point to Github. 
Bazel has difficulty downloading from the gitlab instance, leading to a warning. Use the Github mirror instead.
 2021-05-19 09:24:17 -04:00
Peter Hawkins
1ed321cbcd Update PocketFFT version to fix crash due to undersized aligned allocations. 2021-03-12 08:46:01 -05:00
Skye Wanderman-Milne
ba4f7880a3 Replace use of TF's third_party_http_archive with http_archive in PocketFFT workspace. 
TF got rid of `third_party_http_archive`, and I'm not sure why we couldn't use Bazel's provided `http_archive` instead.
 2021-02-25 17:29:37 -08:00
Peter Hawkins
9b7fd6329b Add PocketFFT mirror to fix download problems for some Bazel versions. 2020-10-23 19:48:00 -04:00
Peter Hawkins
f58f1ee456 [JAX] Use PocketFFT for FFTs on CPU instead of Eigen. 
PocketFFT is the same FFT library used by NumPy (although we are using the C++ variant rather than the C variant.)

For the benchmark in #2952 on my workstation:

Before:
```
907.3490574884647
max:     4.362646594533903e-08
mean:    6.237288307614869e-09
min:     0.0
numpy fft execution time [ms]:   37.088446617126465
jax fft execution time [ms]:     74.93342399597168
```

After:
```
907.3490574884647
max:     1.9057386696477137e-12
mean:    3.9326737908882566e-13
min:     0.0
numpy fft execution time [ms]:   37.756404876708984
jax fft execution time [ms]:     28.128278255462646
```

Fixes https://github.com/google/jax/issues/2952

PiperOrigin-RevId: 338743753
 2020-10-23 14:20:32 -07:00