jax
accb7199
- [XLA:MGPU] Port the entire TiledLayout class to C++.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 day ago
[XLA:MGPU] Port the entire TiledLayout class to C++. Reverts a21715c3f30cdfbd032c173b5fa5f6b32c8294d1 PiperOrigin-RevId: 865304029
References
#34598 - [ROCm] Implement approx_tanh for ROCm using OCML tanh function
#34600 - [ROCm] Enable scaled_matmul tests on ROCm
#34601 - [ROCm] Enable neural network tests on ROCm
#34602 - [ROCm] Fix and enable Pallas ops tests on ROCm
#34603 - [ROCm] Enable miscellaneous tests on ROCm
#34635 - [XLA:MGPU] Port the entire TiledLayout class to C++.
#34689 - [ROCm] Fix ROCm GPU architecture detection and route to Triton backend
#34822 - [ROCm] Added back compatibility test for hipsolver_sytrd
#34829 - [ROCm] Add ROCm LU solver to backward compatibility tests
#34832 - [ROCm] Add ROCm support to annotate_device_placement backward compatibility …
#34862 - [ROCm] Added back compatibility test for hipsolver_geqrf
Author
golechwierowicz
Committer
Google-ML-Automation
Parents
05d9eec3
Loading