ROCm/jax

Pull Requests Commits

[ROCm]: Disable sparse bcsr test

Rahul Batra committed 2 years ago

0bee51e1

[ROCm]: Disable some interpretor test for float16

Rahul Batra committed 2 years ago

a981cc22

[ROCm]: Disable test for rocm in tests_shape_poly_test

Rahul Batra committed 2 years ago

942bf6e7

[ROCm]: Disable multi-platform tests in export_test

Rahul Batra committed 2 years ago

9dede147

[ROCm]: Disable export_harnesses_multi_platform_test on ROCm

Rahul Batra committed 2 years ago

6950003d

[ROCm]: Comment out GPU_LOWERING_ENABLED in sparse_test

Rahul Batra committed 2 years ago

38aaa976

[ROCm]: Disable sparse tests

invalid-email-address committed 2 years ago

29293061

Disable linalg_test::EighTinyNorm

Rahul Batra committed 2 years ago

1b09dca9

[ROCm]: Update build script

Rahul Batra committed 2 years ago

ad002ba3

Prepare for 0.4.21 release

yashk2810 committed 2 years ago

5942e158

Merge pull request #18795 from gnecula:test_export_grad

a-googler committed 2 years ago

d91c13e9

[export] Add and fix a test for exporting higher-order gradients with sharding

gnecula committed 2 years ago

8a2d4a01

Disable export_harnesses_multi_platform_test under sanitizers.

hawkinsp committed 2 years ago

1d95e79f

Merge pull request #18794 from olupton:qualname

a-googler committed 2 years ago

5e0993c2

Use qualified name if possible.

olupton committed 2 years ago

3c0c6b78

Update XLA dependency to use revision

a-googler committed 2 years ago

a137edc4

Update XLA dependency to use revision

a-googler committed 2 years ago

bf084113

Merge pull request #18786 from gnecula:test_export_effects

a-googler committed 2 years ago

61e79cd4

Merge pull request #18761 from gnecula:export_sharding

a-googler committed 2 years ago

b51b80eb

[export] Improve the testing of exporting with effects

gnecula committed 2 years ago

bd7c1aa7

[export] Simplify the handling of shardings in Exported.

gnecula committed 2 years ago

3eb3e2dc

Merge pull request #18783 from gnecula:fix_indexing

a-googler committed 2 years ago

b8228014

Remove the ml_program MLIR dialect from jaxlib.

hawkinsp committed 2 years ago

32fb1b40

Fix bug in indexing with slices that overflow, and add tests.

gnecula committed 2 years ago

d2f62612

Put the input creating via `jnp` outside the pjit cpp cache miss counting code.

yashk2810 committed 2 years ago

04183707

Rewrite `tf.aliasing_output` when wrapping the main StableHLO function

junwhanahn committed 2 years ago

965ba054

Update XLA dependency to use revision

a-googler committed 2 years ago

1aab108c

Reverts f0382a5838f4526d21631e804f6fe576bfc3f97e

yashk2810 committed 2 years ago

f0bc7e0f

Re-enable passing tests on TPU

majnemer committed 2 years ago

86661c85

Merge pull request #18757 from jakevdp:astype

a-googler committed 2 years ago

e61f7a81

Older