Caffe2 - C++ API
A deep learning, cross platform ML framework
mkl_context.cc
1 
17 // #include "caffe2/mkl/utils/mkl_context.h"
18 
19 #include "caffe2/core/event_cpu.h"
20 
21 namespace caffe2 {
22 
23 // MKL events are the same as CPU events
24 
25 REGISTER_EVENT_CREATE_FUNCTION(MKLDNN, EventCreateCPU);
26 REGISTER_EVENT_RECORD_FUNCTION(MKLDNN, EventRecordCPU);
27 REGISTER_EVENT_WAIT_FUNCTION(MKLDNN, MKLDNN, EventWaitCPUCPU);
28 REGISTER_EVENT_WAIT_FUNCTION(MKLDNN, CPU, EventWaitCPUCPU);
29 REGISTER_EVENT_WAIT_FUNCTION(CPU, MKLDNN, EventWaitCPUCPU);
30 REGISTER_EVENT_FINISH_FUNCTION(MKLDNN, EventFinishCPU);
31 
32 REGISTER_EVENT_QUERY_FUNCTION(MKLDNN, EventQueryCPU);
33 REGISTER_EVENT_ERROR_MESSAGE_FUNCTION(MKLDNN, EventErrorMessageCPU);
34 REGISTER_EVENT_SET_FINISHED_FUNCTION(MKLDNN, EventSetFinishedCPU);
35 REGISTER_EVENT_RESET_FUNCTION(MKLDNN, EventResetCPU);
36 
37 } // namespace caffe2
Copyright (c) 2016-present, Facebook, Inc.