pytorch
7c9e78fd - [TensorPipe] Add options for agent, including backend killswitches (#40162)

Commit
4 years ago
[TensorPipe] Add options for agent, including backend killswitches (#40162) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/40162 The only public option is `num_worker_threads`. The other ones are private (as indicated by the leading underscore, is that enough?) and allow to specify a different set and order of transports/channels. These can thus be used to disable a backend (by not specifying it) or by forcing one (by raising its priority). They can therefore be used to work around defective backends, in case we'll find any post-release. ghstack-source-id: 106103238 Test Plan: Built //caffe2:ifbpy and, using TensorPipe's verbose logging, verified that the transports/channels I specified were indeed the ones that were being registered. Differential Revision: D22090661 fbshipit-source-id: 789bbe3bde4444cfa20c40276246e4ab67c50cd0
Author
lw lw
Parents
Loading