pytorch
7492e98c - [Tensorpipe Agent] RPC, RRef tests for Tensorpipe Agent (#38444)

Commit
4 years ago
[Tensorpipe Agent] RPC, RRef tests for Tensorpipe Agent (#38444) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/38444 This enables the RPC/RRef test suites to run with the Tensorpipe RPC Agent. This creates a new fixture to ensure the backend/options used are Tensorpipe, as well as a decorator to skip tests that Tensorpipe currently cannot support due to missing functionality. One small note: the decorator function is a class method of the test class so we can check whether `self.rpc_backend` is tensorpipe. In the class-scope, the `TEST_CONFIG.rpc_backend_name` string is set to Tensorpipe, but outside the class scope, it is PGA, possibly due to importing dist_utils which sets this config to PGA by default. The cleanest solution would be to refactor the backend selection to be more uniform (since currently every backend is set slightly differently), but that would be a longer-term fix. ghstack-source-id: 104321885 Test Plan: Note: A couple of these tests will fail right now due to missing features. I've skipped the ones that regularly fail, but there will be some flaky tests that still fail occasionally. The decorator `@_skip_if_tensorpipe_agent` skips the tests that fail with the Tensorpipe Agent. Remove this decorator from above the tests once they are fixed. Differential Revision: D21412016 fbshipit-source-id: 1e801ac5ccaf87974dd4df92d556895b01468bf3
Author
Parents
Loading