1 #include "caffe2/core/event.h" 4 #include <condition_variable> 10 : status_(EventStatus::EVENT_INITIALIZED) {
12 option.device_type() == PROTO_CPU ||
13 option.device_type() == PROTO_MKLDNN ||
14 option.device_type() == PROTO_IDEEP,
15 "Expected CPU/MKLDNN/IDEEP device type");
20 std::condition_variable cv_completed_;
21 std::atomic<int> status_;
23 std::vector<EventCallbackFunction> callbacks_;
26 void EventCreateCPU(
const DeviceOption& option,
Event* event);
33 void EventFinishCPU(
const Event* event);
35 void EventWaitCPUCPU(
const Event* event,
void* );
37 EventStatus EventQueryCPU(
const Event* event);
39 const std::string& EventErrorMessageCPU(
const Event* event);
41 void EventSetFinishedCPU(
const Event* event,
const char* err_msg);
43 bool EventCanScheduleCPU(
const Event*,
const Event*);
45 void EventResetCPU(
Event*);
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...