jax
7cee4db6
- [Mosaic GPU] Add support for f8 types for WGMMA with lhs in registers under WG semantics.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
11 days ago
[Mosaic GPU] Add support for f8 types for WGMMA with lhs in registers under WG semantics. PiperOrigin-RevId: 855254612
References
#31768 - [ROCm] Support lowering through PJRT_Triton_Extension
#592 - Upstream ci download jax rocm wheels
#34304 - [ROCm] Add ROCm platform support for psend/precv collective operations
#34450 - [ROCm] Upstream ROCm CI Nightly Wheel Testing
#34470 - [ROCm] Enabled testTridiagonal for ROCm devices.
#34474 - [ROCm] Added support for GESVDJ on ROCm devices
#34494 - [ROCm] Enabled ROCm devices to default to Jacobi SVD on smaller matrices.
#34561 - [ROCm] Enable ToeplitzSymmetricConstruction and condition number tests
#34567 - [ROCm]Unskip testMultivariateNormalSingularCovariance on ROCm
#34568 - [ROCm] Skip test_tridiagonal_solve on ROCm
#34575 - [ROCm] Enabled the Triton Pallas tests to run for ROCm.
#34577 - [ROCm] Enabled RNN unit test "test_no_workspace_overflow" for ROCm.
#34578 - [ROCm] Unskip supported dtypes for testConvolutionsPreferredElementType on ROCm
#34598 - Implement approx_tanh for ROCm using OCML tanh function
#34600 - Enable scaled_matmul tests on ROCm
#34601 - Enable neural network tests on ROCm
#34602 - Fix and enable Pallas ops tests on ROCm
#34603 - Enable miscellaneous tests on ROCm
#34615 - [ROCm] Skip test_batch_axis_sharding_jvp on ROCm
Author
allanrenucci
Committer
Google-ML-Automation
Parents
1242ba32
Loading