Aaron Orenstein
|
b6c5562c1f
|
PEP585 update - torch/export (#145165)
See #145101 for details.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/145165
Approved by: https://github.com/bobrenjc93
|
2025-01-19 20:56:55 +00:00 |
|
Edward Z. Yang
|
612122af8f
|
Fix type-safety of torch.nn.Module instances (#141240)
Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/141240
Approved by: https://github.com/Skylion007, https://github.com/malfet
|
2024-11-22 00:05:05 +00:00 |
|
angelayi
|
f14f245747
|
[export] Remove custom forward func in swap (#139126)
Differential Revision: [D65100694](https://our.internmc.facebook.com/intern/diff/D65100694)
Remove the custom forward function and instead move the pytree flatten/unflatten ops into the graph. This allows us to natively run via the interpreter.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/139126
Approved by: https://github.com/avikchaudhuri
|
2024-10-30 16:50:57 +00:00 |
|
angelayi
|
1dc1b85714
|
[export] Move swap to a different file (#137134)
Refactor so that unflattener doesn't become too messy
Differential Revision: [D63719648](https://our.internmc.facebook.com/intern/diff/D63719648/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/137134
Approved by: https://github.com/avikchaudhuri
ghstack dependencies: #136191, #137102
|
2024-10-06 04:28:18 +00:00 |
|