ROCm/jax

Pull Requests Commits

_scaled_matmul lowering for rocm

shurale-nkn committed 83 days ago

3346ee4f

Abort-Detection: Fix halt-for-connection input (#712)

gulsumgudukbay committed 85 days ago

a0cd2621

Abort detection CI workflow (#688)

gulsumgudukbay committed 85 days ago

eec4197a

Skipping testEighTinyNorm due to hipSolver issues (#697)

AratiGanesh committed 85 days ago

85e784dd

Remove 'mean' from unsupported params for jnp.var (#689)

magaonka-amd committed 85 days ago

ee3d2b3c

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

magaonka-amd committed 85 days ago

f97e88e8

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

magaonka-amd committed 85 days ago

766e867b

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

magaonka-amd committed 85 days ago

6dba3c80

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

magaonka-amd committed 85 days ago

7957da31

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd committed 85 days ago

9d1d676e

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

magaonka-amd committed 85 days ago

00496a95

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

magaonka-amd committed 85 days ago

54ae608b

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 85 days ago

9c1e27b1

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

AratiGanesh committed 85 days ago

cb7f8899

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 85 days ago

0f2cba5d

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

magaonka-amd committed 85 days ago

98e03add

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

magaonka-amd committed 85 days ago

31a8e11a

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

magaonka-amd committed 85 days ago

49474b06

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

magaonka-amd committed 85 days ago

5af4aa7b

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

magaonka-amd committed 85 days ago

3d2c74f8

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

magaonka-amd committed 85 days ago

3ddfe507

Enable test_variadic_reduce_window on ROCm (#647)

mminutoli committed 85 days ago

28108b7c

Enable RngShardingTests (#644)

gulsumgudukbay committed 85 days ago

af4d148e

Ruturaj4 committed 85 days ago

03708730

Fix Numpy signatures test (#598)

magaonka-amd committed 85 days ago

0001fb9b

Fix shared memory limit check for ROCm in test_dot (#596)

phambinhfin committed 85 days ago

9f4587c5

Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0) (#597)

phambinhfin committed 85 days ago

83c22830

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 85 days ago

2c5421fa

Let the unit tests use build.py for setting up Bazel commands for unit tests (#582)

charleshofer committed 85 days ago

ced263bc

Add rocm test requirements file (#570)

AratiGanesh committed 85 days ago

2ec72233

Older