jax
Fix batched Cholesky OOM on ROCm by bypassing hipSolver's external hipMalloc
#717
Open
Open
Fix batched Cholesky OOM on ROCm by bypassing hipSolver's external hipMalloc #717
FlemingH
wants to merge
43 commits into
ROCm:amd-main
from
FlemingH:amd-main
Remove nvidia_wheel_versions
097ff78e
Make jaxlib targets visible
19a87c5b
hipblas typedef fix
35b23689
No GPU fail
9bf2dbf6
Wrap HIP inline functions in anonymous namespaces in vendor.h
7d1708e4
SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError
30d7f94d
Add shared utility function get_rocm_version to test_util.py
a5377e54
Fix hipSparse CSR algorithm mappings for ROCm 7
db30afa3
Fix v_pages quantization and adjust test params for ROCm compatibilit…
a44f9428
Address LLVM assertion failure due to a multithreaded use. Update .gi…
01746ea7
Add skip of test_is_finite() on Cuda (#565)
f5555631
Add rocm test requirements file (#570)
8cf787ad
Let the unit tests use build.py for setting up Bazel commands for uni…
17e6022f
adding abort logic to rocm/jax (#590)
b600136d
Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0) (…
02399d0b
Fix shared memory limit check for ROCm in test_dot (#596)
0959b0fe
Fix Numpy signatures test (#598)
b43ca180
fix merge arts
cdb5bcb5
Enable RngShardingTests (#644)
de1ef417
Enable test_variadic_reduce_window on ROCm (#647)
d8179cdf
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
c5016efc
Update sparse test skip messages in v0.8.2 (#653)
4e6626ea
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
694e861d
Update sparse test skip messages in v0.8.2 (#653)
12e07fb9
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
76e576f6
Update sparse test skip messages in v0.8.2 (#653)
237e5ad8
Enable testMultivariateNormalSingularCovariance on ROCm (#666)
da3a3cce
Skip test_tridiagonal_solve on ROCm due to hipSPARSE numerical errors…
06d459eb
Update Skip Reason Outputs (#663)
c30a4494
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
58ce4e11
Update sparse test skip messages in v0.8.2 (#653)
e8307d2c
Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)
64ee74ec
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
f1441325
Update sparse test skip messages in v0.8.2 (#653)
9dd1698e
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
fd1195e3
Update sparse test skip messages in v0.8.2 (#653)
4af53277
Add ROCm encoding for test_struct_encoding_determinism (#683)
bfe02082
Remove 'mean' from unsupported params for jnp.var (#689)
44b8a6c3
Implement approx_tanh for ROCm using OCML tanh function (#691)
7bd4a132
Skipping testEighTinyNorm due to hipSolver issues (#697)
95ae9faa
Abort detection CI workflow (#688)
e355fcd6
Abort-Detection: Fix halt-for-connection input (#712)
d36ebc22
hawkinsp
commented
on 2026-02-27
Fix batched Cholesky OOM on ROCm by routing hipSOLVER workspace throu…
5b647ae9
Login to write a write a comment.
Login via GitHub