pytorch/test/cpp/rpc
Pritam Damania ff6e560301 Add C++ end to end test for RPC and distributed autograd. (#36893)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36893

Adding an end to end test for running a simple training loop in C++
for the distributed RPC framework.

The goal of this change is to enable LeakSanitizer and potentially catch memory
leaks in the Future. Enabling LSAN with python multiprocessing is tricky and we
haven't found a solution for this. As a result, adding a C++ test that triggers
most of the critical codepaths would be good for now.

As an example, this unit test would've caught the memory leak fixed by:
https://github.com/pytorch/pytorch/pull/31030
ghstack-source-id: 107781167

Test Plan:
1) Verify the test catches memory leaks.
2) waitforbuildbot

Reviewed By: mrshenli

Differential Revision: D21112208

fbshipit-source-id: 4eb2a6b409253108f6b6e14352e593d250c7a64d
2020-07-15 12:59:19 -07:00
..
CMakeLists.txt Add C++ end to end test for RPC and distributed autograd. (#36893) 2020-07-15 12:59:19 -07:00
e2e_test_base.h Add C++ end to end test for RPC and distributed autograd. (#36893) 2020-07-15 12:59:19 -07:00
test_e2e_process_group.cpp Add C++ end to end test for RPC and distributed autograd. (#36893) 2020-07-15 12:59:19 -07:00
test_tensorpipe_serialization.cpp [TensorPipe] Improve serialization (#39010) 2020-05-28 10:48:24 -07:00
test_wire_serialization.cpp Back out "Revert D21171334: [pytorch][PR] Change StorageImpl to track byte count rather than element count" (#37893) 2020-05-05 22:43:15 -07:00