ROCm/jax

Pull Requests Commits

[ROCm]: Disable EighRankDeficient Test

Rahul Batra committed 1 year ago

485d06a2

[ROCm]: Disable source info test

Rahul Batra committed 1 year ago

28110c72

[ROCm]: Disable debugger test

Rahul Batra committed 1 year ago

d9c7ea54

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

Rahul Batra committed 1 year ago

f98aa2b2

[ROCm]: Disable sparse bcsr test

Rahul Batra committed 1 year ago

6552ade4

[ROCm]: Disable some interpretor test for float16

Rahul Batra committed 1 year ago

0c88db51

[ROCm]: Disable test for rocm in tests_shape_poly_test

Rahul Batra committed 1 year ago

811a1c61

[ROCm]: Disable multi-platform tests in export_test

Rahul Batra committed 1 year ago

cedb4956

[ROCm]: Disable export_harnesses_multi_platform_test on ROCm

Rahul Batra committed 1 year ago

d684d859

[ROCm]: Comment out GPU_LOWERING_ENABLED in sparse_test

Rahul Batra committed 1 year ago

47e855b4

[ROCm]: Disable sparse tests

invalid-email-address committed 1 year ago

e1b2dedd

Disable linalg_test::EighTinyNorm

Rahul Batra committed 1 year ago

fe9bcfb5

[ROCm]: Add support to continue on fail

Rahul Batra committed 1 year ago

eedc453f

[ROCM]: Generating pytest html logs from unit-tests.

zahiqbal committed 1 year ago

fc5ab90a

[ROCm]: Updates hip headers path for ROCm 6.0

Rahul Batra committed 1 year ago

f15f5e04

[ROCm]: Dockerfile updates

Rahul Batra committed 1 year ago

6635af4d

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

Rahul Batra committed 1 year ago

ebe7fb17

Prepare for emergency jax and jaxlib 0.4.23 release

yashk2810 committed 1 year ago

0017b3a6

Add patch to suppress XLA:GPU logging.

hawkinsp committed 1 year ago

432cd182

Finish jax and jaxlib 0.4.22 release

yashk2810 committed 1 year ago

cccb5650

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

froystig committed 1 year 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

Older