bug fixed logs print copy

zhangmeng

2020-01-17 f7c4a3cfd07adede3308f8d9d3d7315427d90a7c

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#include "caffe2/core/event.h"
 
#include <atomic>
#include <condition_variable>
 
namespace caffe2 {
 
struct CPUEventWrapper {
  explicit CPUEventWrapper(const DeviceOption& option)
      : status_(EventStatus::EVENT_INITIALIZED) {
    CAFFE_ENFORCE(
        option.device_type() == PROTO_CPU ||
            option.device_type() == PROTO_MKLDNN ||
            option.device_type() == PROTO_IDEEP,
        "Expected CPU/MKLDNN/IDEEP device type");
  }
  ~CPUEventWrapper() {}
 
  std::mutex mutex_;
  std::condition_variable cv_completed_;
  std::atomic<int> status_;
  std::string err_msg_;
  std::vector<EventCallbackFunction> callbacks_;
};
 
void EventCreateCPU(const DeviceOption& option, Event* event);
 
void EventRecordCPU(
    Event* event,
    const void* /* unused */,
    const char* err_msg);
 
void EventFinishCPU(const Event* event);
 
void EventWaitCPUCPU(const Event* event, void* /* context */);
 
EventStatus EventQueryCPU(const Event* event);
 
const std::string& EventErrorMessageCPU(const Event* event);
 
void EventSetFinishedCPU(const Event* event, const char* err_msg);
 
bool EventCanScheduleCPU(const Event*, const Event*);
 
void EventResetCPU(Event*);
 
} // namespace caffe2