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

97 Commits

Author SHA1 Message Date
Peter Hawkins
295cecd505 Update XLA dependency to use revision 
e73af4223d.

PiperOrigin-RevId: 571067534
 2023-10-05 10:54:50 -07:00
Skye Wanderman-Milne
b9c602b8d7 Update XLA commit and versions for jax 0.4.17 release 2023-10-03 13:20:25 -07:00
jax authors
72407f6f4d Update XLA dependency to use revision 
7a19856d74.

PiperOrigin-RevId: 570300307
 2023-10-03 00:37:15 -07:00
jax authors
90286f2fe1 Update XLA dependency to use revision 
2233d81842.

PiperOrigin-RevId: 569975863
 2023-10-02 00:58:05 -07:00
jax authors
004e67e1e8 Update XLA dependency to use revision 
9938bdbbf3.

PiperOrigin-RevId: 569829166
 2023-10-01 01:30:44 -07:00
jax authors
bf46b7427f Update XLA dependency to use revision 
f62cc8f27a.

PiperOrigin-RevId: 569686162
 2023-09-30 01:01:55 -07:00
jax authors
88fac56fd0 Update XLA dependency to use revision 
99a225c1b9.

PiperOrigin-RevId: 569418541
 2023-09-29 01:14:27 -07:00
jax authors
2d068a1caa Update XLA dependency to use revision 
0c71a63a86.

PiperOrigin-RevId: 569108876
 2023-09-28 02:12:36 -07:00
jax authors
350b35cbf5 Update XLA dependency to use revision 
6b5136043c.

PiperOrigin-RevId: 568775379
 2023-09-27 01:18:00 -07:00
jax authors
bfc3193da0 Update XLA dependency to use revision 
c4b1450d53.

PiperOrigin-RevId: 568456910
 2023-09-26 01:20:18 -07:00
jax authors
06e87253a0 Update XLA dependency to use revision 
0dc9def695.

PiperOrigin-RevId: 568150266
 2023-09-25 02:04:28 -07:00
jax authors
034c54609c Update XLA dependency to use revision 
26a84cf3bc.

PiperOrigin-RevId: 567976646
 2023-09-24 01:51:04 -07:00
jax authors
e84be656fc Update XLA dependency to use revision 
fbd9c73f0f.

PiperOrigin-RevId: 567820717
 2023-09-23 01:17:31 -07:00
jax authors
705412ccba Update XLA dependency to use revision 
ae24abc144.

PiperOrigin-RevId: 567555060
 2023-09-22 01:39:30 -07:00
jax authors
f304eb89b9 Update XLA dependency to use revision 
ed31887b05.

PiperOrigin-RevId: 567220157
 2023-09-21 01:09:14 -07:00
Peter Hawkins
8c70288b83 Refer to CUDA stubs directly from TSL, rather than using an alias defined in xla/stream_executor. 
Remove the aliases in xla/stream_executor.

PiperOrigin-RevId: 567025507
 2023-09-20 11:21:54 -07:00
jax authors
f8432da5d9 Update XLA dependency to use revision 
44683beb41.

PiperOrigin-RevId: 566886834
 2023-09-20 01:37:41 -07:00
jax authors
3b66fbf841 Update XLA dependency to use revision 
f71cb9db3b.

PiperOrigin-RevId: 566550177
 2023-09-19 01:23:30 -07:00
Yash Katariya
d477b92121 Start jax and jaxlib 0.4.16 release 
PiperOrigin-RevId: 566354738
 2023-09-18 11:05:11 -07:00
jax authors
152af70ab0 Update XLA dependency to use revision 
70ae45e230.

PiperOrigin-RevId: 566222153
 2023-09-18 01:01:57 -07:00
jax authors
bcc545a692 Update XLA dependency to use revision 
fb9e7064da.

PiperOrigin-RevId: 566049124
 2023-09-17 01:00:35 -07:00
jax authors
bf40f75bd5 Update XLA dependency to use revision 
83d9984673.

PiperOrigin-RevId: 565895458
 2023-09-16 01:59:46 -07:00
jax authors
c0a2f74aee Update XLA dependency to use revision 
a0f1d69d2e.

PiperOrigin-RevId: 565609926
 2023-09-15 01:40:29 -07:00
jax authors
cab68db752 Update XLA dependency to use revision 
f33da8b05c.

PiperOrigin-RevId: 565291862
 2023-09-14 01:08:59 -07:00
jax authors
5a15ba90db Update XLA dependency to use revision 
551a620daf.

PiperOrigin-RevId: 564978874
 2023-09-13 02:10:36 -07:00
jax authors
462c0bd30d Update XLA dependency to use revision 
cd7379d9af.

PiperOrigin-RevId: 564660543
 2023-09-12 02:59:25 -07:00
jax authors
bfc12bdda9 Update XLA dependency to use revision 
96beae40a1.

PiperOrigin-RevId: 564327668
 2023-09-11 03:36:30 -07:00
jax authors
292deef6fd Update XLA dependency to use revision 
c227585959.

PiperOrigin-RevId: 563982388
 2023-09-09 04:01:14 -07:00
Peter Hawkins
c3515231b0 Add -lm to Linux flatbuffer builds. 2023-09-08 12:57:46 -04:00
jax authors
601d67ae66 Update XLA dependency to use revision 
cb2f349863.

PiperOrigin-RevId: 563710408
 2023-09-08 04:14:45 -07:00
Peter Hawkins
dbf13252f0 Copybara import of the project: 
--
3905d6123bdc22f505934242363fda426c99c4cf by Peter Hawkins <phawkins@google.com>:

Update flatbuffers.

Use upstream flatbuffer bazel scripts, with a couple of small patches to fix:
* https://github.com/google/flatbuffers/issues/8087 (remove npm references)
* https://github.com/google/flatbuffers/pull/8088 (fix flatc build failure due to main() removal by linker)

COPYBARA_INTEGRATE_REVIEW=https://github.com/google/jax/pull/17502 from hawkinsp:fb 3905d6123bdc22f505934242363fda426c99c4cf
PiperOrigin-RevId: 563543954
 2023-09-07 14:27:25 -07:00
jax authors
e4b9e5cdb5 Update XLA dependency to use revision 
a05649b19f.

PiperOrigin-RevId: 563389073
 2023-09-07 04:36:16 -07:00
jax authors
fae98733aa Update XLA dependency to use revision 
694b1de5e7.

PiperOrigin-RevId: 563272353
 2023-09-06 18:22:40 -07:00
jax authors
cf6ac58e60 Update XLA dependency to use revision 
2af57548fa.

PiperOrigin-RevId: 563065072
 2023-09-06 04:38:51 -07:00
Michael Hudgins
1ac7e75ee6 Move XLA dependency to using tf_http_archive to enable the ability for bazel to use a GCS mirror for the XLA dependency. 
PiperOrigin-RevId: 562783418
 2023-09-05 08:16:01 -07:00
jax authors
daa8ec2c4b Update XLA dependency to use revision 
09ee9c8ff8.

PiperOrigin-RevId: 562746431
 2023-09-05 05:21:10 -07:00
jax authors
172ad93881 Update XLA dependency to use revision 
ed8139cd8a.

PiperOrigin-RevId: 562529695
 2023-09-04 05:04:46 -07:00
jax authors
da87d78ddd Update XLA dependency to use revision 
7bd5f555f0.

PiperOrigin-RevId: 562181905
 2023-09-02 04:51:09 -07:00
jax authors
0ddbe76cba Update XLA dependency to use revision 
7b9aa07fe9.

PiperOrigin-RevId: 561928858
 2023-09-01 05:35:23 -07:00
jax authors
6b4fbe9b71 Update XLA dependency to use revision 
449d45dbcc.

PiperOrigin-RevId: 561630195
 2023-08-31 05:54:20 -07:00
jax authors
c1680aacbf Update XLA dependency to use revision 
61e12edca8.

PiperOrigin-RevId: 561315877
 2023-08-30 05:52:12 -07:00
jax authors
ba8c4c3086 Update XLA dependency to use revision 
db4cc73e3f.

PiperOrigin-RevId: 561015160
 2023-08-29 06:46:00 -07:00
jax authors
7b802dcd2e Update XLA dependency to use revision 
4f675c5715.

PiperOrigin-RevId: 560691873
 2023-08-28 06:30:55 -07:00
jax authors
66c0400152 Update XLA dependency to use revision 
1c22df6753.

PiperOrigin-RevId: 560496105
 2023-08-27 06:22:28 -07:00
jax authors
08ca945271 Update XLA dependency to use revision 
7a371ed44a.

PiperOrigin-RevId: 560339802
 2023-08-26 06:52:36 -07:00
Peter Hawkins
9a5b808853 Update nanobind version to 1.5.2. 2023-08-25 14:45:32 +00:00
jax authors
87cec1a6e8 Update XLA dependency to use revision 
14c0e8565a.

PiperOrigin-RevId: 560077922
 2023-08-25 06:51:19 -07:00
Peter Hawkins
70b7d50181 Switch jaxlib to use nanobind instead of pybind11. 
nanobind has a number of advantages (https://nanobind.readthedocs.io/en/latest/why.html), notably speed of compilation and dispatch, but the main reason to do this for these bindings is because nanobind can target the Python Stable ABI starting with Python 3.12. This means that we will not need to ship per-Python version CUDA plugins starting with Python 3.12.

PiperOrigin-RevId: 559898790
 2023-08-24 16:07:56 -07:00
jax authors
3f60993b27 Update XLA dependency to use revision 
116ceaeccd.

PiperOrigin-RevId: 559729371
 2023-08-24 06:03:10 -07:00
jax authors
86f5de855b Update XLA dependency to use revision 
a52ced2433.

PiperOrigin-RevId: 559407111
 2023-08-23 06:36:17 -07:00