jax
fd7de07b - [Pallas:MGPU] Make sure the ue8m0 reciprocal optimization kicks in

Commit

1 day ago

[Pallas:MGPU] Make sure the ue8m0 reciprocal optimization kicks in PiperOrigin-RevId: 881382732

References

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

#35761 - [Pallas:MGPU] Make sure the ue8m0 reciprocal optimization kicks in

#35785 - [ROCm] Fix and simplify jax rocm plugin init script

#35821 - [ROCm] Use rocm plugin and pjrt wheels for pytest job in continuous wheel build pipeline

Author

apaszke

apaszke

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading