diff options
Diffstat (limited to 'torch/csrc/autograd/engine.h')
| -rw-r--r-- | torch/csrc/autograd/engine.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/torch/csrc/autograd/engine.h b/torch/csrc/autograd/engine.h index db8b3357ac..94490303cc 100644 --- a/torch/csrc/autograd/engine.h +++ b/torch/csrc/autograd/engine.h @@ -57,7 +57,7 @@ protected: ReadyQueue& ready_queue(int device); void start_threads(); virtual void thread_init(int device); - virtual void thread_main(GraphTask *task); + virtual void thread_main(GraphTask *graph_task); virtual void thread_on_exception(FunctionTask& task, std::exception& e); std::once_flag start_threads_flag; |