jax
d37f6c8d
- [Pallas:MGPU] Fix signature of print_layout methods
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
45 days ago
[Pallas:MGPU] Fix signature of print_layout methods When transforms are non-empty, their args will be placed before fmt, not after. PiperOrigin-RevId: 959025302
References
#39712 - [Pallas:MGPU] Fix signature of print_layout methods
Author
apaszke
Committer
Google-ML-Automation
Parents
be0dd453
Loading