ROCm/jax

Pull Requests Commits

Fix C++23 build errors (#257) (#318)

mrodden committed 331 days ago

85e80e42

Fix permissions on /tmp (#320)

gabeweisz committed 331 days ago

7aa324ce

Update offload ROCm architectures (#311)

Ruturaj4 committed 331 days ago

ca4cfe37

Set higher tolerance for failing fp8 tests (#306)

Ruturaj4 committed 331 days ago

defaf1a5

Fixes for 0.5.0 build (#308)

mrodden committed 331 days ago

52912e63

Add support for ROCm wheel based install (#283)

mrodden committed 331 days ago

52f0201e

Fix and enable RNN tests on JAX 0.5.0 branch (#301)

Ruturaj4 committed 331 days ago

367c08c1

Fix the latest XLA has for 0.5.0 JAX branch (#302)

Ruturaj4 committed 331 days ago

6dee2704

disable c23-extension warning so clang-19 works

gabeweisz committed 340 days ago

3d4f91d0

Fix dev build script (#286)

gulsumgudukbay committed 344 days ago

Verified d9ec58fd

Fix auditwheel version issue (#290)

mrodden committed 344 days ago

Verified 1c96ac20

Deprecate unused pre-production gfx versions (#272)

JehandadKhan committed 348 days ago

Verified a496792a

[pallas:triton] Fix atomic min/max lowering for unsigned integers and float types (#264)

draganmladjenovic committed 351 days ago

Verified 8445686b

removing pytest-csv and removing csv logs from unit testing (#261)

zahiqbal committed 356 days ago

Verified 9ce2aec6

add gfx1101 target (#249) (#254)

JehandadKhan committed 356 days ago

Verified 48a36151

Fix sha256 hash to point to the latest xla commit (#251)

Ruturaj4 committed 359 days ago

Verified 772bd4ed

Skip tests on ROCm (#235)

Ruturaj4 committed 1 year ago

Verified 790fe78b

Fix invalid lowerings for ROCm in Pallas (#223)

mrodden committed 1 year ago

Verified 6b351552

Use HIPBLAS_V2 (#222)

Ruturaj4 committed 1 year ago

Verified 1fc3d157

Merge pull request #215 from ROCm/5.0_xla_sha_fix

zahiqbal committed 1 year ago

Verified b6843ff8

Update workspace.bzl

Ruturaj4 committed 1 year ago

Verified 1c9f9073

Update XLA hash for 0.5.0 branch (#212)

Ruturaj4 committed 1 year ago

Verified cdf1ef5a

Merge pull request #211 from ROCm/jax_5.0_xla_sha_fix

zahiqbal committed 1 year ago

Verified 81ac86ba

Update XLA hash for 0.5.0 branch

Ruturaj4 committed 1 year ago

b51b0c76

Release JAX 0.5.0

hawkinsp committed 1 year ago

c25fb92c

Reverts f1b894d14a28ac22a037fb79177b991275c75a18

Google-ML-Automation committed 1 year ago

a527aba6

[sharding_in_types] Error out for reshape for splits like this: `(4, 6, 8)` -> `(4, 4, 2, 6)`

yashk2810 committed 1 year ago

ce85b898

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

7cac76d3

[Mosaic GPU] Delete unused declarations of `mosaic_gpu_memcpy_async_h2d`.

bchetioui committed 1 year ago

d3be190e

[mosaic_gpu] Added a serialization pass

superbobry committed 1 year ago

d34c40f6

Older