ROCm/jax

Pull Requests Commits

gfx950 platform support

zahiqbal committed 1 year ago

a3864eac

removed commented statement in linalg_test

gulsumgudukbay committed 1 year ago

a0fc9739

unskip tests in tests/lax_numpy_test and tests/pallas/gpy_ops_test.py and fix decorator order in tests/linalg_test

gulsumgudukbay committed 1 year ago

a5c5ac78

fix non-positive determinant error in tests/scipy_spatial_test.py::...::testRotationFromMatrix0 (#362)

gulsumgudukbay committed 1 year ago

Verified 1f85d7db

Fix bytes_limit key of mem_stats (#243)

Ruturaj4 committed 1 year ago

5c681d85

buildkit option for docker build

kiran-thumma committed 1 year ago

f7d84975

Ignore future scipy warning and not treat it as an error

Ruturaj4 committed 1 year ago

71259488

pallas/ops_test health check suppressed for test_select_n

gulsumgudukbay committed 1 year ago

509de786

Sync get-rocm script

mrodden committed 1 year ago

8a92b394

skip sparse_nm tests on ROCm

Ruturaj4 committed 1 year ago

1f0e2dc2

Update XLA commit hash (#342)

charleshofer committed 1 year ago

Verified 0a2f2805

Port PR CI workflow from rocm-main (#312)

charleshofer committed 1 year ago

Verified 73f35188

Fix C++23 build errors (#257) (#318)

mrodden committed 1 year ago

Verified 0074fed2

Fix permissions on /tmp (#320)

gabeweisz committed 1 year ago

Verified 82a2ec71

Update offload ROCm architectures (#311)

Ruturaj4 committed 1 year ago

Verified 7613a24b

Set higher tolerance for failing fp8 tests (#306)

Ruturaj4 committed 1 year ago

Verified 6a539471

Fixes for 0.5.0 build (#308)

mrodden committed 1 year ago

Verified 887d1ede

Add support for ROCm wheel based install (#283)

mrodden committed 1 year ago

Verified 0218c7bd

Fix and enable RNN tests on JAX 0.5.0 branch (#301)

Ruturaj4 committed 1 year ago

Verified 41bcf9f1

Fix the latest XLA has for 0.5.0 JAX branch (#302)

Ruturaj4 committed 1 year ago

Verified ce337284

Fix dev build script (#286)

gulsumgudukbay committed 1 year ago

Verified d9ec58fd

Fix auditwheel version issue (#290)

mrodden committed 1 year ago

Verified 1c96ac20

Deprecate unused pre-production gfx versions (#272)

JehandadKhan committed 1 year ago

Verified a496792a

[pallas:triton] Fix atomic min/max lowering for unsigned integers and float types (#264)

draganmladjenovic committed 1 year ago

Verified 8445686b

removing pytest-csv and removing csv logs from unit testing (#261)

zahiqbal committed 1 year ago

Verified 9ce2aec6

add gfx1101 target (#249) (#254)

JehandadKhan committed 1 year ago

Verified 48a36151

Fix sha256 hash to point to the latest xla commit (#251)

Ruturaj4 committed 1 year ago

Verified 772bd4ed

Skip tests on ROCm (#235)

Ruturaj4 committed 1 year ago

Verified 790fe78b

Fix invalid lowerings for ROCm in Pallas (#223)

mrodden committed 1 year ago

Verified 6b351552

Use HIPBLAS_V2 (#222)

Ruturaj4 committed 1 year ago

Verified 1fc3d157

Older