mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
c3f2f3294e
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50977 Adds a `blocking` flag that can be set to False to make this API return a `Future` to the type. This is to make this function non-blocking, mostly for a future change that will allow `rref.rpc_async()` to be completely non-blocking (it currently calls and waits for this function that issues an RPC in-line). ghstack-source-id: 121021433 Test Plan: Modified UT Reviewed By: mrshenli Differential Revision: D25944582 fbshipit-source-id: e3b48a52af2d4578551a30ba6838927b489b1c03 |
||
|---|---|---|
| .. | ||
| codegen | ||
| data | ||
| distributed | ||
| generated | ||
| test_module | ||
| __init__.py | ||
| autocast_test_lists.py | ||
| common_cuda.py | ||
| common_device_type.py | ||
| common_distributed.py | ||
| common_jit.py | ||
| common_methods_invocations.py | ||
| common_nn.py | ||
| common_quantization.py | ||
| common_quantized.py | ||
| common_utils.py | ||
| dist_utils.py | ||
| expecttest.py | ||
| hypothesis_utils.py | ||
| jit_metaprogramming_utils.py | ||
| jit_utils.py | ||
| mypy_wrapper.py | ||
| te_utils.py | ||