Searched refs:DistEngine (Results 1 – 8 of 8) sorted by relevance
27 class TORCH_API DistEngine {30 static DistEngine& getInstance();59 DistEngine(const DistEngine&) = delete;60 DistEngine& operator=(const DistEngine&) = delete;61 DistEngine(DistEngine&&) = delete;62 DistEngine& operator=(DistEngine&&) = delete;66 DistEngine();67 ~DistEngine();167 DistEngine::getInstance().cleanupBackwardPass(autogradContext_); in ~BackwardPassCleanupGuard()
78 void DistEngine::globalCpuThread( in globalCpuThread()112 DistEngine::DistEngine() in DistEngine() function in torch::distributed::autograd::DistEngine117 &DistEngine::globalCpuThread, in DistEngine()137 DistEngine::~DistEngine() { in ~DistEngine()144 DistEngine& DistEngine::getInstance() { in getInstance()146 static DistEngine* engine = new DistEngine(); in getInstance()150 void DistEngine::validateRootsAndRetrieveEdges( in validateRootsAndRetrieveEdges()180 void DistEngine::computeDependencies( in computeDependencies()346 void DistEngine::execute_graph_task_until_ready_queue_empty( in execute_graph_task_until_ready_queue_empty()402 c10::intrusive_ptr<c10::ivalue::Future> DistEngine::[all …]
65 auto& engine = DistEngine::getInstance(); in TEST_F()84 auto& engine = DistEngine::getInstance(); in TEST_F()
19 DistEngine::getInstance().execute(context_id, roots, retain_graph); in backward()
133 []() { return DistEngine::getInstance().getDebugInfo(); }, in dist_autograd_init()
82 friend class DistEngine; variable
155 autograd::DistEngine::getInstance().execute( in runTrainingLoop()
373 auto execFuture = DistEngine::getInstance().executeSendFunctionAsync( in processBackwardAutogradReq()