jax
eaf6efc1 - [Pallas] Fix cost estimate einsum bug.

Commit

45 days ago

[Pallas] Fix cost estimate einsum bug. PiperOrigin-RevId: 839859433

References

#31768 - [ROCm] Support lowering through PJRT_Triton_Extension

#33628 - [Pallas] Fix cost estimate einsum bug.

#34135 - [ROCm] update to test if there are GPU events when doing profiling on…

#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

Author

justinjfu

justinjfu

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading