Remove nvidia_wheel_versions - SemanticDiff

Commit

64 days ago

Remove nvidia_wheel_versions

References

#556 - Porting v0.7.1 to v0.8.0

#579 - Create rocm-test-requirements.txt

#581 - Fix/pallas tests shared memory

#584 - Use plain bazel to test jax, use hermetic rocm dependency

#585 - update a test for checking zero ROCm GPU event

#591 - Adding run_pytest_rocm.sh

#599 - Use xla from rocm-jaxlib-v0.8.0 release branch

#634 - Skip tridiagonal solve tests on ROCm

#636 - Skip test_get_vmap for slice-only case with mismatched batch dims

#639 - Enable testMultivariateNormalSingularCovariance on ROCm

#640 - Enable nn tests on ROCm

Author

charleshofer

charleshofer

Committer

charleshofer

charleshofer

Parents

Loading