xref: /aosp_15_r20/external/pytorch/test/cpp/rpc/e2e_test_base.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include "e2e_test_base.h"
2 
3 namespace torch {
4 namespace distributed {
5 namespace rpc {
6 
getDistAutogradContainer()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