ROCm/jax

Pull Requests Commits

Add rocm perf yml file

Ruturaj4 committed 228 days ago

d6c595d3

Add is_device_rocm() check to CSR matmat and matvec tests

psanal35 committed 257 days ago

Verified 3c4fd69b

Revert "Add is_device_rocm() check to CSR matmat and matvec test skip conditi…"

mrodden committed 261 days ago

e51fbf06

Add is_device_rocm() check to CSR matmat and matvec test skip conditions (follow-up to #380)

psanal35 committed 261 days ago

3a824c12

Generalize dtype check for future-proofing and fix import order

psanal35 committed 268 days ago

5582e2aa

Use ROCM_PATH if set with fallback to /opt/rocm, replace assert, note fix in ROCm 6.4.0

psanal35 committed 268 days ago

6fb1f413

Skip CSR matmat and matvec float tests on ROCm <6.4 (NaN issue with beta==0)

psanal35 committed 268 days ago

0e9e57cf

removed commented statement in linalg_test

gulsumgudukbay committed 271 days 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 271 days ago

a5c5ac78

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

gulsumgudukbay committed 275 days ago

Verified 1f85d7db

Fix bytes_limit key of mem_stats (#243)

Ruturaj4 committed 276 days ago

5c681d85

buildkit option for docker build

kiran-thumma committed 276 days ago

f7d84975

Ignore future scipy warning and not treat it as an error

Ruturaj4 committed 276 days ago

71259488

pallas/ops_test health check suppressed for test_select_n

gulsumgudukbay committed 276 days ago

509de786

Sync get-rocm script

mrodden committed 276 days ago

8a92b394

skip sparse_nm tests on ROCm

Ruturaj4 committed 277 days ago

1f0e2dc2

Update XLA commit hash (#342)

charleshofer committed 285 days ago

Verified 0a2f2805

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

charleshofer committed 290 days ago

Verified 73f35188

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

mrodden committed 297 days ago

Verified 0074fed2

Fix permissions on /tmp (#320)

gabeweisz committed 297 days ago

Verified 82a2ec71

Update offload ROCm architectures (#311)

Ruturaj4 committed 298 days ago

Verified 7613a24b

Set higher tolerance for failing fp8 tests (#306)

Ruturaj4 committed 299 days ago

Verified 6a539471

Fixes for 0.5.0 build (#308)

mrodden committed 302 days ago

Verified 887d1ede

Add support for ROCm wheel based install (#283)

mrodden committed 302 days ago

Verified 0218c7bd

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

Ruturaj4 committed 303 days ago

Verified 41bcf9f1

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

Ruturaj4 committed 303 days ago

Verified ce337284

Fix dev build script (#286)

gulsumgudukbay committed 306 days ago

Verified d9ec58fd

Fix auditwheel version issue (#290)

mrodden committed 306 days ago

Verified 1c96ac20

Deprecate unused pre-production gfx versions (#272)

JehandadKhan committed 310 days ago

Verified a496792a

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

draganmladjenovic committed 313 days ago

Verified 8445686b

Older