ROCm/jax

Pull Requests Commits

Switch to hermetic llvm for rocm

alekstheod committed 3 days ago

da25bf17

Rename the benchmarks workflow consistently (#770)

psanal35 committed 11 days ago

Verified eb58ba29

Add placeholder for nightly benchmark workflow (#768)

psanal35 committed 11 days ago

Verified 8d4fbef1

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

psanal35 committed 15 days ago

2422c280

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

WBobby committed 15 days ago

6283bbcf

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

gulsumgudukbay committed 15 days ago

3f5828e9

Abort detection CI workflow (#688)

gulsumgudukbay committed 15 days ago

c21c3b8d

Skipping testEighTinyNorm due to hipSolver issues (#697)

AratiGanesh committed 15 days ago

7601b868

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

magaonka-amd committed 15 days ago

9d4fce1a

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

magaonka-amd committed 15 days ago

3e79165b

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

magaonka-amd committed 15 days ago

d0a11b30

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

magaonka-amd committed 15 days ago

2b8c7fe3

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

magaonka-amd committed 15 days ago

837654d2

Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd committed 15 days ago

130ca422

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

magaonka-amd committed 15 days ago

411d4faa

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

magaonka-amd committed 15 days ago

7ec9fe0f

Update Skip Reason Outputs (#663)

gulsumgudukbay committed 15 days ago

3757b64d

Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh committed 15 days ago

5c681b10

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

magaonka-amd committed 15 days ago

2c12a03c

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

magaonka-amd committed 15 days ago

82a1e816

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

magaonka-amd committed 15 days ago

c2ea7b45

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

magaonka-amd committed 15 days ago

489fcf60

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

magaonka-amd committed 15 days ago

81842f41

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

magaonka-amd committed 15 days ago

f360e135

Enable test_variadic_reduce_window on ROCm (#647)

mminutoli committed 15 days ago

4eb74735

Enable RngShardingTests (#644)

gulsumgudukbay committed 15 days ago

8a9adefd

Ruturaj4 committed 15 days ago

3b3b31cf

Fix Numpy signatures test (#598)

magaonka-amd committed 15 days ago

ad47e174

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

phambinhfin committed 15 days ago

82bf13e1

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

phambinhfin committed 15 days ago

9b5d7088

Older