ROCm/jax

Pull Requests Commits

[ROCm]: Add support to continue on fail

Rahul Batra committed 2 years ago

6cbedd2b

Merge pull request #19489 from skye:version

a-googler committed 2 years ago

f67d4823

Update version numbers for 0.4.24 release

skye committed 2 years ago

368f594c

[xla:gpu] Make cu_threefry2x32 custom call compatible with command buffers

anlunx committed 2 years ago

88f5eaca

Merge pull request #19484 from jakevdp:full-tests

a-googler committed 2 years ago

85fc4a95

[Pallas] Add support for trivial vmap of scalar prefetch

sharadmv committed 2 years ago

037bc5ed

Test: move lax.full sharding tests to multi_device_test

jakevdp committed 2 years ago

8476b44a

Merge pull request #19481 from gnecula:poly_changelog

a-googler committed 2 years ago

7e80aac3

[shape_poly] Forgot to update CHANGELOG for #19235.

gnecula committed 2 years ago

af306680

Merge pull request #19473 from jakevdp:shard-map-run

a-googler committed 2 years ago

c2863726

Merge pull request #19285 from gnecula:poly_constraints

a-googler committed 2 years ago

8708d009

The Triton autotuner ignores configs that use too much shmem

a-googler committed 2 years ago

5761e393

[shape_poly] Add support for symbolic constraints on dimension variables

gnecula committed 2 years ago

24201ef9

Exclude modules with host callbacks from inter-host sharing.

a-googler committed 2 years ago

34d22fc4

Update XLA dependency to use revision

a-googler committed 2 years ago

f4b483cf

Merge pull request #19474 from mattjj:shmap-disable-jit-fix

a-googler committed 2 years ago

507239d0

[shard-map] fix disable_jit-of-shmap

mattjj committed 2 years ago

5275ef6e

DOC: run shard_map docs as part of the doc build

jakevdp committed 2 years ago

a3abd178

Merge pull request #19466 from jakevdp:full-like-sharding

a-googler committed 2 years ago

70aec849

Merge pull request #19472 from mattjj:shmap-tutorial-typos

a-googler committed 2 years ago

f21022b1

Merge pull request #19438 from sharadmv:add-init

a-googler committed 2 years ago

1df92c95

[shard-map] fix typos in docs

mattjj committed 2 years ago

a61f7807

[Pallas/TPU] Add vmem limit test

sharadmv committed 2 years ago

76554bc9

[XLA:Python] Fail with an AttributeError if __cuda_array_interface__ is called on a sharded array.

hawkinsp committed 2 years ago

dfda948f

Dedupe shardings before passing them to _get_and_check_device_assignment

superbobry committed 2 years ago

46f796b3

Merge pull request #19455 from ROCmSoftwarePlatform:rocm-jaxlib-docker-rocm6.0-updates

a-googler committed 2 years ago

8226ff38

Disable NumPy interoperability DLPack test for bools for NumPy versions older than 1.25.

hawkinsp committed 2 years ago

71ee3abe

Disabled lax_scipy_special_functions_test under ASAN on GPU.

hawkinsp committed 2 years ago

25e4acfe

lax.full_like: add sharding argument

jakevdp committed 2 years ago

e76b87ea

Merge pull request #19464 from jakevdp:actions-cache

a-googler committed 2 years ago

a415b567

Older