mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
20 lines
467 B
C++
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
|