port rocm/jax 0.9.2 cherrypicks to amd-main branch (#740) #741
magaonka-amd
wants to merge
42 commits into
ROCm:amd-main
from
magaonka-amd:merge/pr740-to-amd-main
Remove nvidia_wheel_versions
6a1daf5d
Make jaxlib targets visible
cb422e0d
hipblas typedef fix
9ce6d54c
No GPU fail
2f63e7b4
Wrap HIP inline functions in anonymous namespaces in vendor.h
e4d5cde2
SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError
a4226671
Add shared utility function get_rocm_version to test_util.py
90a70d0d
Fix hipSparse CSR algorithm mappings for ROCm 7
7be1d70e
Fix v_pages quantization and adjust test params for ROCm compatibilit…
f5d027ae
Address LLVM assertion failure due to a multithreaded use. Update .gi…
e8e9b538
Add skip of test_is_finite() on Cuda (#565)
0b758c6c
Add rocm test requirements file (#570)
c8241809
Let the unit tests use build.py for setting up Bazel commands for uni…
647bb5c9
adding abort logic to rocm/jax (#590)
72f8c81f
Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0) (…
1b9da37b
Fix shared memory limit check for ROCm in test_dot (#596)
49c0e655
Fix Numpy signatures test (#598)
fc75e210
fix merge arts
e087832c
Enable RngShardingTests (#644)
120974f3
Enable test_variadic_reduce_window on ROCm (#647)
88579ce0
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
e23dee16
Update sparse test skip messages in v0.8.2 (#653)
c20c4e9a
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
891eca23
Update sparse test skip messages in v0.8.2 (#653)
3465af49
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
4b235392
Update sparse test skip messages in v0.8.2 (#653)
4edf90fb
Enable testMultivariateNormalSingularCovariance on ROCm (#666)
33e02940
Skip test_tridiagonal_solve on ROCm due to hipSPARSE numerical errors…
cbeac55e
Update Skip Reason Outputs (#663)
1b420973
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
0f037920
Update sparse test skip messages in v0.8.2 (#653)
2234070d
Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)
8bdd8955
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
6fb2109a
Update sparse test skip messages in v0.8.2 (#653)
e9ec1b39
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
4b2b54f4
Update sparse test skip messages in v0.8.2 (#653)
5e0244de
Remove 'mean' from unsupported params for jnp.var (#689)
78cd5430
Skipping testEighTinyNorm due to hipSolver issues (#697)
d5e4bb5b
Abort detection CI workflow (#688)
dcf0715d
Abort-Detection: Fix halt-for-connection input (#712)
1db96e1c
fix: add rocm_sysdeps/lib to wheel RUNPATH (#737)
7d9fd7e1
Cherry-pick fixes for JAX 0.9.2 release from JAX 0.9.1 (#740)
1eb823e8
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub