jax
b5762449
- Make nvidia version data optional for ROCm builds
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
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
Committer
charleshofer
Parents
28e04303
Loading