Adam J. Stewart
dfde877c0b
Add type hints for a few random functions/classes
...
Adds type hints for a few functions/classes that we use in [TorchGeo](https://github.com/microsoft/torchgeo ).
Pull Request resolved: https://github.com/pytorch/pytorch/pull/74171
Approved by: https://github.com/jbschlosser , https://github.com/anjali411
2022-05-04 13:53:00 +00:00
PyTorch MergeBot
80fe96c860
Revert "Add type hints for a few random functions/classes"
...
This reverts commit cdb40eb528https://github.com/pytorch/pytorch/pull/74171 on behalf of https://github.com/zengk95
2022-04-21 21:07:15 +00:00
Adam J. Stewart
cdb40eb528
Add type hints for a few random functions/classes
...
Adds type hints for a few functions/classes that we use in [TorchGeo](https://github.com/microsoft/torchgeo ).
Pull Request resolved: https://github.com/pytorch/pytorch/pull/74171
Approved by: https://github.com/jbschlosser
2022-04-21 20:09:40 +00:00
Calvin McCarter
bdf439a958
Adds _LazyInstanceNorm and LazyInstanceNormXd ( #60982 )
...
Summary:
Signed-off-by: Calvin McCarter <calvin@lightmatter.co>
Fixes https://github.com/pytorch/pytorch/issues/60981
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60982
Reviewed By: albanD
Differential Revision: D29810547
Pulled By: jbschlosser
fbshipit-source-id: d933d4c7fe5cf7be9b09a5ab93f740b94cf08cc1
2021-07-21 06:45:45 -07:00
Hameer Abbasi
46e4b2dbda
Convert assert -> cast. ( #57458 )
...
Summary:
Fixes https://github.com/pytorch/pytorch/issues/55868 .
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57458
Reviewed By: mruberry
Differential Revision: D28365745
Pulled By: walterddr
fbshipit-source-id: 35cc3fa85f87b0ef98cf970f620ab909d240c7be
2021-05-12 13:54:16 -07:00
Sam Estep
75024e228c
Add lint for unqualified type: ignore ( #56290 )
...
Summary:
The other half of https://github.com/pytorch/pytorch/issues/56272 .
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56290
Test Plan:
CI should pass on the tip of this PR, and we know that the lint works because the following CI runs (before this PR was finished) failed:
- https://github.com/pytorch/pytorch/runs/2384511062
- https://github.com/pytorch/pytorch/actions/runs/765036024
Reviewed By: seemethere
Differential Revision: D27867219
Pulled By: samestep
fbshipit-source-id: e648f07b6822867e70833e23ddafe7fb7eaca235
2021-04-21 08:07:23 -07:00
Pritam Damania
f39f258dfd
Ensure DDP + Pipe works with find_unused_parameters. ( #49908 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/49908
As described in https://github.com/pytorch/pytorch/issues/49891 , DDP +
Pipe doesn't work with find_unused_parameters.
This PR adds a simple fix to enable this functionality. This only currently
works for Pipe within a single host and needs to be re-worked once we support
cross host Pipe.
ghstack-source-id: 119573413
Test Plan:
1) unit tests added.
2) waitforbuildbot
Reviewed By: rohan-varma
Differential Revision: D25719922
fbshipit-source-id: 948bcc758d96f6b3c591182f1ec631830db1b15c
2021-01-11 16:52:37 -08:00
Pritam Damania
9d91360b5d
Cleanup APIs for pipeline parallelism. ( #48630 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/48630
1) Make torch.distributed.pipeline package public.
2) Make several helper methods private.
ghstack-source-id: 118820803
Test Plan: waitforbuildbot
Reviewed By: rohan-varma
Differential Revision: D25235688
fbshipit-source-id: c32833ebf090ddbd4eaf06fcb5e3f9d421623a60
2020-12-18 15:17:13 -08:00
