jax
b5762449 - Make nvidia version data optional for ROCm builds

Commit

86 days ago

Make nvidia version data optional for ROCm builds

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

#670 - Test pytest abort plugin

#701 - Skip test_tridiagonal_solve_grad test 0.8.0

#708 - Skip test_batch_axis_sharding_jvp13 test 0.8.0

Author

phambinhfin

phambinhfin

Committer

charleshofer

charleshofer

Parents

Loading