mkl_context.cc

// #include "caffe2/mkl/utils/mkl_context.h"

#include "caffe2/core/event_cpu.h"

namespace caffe2 {

// MKL events are the same as CPU events

REGISTER_EVENT_CREATE_FUNCTION(MKLDNN, EventCreateCPU);
REGISTER_EVENT_RECORD_FUNCTION(MKLDNN, EventRecordCPU);
REGISTER_EVENT_WAIT_FUNCTION(MKLDNN, MKLDNN, EventWaitCPUCPU);
REGISTER_EVENT_WAIT_FUNCTION(MKLDNN, CPU, EventWaitCPUCPU);
REGISTER_EVENT_WAIT_FUNCTION(CPU, MKLDNN, EventWaitCPUCPU);
REGISTER_EVENT_FINISH_FUNCTION(MKLDNN, EventFinishCPU);

REGISTER_EVENT_QUERY_FUNCTION(MKLDNN, EventQueryCPU);
REGISTER_EVENT_ERROR_MESSAGE_FUNCTION(MKLDNN, EventErrorMessageCPU);
REGISTER_EVENT_SET_FINISHED_FUNCTION(MKLDNN, EventSetFinishedCPU);
REGISTER_EVENT_RESET_FUNCTION(MKLDNN, EventResetCPU);

} // namespace caffe2