diff options
| author | Adam Paszke <adam.paszke@gmail.com> | 2017-11-20 18:44:45 +0100 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2017-11-20 12:44:45 -0500 |
| commit | cf407213f929d2beca4e82135f00738b6e234712 (patch) | |
| tree | 62e90ae95b87fbe0cfb2a9658a1fca004913bb35 /torch/csrc/autograd/engine.h | |
| parent | 38cd6b3bd0d5b4e33693b96731edce1cfa825d62 (diff) | |
| download | pytorch-cf407213f929d2beca4e82135f00738b6e234712.tar.gz pytorch-cf407213f929d2beca4e82135f00738b6e234712.tar.bz2 pytorch-cf407213f929d2beca4e82135f00738b6e234712.zip | |
Clean up stochastic function related dead code (#3782)
Diffstat (limited to 'torch/csrc/autograd/engine.h')
| -rw-r--r-- | torch/csrc/autograd/engine.h | 8 |
1 files changed, 1 insertions, 7 deletions
diff --git a/torch/csrc/autograd/engine.h b/torch/csrc/autograd/engine.h index ff1ae45cbd..9d1c3aaea5 100644 --- a/torch/csrc/autograd/engine.h +++ b/torch/csrc/autograd/engine.h @@ -27,7 +27,6 @@ struct Engine { virtual ~Engine(); using ready_queue_type = std::deque<std::pair<std::shared_ptr<Function>, InputBuffer>>; - using function_queue = std::vector<Function*>; using dependencies_type = std::unordered_map<Function*, int>; using pre_callback_type = std::function<bool (Function*, variable_list&)>; @@ -47,12 +46,7 @@ struct Engine { void queue_callback(std::function<void()> callback); protected: - function_queue find_roots( - const function_list& roots, - variable_list& inputs, - GraphTask& task); - void find_stochastic_functions(function_queue& queue, Function* graph_root, GraphTask& task); - void compute_dependencies(function_queue queue, GraphTask& task); + void compute_dependencies(Function* root, GraphTask& task); void evaluate_function(FunctionTask& task); ReadyQueue& ready_queue(int device); void start_threads(); |