jax
b77014fd
- [Pallas] Fix ROCm GPU architecture detection and route to Triton backend
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
123 days ago
[Pallas] Fix ROCm GPU architecture detection and route to Triton backend
References
#671 - Rocm jaxlib v0.9.0
#672 - [ROCm][Pallas] Fix ROCm GPU architecture detection and route to Triton backend
#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
Ruturaj4
Committer
Ruturaj4
Parents
98cc66ca
Loading