1 #ifndef CAFFE2_CORE_NET_ASYNC_SCHEDULING_H_ 2 #define CAFFE2_CORE_NET_ASYNC_SCHEDULING_H_ 4 #include "caffe2/core/net_async_base.h" 11 const std::shared_ptr<const NetDef>& net_def,
18 bool DoRunAsync()
override;
20 void pollAndSchedule(
int task_id);
21 void schedule(
int task_id);
23 virtual void finishRun();
24 int updateParentCount(
int child_id);
26 std::mutex running_mutex_;
27 std::condition_variable running_cv_;
28 std::atomic<bool> running_;
29 std::atomic<bool> success_;
31 std::mutex cleanup_mutex_;
32 std::atomic<bool> cleanup_;
34 std::atomic<int> processed_tasks_num_;
35 std::mutex exception_mutex_;
36 std::vector<std::string> exception_messages_;
43 #endif // CAFFE2_CORE_NET_ASYNC_SCHEDULING_H_
Workspace is a class that holds all the related objects created during runtime: (1) all blobs...
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...