ROCm/jax

Pull Requests Commits

[ROCM]: enable linalg_test.py::test_tridiagonal_solve.

zahiqbal committed 2 years ago

fb4b491f

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

Finalize a number of deprecations from JAX 0.4.19

Jake VanderPlas committed 2 years ago

9b9aa1ef

Merge pull request #19460 from google:dependabot/github_actions/actions/upload-artifact-4.2.0

a-googler committed 2 years ago

416206b8

Merge pull request #19463 from jakevdp:numpy-upstream

a-googler committed 2 years ago

ca635815

Bump actions/cache from 3.3.3 to 4.0.0

jakevdp committed 2 years ago

3392b642

TST: add unimplemented parameters in numpy signatures test

jakevdp committed 2 years ago

34901b22

Merge pull request #19445 from jakevdp:full-device

a-googler committed 2 years ago

e23ee15d

Older