Fix batched Cholesky OOM on ROCm by bypassing hipSolver's external hipMalloc#717

Open

FlemingH wants to merge 43 commits intoROCm:amd-mainfrom

FlemingH:amd-main

Commits on Feb 24, 2026

Remove nvidia_wheel_versions

charleshofer
authored and
mminutoli
committed
Make jaxlib targets visible

charleshofer
authored and
mminutoli
committed
hipblas typedef fix

charleshofer
authored and
mminutoli
committed
No GPU fail

charleshofer
authored and
mminutoli
committed
Wrap HIP inline functions in anonymous namespaces in vendor.h
mminutoli
committed
SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError

dsicarov-amd
authored and
mminutoli
committed
Add shared utility function get_rocm_version to test_util.py

charleshofer
authored and
mminutoli
committed
Fix hipSparse CSR algorithm mappings for ROCm 7

phambinhfin
authored and
mminutoli
committed
Fix v_pages quantization and adjust test params for ROCm compatibilit… (#560)

phambinhfin
authored and
mminutoli
committed
Address LLVM assertion failure due to a multithreaded use. Update .gitignore (#563)

Arech8
authored and
mminutoli
committed
Add skip of test_is_finite() on Cuda (#565)

Arech8
authored and
mminutoli
committed
Add rocm test requirements file (#570)

AratiGanesh
authored and
mminutoli
committed
Let the unit tests use build.py for setting up Bazel commands for unit tests (#582)

charleshofer
authored and
mminutoli
committed
adding abort logic to rocm/jax (#590)

gulsumgudukbay
authored and
mminutoli
committed
Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0) (#597)

phambinhfin
authored and
mminutoli
committed
Fix shared memory limit check for ROCm in test_dot (#596)

phambinhfin
authored and
mminutoli
committed
Fix Numpy signatures test (#598)

authored and
mminutoli
committed
fix merge arts

Ruturaj4
authored and
mminutoli
committed
Enable RngShardingTests (#644)

gulsumgudukbay
authored and
mminutoli
committed
Enable test_variadic_reduce_window on ROCm (#647)
mminutoli
committed
Skip sparse tests on ROCm due to hipSPARSE issue (#652)

magaonka-amd
authored and
mminutoli
committed
Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd
authored and
mminutoli
committed
Skip sparse tests on ROCm due to hipSPARSE issue (#652)

magaonka-amd
authored and
mminutoli
committed
Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd
authored and
mminutoli
committed
Skip sparse tests on ROCm due to hipSPARSE issue (#652)

magaonka-amd
authored and
mminutoli
committed
Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd
authored and
mminutoli
committed
Enable testMultivariateNormalSingularCovariance on ROCm (#666)

AratiGanesh
authored and
mminutoli
committed
Skip test_tridiagonal_solve on ROCm due to hipSPARSE numerical errors (#668)

AratiGanesh
authored and
mminutoli
committed
Update Skip Reason Outputs (#663)

gulsumgudukbay
authored and
mminutoli
committed
Skip sparse tests on ROCm due to hipSPARSE issue (#652)

magaonka-amd
authored and
mminutoli
committed
Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd
authored and
mminutoli
committed
Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)

magaonka-amd
authored and
mminutoli
committed
Skip sparse tests on ROCm due to hipSPARSE issue (#652)

magaonka-amd
authored and
mminutoli
committed
Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd
authored and
mminutoli
committed
Skip sparse tests on ROCm due to hipSPARSE issue (#652)

magaonka-amd
authored and
mminutoli
committed
Update sparse test skip messages in v0.8.2 (#653)

magaonka-amd
authored and
mminutoli
committed
Add ROCm encoding for test_struct_encoding_determinism (#683)

AratiGanesh
authored and
mminutoli
committed
Remove 'mean' from unsupported params for jnp.var (#689)

magaonka-amd
authored and
mminutoli
committed
Implement approx_tanh for ROCm using OCML tanh function (#691)

magaonka-amd
authored and
mminutoli
committed
Skipping testEighTinyNorm due to hipSolver issues (#697)

AratiGanesh
authored and
mminutoli
committed
Abort detection CI workflow (#688)

gulsumgudukbay
authored and
mminutoli
committed
Abort-Detection: Fix halt-for-connection input (#712)

gulsumgudukbay
authored and
mminutoli
committed

Commits on Mar 2, 2026

Fix batched Cholesky OOM on ROCm by routing hipSOLVER workspace through XLA allocator
FlemingH
committed