Aaron Orenstein
|
db4ce78d46
|
PEP585: More UP006 fixes (#146392)
This should be the final PR before we can enable RUFF UP006.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/146392
Approved by: https://github.com/justinchuby, https://github.com/albanD, https://github.com/Skylion007
|
2025-02-20 06:18:13 +00:00 |
|
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 |
|
Gagan Jain
|
a6287b5c27
|
Fixing issue in move pass for copying Parameter (#138855)
Summary: Fixing bug for Parameter copy during move pass of exported graph.
Test Plan:
UT
runs on APS models.
Differential Revision: D64876951
Pull Request resolved: https://github.com/pytorch/pytorch/pull/138855
Approved by: https://github.com/pianpwk
Co-authored-by: Gagan Jain <gaganj@meta.com>
|
2024-10-25 17:57:27 +00:00 |
|
Yiming Zhou
|
2cfc2da527
|
[export] Make move_to_device_pass function public (#134263)
Summary:
This is a follow-up of https://github.com/pytorch/pytorch/pull/133660
Here we make the `move_to_device_pass()` function publich so users can call it by `from torch.export.passes import move_to_device_pass`
Test Plan: CI
Differential Revision: D61671310
Pull Request resolved: https://github.com/pytorch/pytorch/pull/134263
Approved by: https://github.com/angelayi
|
2024-08-23 23:18:30 +00:00 |
|