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

74 Commits

Author SHA1 Message Date
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
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