Searched refs:local_graph_task (Results 1 – 2 of 2) sorted by relevance
532 std::shared_ptr<GraphTask> local_graph_task; in thread_main() local545 local_graph_task = task.base_.lock(); in thread_main()546 if (!local_graph_task) { in thread_main()554 if (task.fn_ && !local_graph_task->has_error_.load()) { in thread_main()558 at::ThreadLocalStateGuard tls_guard(local_graph_task->thread_locals_); in thread_main()560 &local_graph_task->warning_handler_); in thread_main()567 GraphTaskGuard guard(local_graph_task); in thread_main()576 local_graph_task, in thread_main()579 local_graph_task->cpu_ready_queue_); in thread_main()583 thread_on_exception(local_graph_task, task.fn_, e); in thread_main()[all …]
365 std::shared_ptr<GraphTask> local_graph_task; in execute_graph_task_until_ready_queue_empty() local371 if (!(local_graph_task = task.base_.lock())) { in execute_graph_task_until_ready_queue_empty()374 if (task.fn_ && !local_graph_task->has_error_.load()) { in execute_graph_task_until_ready_queue_empty()375 at::ThreadLocalStateGuard tls_guard(local_graph_task->thread_locals_); in execute_graph_task_until_ready_queue_empty()377 GraphTaskGuard guard(local_graph_task); in execute_graph_task_until_ready_queue_empty()379 local_graph_task, task.fn_.get(), task.inputs_, cpu_ready_queue); in execute_graph_task_until_ready_queue_empty()381 engine_.thread_on_exception(local_graph_task, task.fn_, e); in execute_graph_task_until_ready_queue_empty()390 --local_graph_task->outstanding_tasks_; in execute_graph_task_until_ready_queue_empty()