bobrenjc93
|
53e4d7b6a2
|
remove allow-untyped-defs for torch/_lazy/device_context.py (#143367)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143367
Approved by: https://github.com/aorenste
ghstack dependencies: #143366
|
2024-12-17 18:54:03 +00:00 |
|
Oguz Ulgen
|
72d2dba992
|
Add None return type to init (#132335)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/132335
Approved by: https://github.com/albanD
|
2024-08-01 15:26:45 +00:00 |
|
Aaron Orenstein
|
afe15d2d2f
|
Flip default value for mypy disallow_untyped_defs [3/11] (#127840)
See #127836 for details.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/127840
Approved by: https://github.com/oulgen
|
2024-06-08 18:28:01 +00:00 |
|
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 |
|