ROCm/jax

Pull Requests Commits

remove flag for comm splitting.

mminutoli committed 27 days ago

24c32952

fixup! Add download-jax-rocm-wheels composite action

mminutoli committed 27 days ago

eaf81af3

Enable pre-releases in the download command.

mminutoli committed 27 days ago

db6af54f

Disable authentication on gcloud.

mminutoli committed 27 days ago

04400974

Fix image name.

mminutoli committed 27 days ago

df9c9aeb

[TESTING ONLY] Bypass ROCm plugin version check and strip nightly workflow

mminutoli committed 27 days ago

4d4af791

fixup! Add download-jax-rocm-wheels composite action

mminutoli committed 27 days ago

c7a5a2ec

fixup! Add ROCm pytest workflow to nightly/release CI

mminutoli committed 27 days ago

42553c7e

fixup! Add ROCm pytest workflow to nightly/release CI

mminutoli committed 27 days ago

0a12efc0

fixup! Add pytest_rocm.yml workflow for ROCm testing

mminutoli committed 27 days ago

74ed227f

fixup! Add download-jax-rocm-wheels composite action

mminutoli committed 27 days ago

a4df3e0d

fixup! Add pytest_rocm.yml workflow for ROCm testing

mminutoli committed 27 days ago

48320217

fixup! Add ROCm pytest workflow to nightly/release CI

mminutoli committed 27 days ago

1a1a1f53

fixup! Add pytest_rocm.yml workflow for ROCm testing

mminutoli committed 27 days ago

84246843

skip failing tests for ToT

gulsumgudukbay committed 27 days ago

496b8d01

Add ROCm pytest workflow to nightly/release CI

mminutoli committed 27 days ago

0d85e9c8

Add pytest_rocm.yml workflow for ROCm testing

mminutoli committed 27 days ago

b5c4804b

Add download-jax-rocm-wheels composite action

mminutoli committed 27 days ago

606eec60

Add ROCm wheel patterns to install_wheels_locally.sh

mminutoli committed 27 days ago

496630a3

Complete run_pytest_rocm.sh for CI integration

mminutoli committed 27 days ago

10dc836d

adding run_pytest_rocm.sh

gulsumgudukbay committed 27 days ago

61441aad

[Pallas] Fix ROCm GPU architecture detection and route to Triton backend

Ruturaj4 committed 27 days ago

3b825f42

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 27 days ago

e9dda85c

Enable array interoperability tests on ROCm platform (#660)

magaonka-amd committed 27 days ago

10f0b01a

Skip test_tridiagonal_solve on ROCm due to hipSPARSE numerical errors (#668)

AratiGanesh committed 27 days ago

5c09fa3b

Skip test_batch_axis_sharding_jvp because of hipSPARSE issue (#667)

AratiGanesh committed 27 days ago

109e971b

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 27 days ago

9dde5956

Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd committed 27 days ago

17ffc6cd

Skip sparse tests on ROCm due to hipSPARSE issue (#652)

magaonka-amd committed 27 days ago

337dfccd

Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd committed 27 days ago

9fdc7cb5

Older