ROCm/jax

Pull Requests Commits

Add ROCm benchmark workflow for MaxText

psanal35 committed 2 days ago

6f08c2a6

Temporarily disable the cron trigger for the cont. wheel tests workflow

psanal35 committed 28 days ago

ddcf4f41

fix: add rocm_sysdeps/lib to wheel RUNPATH (#737)

WBobby committed 28 days ago

db0dd97b

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

gulsumgudukbay committed 28 days ago

e97e0ed3

Abort detection CI workflow (#688)

gulsumgudukbay committed 28 days ago

f9528df2

Skipping testEighTinyNorm due to hipSolver issues (#697)

AratiGanesh committed 28 days ago

9ee94768

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

magaonka-amd committed 28 days ago

486782bd

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

magaonka-amd committed 28 days ago

7a2494ed

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

magaonka-amd committed 28 days ago

0609c7c0

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

magaonka-amd committed 28 days ago

8e1ebcba

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

magaonka-amd committed 28 days ago

ed839d83

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd committed 28 days ago

9bfe7d66

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

magaonka-amd committed 28 days ago

ff5dfaad

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

magaonka-amd committed 28 days ago

9a5723d5

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 28 days ago

235f616c

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 28 days ago

cef5f06b

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

magaonka-amd committed 28 days ago

68f67b58

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

magaonka-amd committed 28 days ago

9c6283d3

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

magaonka-amd committed 28 days ago

b4de9f9d

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

magaonka-amd committed 28 days ago

6a47a467

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

magaonka-amd committed 28 days ago

3264ecbf

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

magaonka-amd committed 28 days ago

8438836f

Enable test_variadic_reduce_window on ROCm (#647)

mminutoli committed 28 days ago

6121864d

Enable RngShardingTests (#644)

gulsumgudukbay committed 28 days ago

4ec5da21

Ruturaj4 committed 28 days ago

55bbbdfb

Fix Numpy signatures test (#598)

magaonka-amd committed 28 days ago

cda6fb97

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

phambinhfin committed 28 days ago

40323a1a

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

phambinhfin committed 28 days ago

2c1ba3a2

adding abort logic to rocm/jax (#590)

gulsumgudukbay committed 28 days ago

6c52a272

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

charleshofer committed 28 days ago

dba3b37d

Older