Antonio Kim
|
bab1304f59
|
Add step closures (#84300)
Ports over the step closure functionality from PyTorch/XLA to Lazy Tensor Core:
References:
205ae574c0/torch_xla/core/xla_model.py (L852-L900)
205ae574c0/torch_xla/utils/closures.py (L7-L83)
CC: @wconstab @JackCaoG @Krovatkin
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84300
Approved by: https://github.com/JackCaoG, https://github.com/wconstab
|
2022-09-06 20:55:34 +00:00 |
|