1 | #include "e2e_test_base.h" |
---|---|
2 | |
3 | namespace torch { |
4 | namespace distributed { |
5 | namespace rpc { |
6 | |
7 | DistAutogradContainer* getDistAutogradContainer() { |
8 | static DistAutogradContainer* autogradContainer = |
9 | &DistAutogradContainer::init(0); |
10 | return autogradContainer; |
11 | } |
12 | |
13 | const char* TestE2EBase::serverAddress = "127.0.0.1"; |
14 | const size_t TestE2EBase::numIters = 100; |
15 | const size_t TestE2EBase::numWorkers = 1; |
16 | |
17 | } // namespace rpc |
18 | } // namespace distributed |
19 | } // namespace torch |
20 |