mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 00:56:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/lib/mlir
History
Sergei Lebedev a755f1db83 Import from `mlir.dialects` lazily 
These imports jointly account for ~0.3s of import time internally.

PiperOrigin-RevId: 662588167
2024-08-13 11:22:41 -07:00
..
dialects
Import from `mlir.dialects` lazily
2024-08-13 11:22:41 -07:00
__init__.py
Switch Mosaic GPU to a custom pass pipeline and improve the lowering of GPU launch
2024-04-24 03:27:45 -07:00