jax
[Testing do not merge] Add ROCm wheel build and test pipeline to continuous CI
#723
Open
Open
[Testing do not merge] Add ROCm wheel build and test pipeline to continuous CI #723
alekstheod
wants to merge
58 commits into
amd-main
from
implement_periodic_build_wheels_pipeline_rocm_internal_v2
Remove nvidia_wheel_versions
2fe53184
Make jaxlib targets visible
c7ec794c
hipblas typedef fix
789df45d
No GPU fail
d0e32075
Wrap HIP inline functions in anonymous namespaces in vendor.h
caf56b60
SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError
d8deb09f
Add shared utility function get_rocm_version to test_util.py
6eee40dd
Fix hipSparse CSR algorithm mappings for ROCm 7
3128e0e7
Fix v_pages quantization and adjust test params for ROCm compatibilit…
cfddfa60
Address LLVM assertion failure due to a multithreaded use. Update .gi…
cb904b38
Add skip of test_is_finite() on Cuda (#565)
b78fe9a6
Add rocm test requirements file (#570)
514fec32
Let the unit tests use build.py for setting up Bazel commands for uni…
b64a5e40
adding abort logic to rocm/jax (#590)
640e4a21
Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0) (…
6eb4fdf5
Fix shared memory limit check for ROCm in test_dot (#596)
14b0125c
Fix Numpy signatures test (#598)
92211a5d
fix merge arts
b202e1f4
Enable RngShardingTests (#644)
4c60b2b0
Enable test_variadic_reduce_window on ROCm (#647)
22ac32aa
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
8ecdd8b7
Update sparse test skip messages in v0.8.2 (#653)
cc543662
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
1b6c9e72
Update sparse test skip messages in v0.8.2 (#653)
f0e90395
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
0cc3b988
Update sparse test skip messages in v0.8.2 (#653)
47321964
Enable testMultivariateNormalSingularCovariance on ROCm (#666)
e5714928
Skip test_tridiagonal_solve on ROCm due to hipSPARSE numerical errors…
84ad452e
Update Skip Reason Outputs (#663)
3c295779
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
7648e11b
Update sparse test skip messages in v0.8.2 (#653)
e0d2d1e9
Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)
a186d625
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
9983dd00
Update sparse test skip messages in v0.8.2 (#653)
c45c49a9
Skip sparse tests on ROCm due to hipSPARSE issue (#652)
14d1b171
Update sparse test skip messages in v0.8.2 (#653)
ec394330
Add ROCm encoding for test_struct_encoding_determinism (#683)
d18c7571
Remove 'mean' from unsupported params for jnp.var (#689)
717cd20f
Implement approx_tanh for ROCm using OCML tanh function (#691)
02ae6c5b
Skipping testEighTinyNorm due to hipSolver issues (#697)
f2f9dd18
Abort detection CI workflow (#688)
6957e42d
Abort-Detection: Fix halt-for-connection input (#712)
7d684aa5
Add ROCm wheel build and test pipeline to continuous CI
1c480676
Restore docker
c732a777
alekstheod
changed the title
Trigger CI/CD pipeline
aa979f98
Trigger CI/CD pipeline
f2b145ca
alekstheod
changed the title
Trigger CI/CD pipeline
5555a769
alekstheod
changed the title
Trigger change
4b93a667
Trigger CI/CD pipeline
4a374169
Switch docker
250595b7
Add mising file
9b75f92b
Add missing wildcard
76ac8c03
Remove audit step
3965854a
Add step installing aws cli
232efa22
Merge branch 'main' of github.com:jax-ml/jax into implement_periodic_…
a4958bb5
Trigger CI/CD pipeline
23bf7d9d
Switch to oidc
9a3bb3b0
Trigger CI/CD pipeline
b4f4b37c
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub