jax
Fixes for CI Nightly Bazel Workflow
#743
Closed

Fixes for CI Nightly Bazel Workflow #743

tsrw2048 wants to merge 43 commits into amd-main from ci-bazel-test-fix
tsrw2048
charleshofer Remove nvidia_wheel_versions
b28b4779
charleshofer Make jaxlib targets visible
22da2c2e
charleshofer hipblas typedef fix
b8961e3c
charleshofer No GPU fail
fad31022
mminutoli Wrap HIP inline functions in anonymous namespaces in vendor.h
7e698f1a
dsicarov-amd SWDEV-512768 - Replace hipGetLastError with hipExtGetLastError
b406dbc8
charleshofer Add shared utility function get_rocm_version to test_util.py
c2d73d18
phambinhfin Fix hipSparse CSR algorithm mappings for ROCm 7
759f4458
phambinhfin Fix v_pages quantization and adjust test params for ROCm compatibilit…
52e87274
Arech8 Address LLVM assertion failure due to a multithreaded use. Update .gi…
b533c1c9
Arech8 Add skip of test_is_finite() on Cuda (#565)
fc587529
AratiGanesh Add rocm test requirements file (#570)
3b6aab39
charleshofer Let the unit tests use build.py for setting up Bazel commands for uni…
c9cfe7ae
gulsumgudukbay adding abort logic to rocm/jax (#590)
8303140f
phambinhfin Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0) (…
7628bff8
phambinhfin Fix shared memory limit check for ROCm in test_dot (#596)
6fa95347
magaonka-amd Fix Numpy signatures test (#598)
d3a731a7
Ruturaj4 fix merge arts
17e10b1c
gulsumgudukbay Enable RngShardingTests (#644)
7b7f106f
mminutoli Enable test_variadic_reduce_window on ROCm (#647)
fd03607e
magaonka-amd Skip sparse tests on ROCm due to hipSPARSE issue (#652)
c21d1230
magaonka-amd Update sparse test skip messages in v0.8.2 (#653)
4b02bff7
magaonka-amd Skip sparse tests on ROCm due to hipSPARSE issue (#652)
6106f30b
magaonka-amd Update sparse test skip messages in v0.8.2 (#653)
5a7dfb3d
magaonka-amd Skip sparse tests on ROCm due to hipSPARSE issue (#652)
605d3eab
magaonka-amd Update sparse test skip messages in v0.8.2 (#653)
0b74afb0
AratiGanesh Enable testMultivariateNormalSingularCovariance on ROCm (#666)
f0722943
AratiGanesh Skip test_tridiagonal_solve on ROCm due to hipSPARSE numerical errors…
4390cfba
gulsumgudukbay Update Skip Reason Outputs (#663)
0559c4e1
magaonka-amd Skip sparse tests on ROCm due to hipSPARSE issue (#652)
fc1ae633
magaonka-amd Update sparse test skip messages in v0.8.2 (#653)
566db14b
magaonka-amd Skip testCudaArrayInterfaceOnNonCudaFails on ROCm platform (#677)
c6748183
magaonka-amd Skip sparse tests on ROCm due to hipSPARSE issue (#652)
d6fe37cd
magaonka-amd Update sparse test skip messages in v0.8.2 (#653)
c693f9f3
magaonka-amd Skip sparse tests on ROCm due to hipSPARSE issue (#652)
9b8c58a7
magaonka-amd Update sparse test skip messages in v0.8.2 (#653)
736275f8
magaonka-amd Remove 'mean' from unsupported params for jnp.var (#689)
eea3302f
AratiGanesh Skipping testEighTinyNorm due to hipSolver issues (#697)
54510f25
gulsumgudukbay Abort detection CI workflow (#688)
3fa5fa33
gulsumgudukbay Abort-Detection: Fix halt-for-connection input (#712)
6c448c55
WBobby fix: add rocm_sysdeps/lib to wheel RUNPATH (#737)
3a5fb555
tsrw2048 tsrw2048 requested a review from mminutoli mminutoli 47 days ago
tsrw2048 Added fix to Bazel ROCm CI script which will download JAX wheels from…
75a21d7f
tsrw2048 Bazel workflow on ROCm now correctly processes freethreaded Python ve…
927d3bb4
tsrw2048 tsrw2048 force pushed from 9a73a2be to 927d3bb4 46 days ago
psanal35 psanal35 force-pushed the amd-main branch from 96ce5393 to ddcf4f41 40 days ago
mminutoli mminutoli force-pushed the amd-main branch from ddcf4f41 to 2422c280 12 days ago
tsrw2048
tsrw2048 tsrw2048 closed this 10 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone