Files
pytorch/test/cpp/rpc/e2e_test_base.cpp
Luca Wehrstedt 8493b0d5d6 Enroll TensorPipe agent in C++-only E2E test (#42680)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/42680

ghstack-source-id: 109544678

Test Plan: CI

Reviewed By: mrshenli

Differential Revision: D22978714

fbshipit-source-id: 04d6d190c240c6ead9bd9f3b7f3a5f964d7451e8
2020-08-13 07:07:30 -07:00

20 lines
467 B
C++

#include "e2e_test_base.h"
namespace torch {
namespace distributed {
namespace rpc {
DistAutogradContainer* getDistAutogradContainer() {
static DistAutogradContainer* autogradContainer =
&DistAutogradContainer::init(0);
return autogradContainer;
}
const char* TestE2EBase::serverAddress = "127.0.0.1";
const size_t TestE2EBase::numIters = 100;
const size_t TestE2EBase::numWorkers = 1;
} // namespace rpc
} // namespace distributed
} // namespace torch