1 #include "caffe2/core/event.h" 2 #include "caffe2/core/operator.h" 10 : status_(EventStatus::EVENT_INITIALIZED) {
12 option.device_type() == CPU || option.device_type() == MKLDNN,
13 "Expected CPU/MKLDNN device type");
18 std::condition_variable cv_completed_;
19 std::atomic<int> status_;
23 void EventCreateCPU(
const DeviceOption& option,
Event* event);
30 void EventFinishCPU(
const Event* event);
32 void EventWaitCPUCPU(
const Event* event,
void* );
34 EventStatus EventQueryCPU(
const Event* event);
36 const std::string& EventErrorMessageCPU(
const Event* event);
38 void EventSetFinishedCPU(
const Event* event,
const char* err_msg);
40 bool EventCanScheduleCPU(
const Event*,
const Event*);
42 void EventResetCPU(
Event*);
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...