jax
2eba8d26 - Don't dump the MLIR module unless we have debug logging enabled in export.py

Commit

14 days ago

Don't dump the MLIR module unless we have debug logging enabled in export.py PiperOrigin-RevId: 899130336

References

#36733 - Don't dump the MLIR module unless we have debug logging enabled in export.py

#36909 - [ROCm] Skip unit tests in `testEighIdentity` because of hipSolver issue.

#37071 - [ROCm] Switch to hermetic llvm

#37072 - [ROCm] Update rules_ml_toolchain reference

#37085 - Upgrade upstream ROCm CI from 7.2.0 to 7.2.2

#37186 - [ROCm] aiter mha kernels (ASM+CK) integration (#747)

#37198 - [ROCm] Introduce shim script for rocm jax upstream CI

Author

hawkinsp

hawkinsp

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading