ROCm/jax

Pull Requests Commits

skip failing tests for ToT

gulsumgudukbay committed 123 days ago

efe53955

Strip down workflow to execute only rocm piece.

mminutoli committed 123 days ago

467d6edf

Add ROCm pytest workflow to nightly/release CI

mminutoli committed 123 days ago

3dd869aa

Add pytest_rocm.yml workflow for ROCm testing

mminutoli committed 123 days ago

ca5fc7ea

Add download-jax-rocm-wheels composite action

mminutoli committed 123 days ago

40bc0185

Add ROCm wheel patterns to install_wheels_locally.sh

mminutoli committed 123 days ago

1988617a

Complete run_pytest_rocm.sh for CI integration

mminutoli committed 123 days ago

2b54786a

adding run_pytest_rocm.sh

gulsumgudukbay committed 123 days ago

2fe0dc10

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

Ruturaj4 committed 123 days ago

8ccded06

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 123 days ago

7b2e61b5

Enable array interoperability tests on ROCm platform (#660)

magaonka-amd committed 123 days ago

e02db514

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

AratiGanesh committed 123 days ago

919cf613

Skip test_batch_axis_sharding_jvp because of hipSPARSE issue (#667)

AratiGanesh committed 123 days ago

3ec6a7d8

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 123 days ago

356166b7

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

magaonka-amd committed 123 days ago

cf50ecc5

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

magaonka-amd committed 123 days ago

af0f7f97

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

magaonka-amd committed 123 days ago

6eefa201

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

magaonka-amd committed 123 days ago

f6c1fac4

Fix test_cuda_array_interface test skip condition (#657)

magaonka-amd committed 123 days ago

bd6c64f6

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

magaonka-amd committed 123 days ago

665eadb5

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

magaonka-amd committed 123 days ago

d7eb9791

Added changes from PR #626 and PR #645. This also fixes merge conflicts from upstream.

tsrw2048 committed 123 days ago

53486775

Enabled RNN unit test: test_no_workspace_overflow for ROCm devices (#624)

tsrw2048 committed 123 days ago

48b3d9fa

Enabled test for condition number on ROCm devices. (#613)

tsrw2048 committed 123 days ago

730d7120

Unskip supported dtypes for testConvolutionsPreferredElementType (#649)

gulsumgudukbay committed 123 days ago

78edd928

Enable test_variadic_reduce_window on ROCm (#647)

magaonka-amd committed 123 days ago

2cb7590c

Enable reduce_window tests on ROCm (#643)

magaonka-amd committed 123 days ago

645e6b15

Enable RngShardingTests (#644)

gulsumgudukbay committed 123 days ago

75c929e7

Backport cuda_array_interface , testDotAlgorithm, Optimizer test fixes from rocm-jaxlib-v0.8.0 (#635)

magaonka-amd committed 123 days ago

acc7af87

Fix KeyError for bytes_reservable_limit on ROCm

magaonka-amd committed 123 days ago

ac3e1609

Older