jax
f4df93f6
- No GPU fail
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
63 days ago
No GPU fail
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
#587 - Mark ROCm support for gtsv2 as stable
#591 - Adding run_pytest_rocm.sh
#599 - Use xla from rocm-jaxlib-v0.8.0 release branch
#604 - Add proper ROCm skip for cuDNN attention tests
#614 - Implement approx_tanh for ROCm using OCML tanh function
#626 - Enabled triton_pallas_test.py unit test file to run
Author
charleshofer
Committer
charleshofer
Parents
9ed3820b
Loading