8 Commits

Author	SHA1	Message	Date
Nikita Shulga	8811e4d00d	Add/fix typing annotations to some functions (#39075) ... Summary: Add missing typing imports to some jit tests Add typing annotations to `torch.testing._compare_scalars_internal` and `torch.testing._internal.assertTrue` Pull Request resolved: https://github.com/pytorch/pytorch/pull/39075 Differential Revision: D21882468 Pulled By: malfet fbshipit-source-id: dd9858eb8e11a38411544cc64daf36fced807d76	2020-06-04 13:40:04 -07:00
Will Constable	6fd48e24f1	Add support, test for kwargs in jit._fork (#38357) (#38665) ... Summary: Closing 38357 Pull Request resolved: https://github.com/pytorch/pytorch/pull/38665 Reviewed By: suo Differential Revision: D21643697 Pulled By: wconstab fbshipit-source-id: c292c037f87bc2bb69a4ca163d7107d5396c53a2	2020-05-19 13:02:46 -07:00
Wanchao Liang	c38dcd45d7	[jit] fix return different types bug in tracing module calls (#37190) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/37190 if module call return different types, we need to record them correctly Test Plan: Imported from OSS Differential Revision: D21214871 Pulled By: wanchaol fbshipit-source-id: 46ba98f08ed4ade22f9740cb3fca84b29557e125	2020-04-24 14:49:28 -07:00
Wanchao Liang	5362a0b948	[jit] fix lifting bug in tracing module calls (#37189) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/37189 This fix bug in tracing module calls to correct lift values with its correponding value type, rather than the default tensor type. Test Plan: Imported from OSS Differential Revision: D21214872 Pulled By: wanchaol fbshipit-source-id: f635154851365e2d7b88186d6e47634123eac42f	2020-04-24 14:47:54 -07:00
Shihao Xu	cae6bdf199	[JIT] Mark aten::wait as having side effect, since it can represent RPC message received (#35695) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/35695 aten::wait was optimized out, causing RPC futures are not waited on. Test Plan: ``` buck test mode/dev-nosan //caffe2/torch/fb/distributed/model_parallel/tests:test_dist_optim ``` ``` buck test mode/dev-nosan //caffe2/test/distributed/rpc/jit:rpc_fork buck build mode/dev-nosan //caffe2/test/distributed/rpc/jit:rpc_fork && \ buck-out/gen/caffe2/test/distributed/rpc/jit/rpc_fork\#binary.par \ -r test_python_future_with_jit ``` ``` buck build mode/dev-nosan //caffe2/test:jit && \ buck-out/gen/caffe2/test/jit\#binary.par -r test_trace_fork_wait_inline ``` ``` buck build mode/dev-nosan //caffe2/test:jit && \ buck-out/gen/caffe2/test/jit\#binary.par -r test_trace_fork_wait_inline_onnx ``` Differential Revision: D9562716 fbshipit-source-id: 35b2c971efa42949ffdf0910bd75a927eee8d965	2020-03-31 22:17:25 -07:00
Wanchao Liang	4a84ac5f5d	[jit] make Future type annotation available in Python (#27637) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/27637 Fixes https://github.com/pytorch/pytorch/issues/26578 Test Plan: Imported from OSS Differential Revision: D20626866 fbshipit-source-id: 20d6a3a719fddcb33e0e17a56d7123535fa20d65	2020-03-24 14:36:05 -07:00
Pritam Damania	f050b16dd9	Move pytorch distributed tests to separate folder for contbuild. (#30445) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/30445 Create distributed and rpc directories under caffe/test for better management of unit tests. Differential Revision: D18702786 fbshipit-source-id: e9daeed0cfb846ef68806f6decfcb57c0e0e3606	2020-01-22 21:16:59 -08:00
Michael Suo	cc457ca30f	split remaining "easy" tests (#29249) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/29249 This splits out all the tests that are "easy", leaving `TestJit`, `TestScript`, the autogenerated tests, and a small docs test. Splitting those into reasonable chunks is more effort which is less mechanical. Differential Revision: D18339007 Test Plan: Imported from OSS Pulled By: suo fbshipit-source-id: 69164b9f9a2c379fe8923a846c98dd3c37ccb70e	2019-11-06 13:23:01 -08:00