jax
accb7199 - [XLA:MGPU] Port the entire TiledLayout class to C++.

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

golechwierowicz

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading