ROCm/jax

Pull Requests Commits

squashed all commit ci build changes

kiran-thumma committed 1 year ago

f4d64a2f

[ROCm]: Disable EighRankDeficient Test

Rahul Batra committed 2 years ago

86a42556

[ROCm]: Disable source info test

Rahul Batra committed 2 years ago

a8e6a149

[ROCm]: Disable debugger test

Rahul Batra committed 2 years ago

759e6780

[ROCm]: Disable sparse_bcoo_bcsr_test.py::test_bcsr_dot_general for complex types

Rahul Batra committed 2 years ago

b33de545

[ROCm]: Disable sparse bcsr test

Rahul Batra committed 2 years ago

2aba07d5

[ROCm]: Disable some interpretor test for float16

Rahul Batra committed 2 years ago

4dd6dc21

[ROCm]: Disable test for rocm in tests_shape_poly_test

Rahul Batra committed 2 years ago

379eb1dc

[ROCm]: Disable multi-platform tests in export_test

Rahul Batra committed 2 years ago

890adee5

[ROCm]: Disable export_harnesses_multi_platform_test on ROCm

Rahul Batra committed 2 years ago

e1c7529d

[ROCm]: Comment out GPU_LOWERING_ENABLED in sparse_test

Rahul Batra committed 2 years ago

1205ce7d

[ROCm]: Disable sparse tests

invalid-email-address committed 2 years ago

93207de7

Disable linalg_test::EighTinyNorm

Rahul Batra committed 2 years ago

f8320ba6

[ROCm]: Updates hip headers path for ROCm 6.0

Rahul Batra committed 2 years ago

f15f5e04

[ROCm]: Dockerfile updates

Rahul Batra committed 2 years ago

6635af4d

[ROCm]: Lower sparse(some) ops correctly for ROCm

Rahul Batra committed 2 years ago

ebe7fb17

Prepare for emergency jax and jaxlib 0.4.23 release

yashk2810 committed 2 years ago

0017b3a6

Add patch to suppress XLA:GPU logging.

hawkinsp committed 2 years ago

432cd182

Finish jax and jaxlib 0.4.22 release

yashk2810 committed 2 years ago

cccb5650

split the random generalized normal test and skip its K-S half

froystig committed 2 years ago

a85dd62c

Update thinking-in-jax working-with-pytrees

8bitmp3 committed 2 years ago

7edc80d6

Merge pull request #18581 from 8bitmp3:jax-docs-thinking-in-jax

a-googler committed 2 years ago

29ed3cd4

Reverts 3c07c10a9a55f9a32390dd10cf3f420bdf3f1ed8

yashk2810 committed 2 years ago

b52bcc16

Upgrade How to think in JAX

8bitmp3 committed 2 years ago

a4003bf0

Fix the pjit flakey test. The test was weirdly written in the first place. The current suspicion is that `x.copy()` inside jit made it flakey. Also delete some duplicate tests from the time of migrating to jax.Array

yashk2810 committed 2 years ago

e8888065

Start jax and jaxlib 0.4.22 release

yashk2810 committed 2 years ago

2e633522

Remove the `jax_require_devices_during_lowering flag since it was temporary. Added the semi-breaking change to Changelog.md.

yashk2810 committed 2 years ago

3c07c10a

Merge pull request #18869 from 8bitmp3:small-fix-jax-pytrees

a-googler committed 2 years ago

0281596e

Merge pull request #18582 from 8bitmp3:jax-docs-debugging-101

a-googler committed 2 years ago

d269c581

Merge pull request #18961 from mattjj:issue18955

a-googler committed 2 years ago

4459991d

Older