jax
6bd1a15e
- Let the unit tests use build.py for setting up Bazel commands for unit tests (#582)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
102 days ago
Let the unit tests use build.py for setting up Bazel commands for unit tests (#582)
References
#664 - update with ROCm profiling tests with kernel_details
#671 - Rocm jaxlib v0.9.0
#699 - Skip testEighTinyNorm test 0.9.0
#725 - Skip test_pmap on ROCm due to IndivisibleError with new pmap SPMD tiling
#732 - update test
#734 - Add ROCm gesdd SVD backend, export back-compat data, and relaxed GPU …
#739 - Fix: Scaled Matmul at rocm 9.0
#765 - jaxlib/gpu/rnn_kernels.cc: remove premature input_tensor_desc destruction (ROCM-21859)
Author
charleshofer
Committer
Ruturaj4
Parents
835856e0
Loading