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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
#include "caffe2/core/common.h"
#include "caffe2/core/net.h"
#include "caffe2/core/observer.h"
#include "caffe2/core/operator.h"
#include "time_observer.h"
#include <gtest/gtest.h>
#include <chrono>
#include <thread>
namespace caffe2 {
namespace {
class SleepOp final : public OperatorBase {
public:
using OperatorBase::OperatorBase;
bool Run(int /* unused */) override {
StartAllObservers();
std::this_thread::sleep_for(std::chrono::milliseconds(3000));
StopAllObservers();
return true;
}
};
REGISTER_CPU_OPERATOR(SleepOp, SleepOp);
REGISTER_CUDA_OPERATOR(SleepOp, SleepOp);
OPERATOR_SCHEMA(SleepOp)
.NumInputs(0, INT_MAX)
.NumOutputs(0, INT_MAX)
.AllowInplace({{0, 0}, {1, 1}});
unique_ptr<NetBase> CreateNetTestHelper(Workspace* ws) {
NetDef net_def;
{
auto& op = *(net_def.add_op());
op.set_type("SleepOp");
op.add_input("in");
op.add_output("hidden");
}
{
auto& op = *(net_def.add_op());
op.set_type("SleepOp");
op.add_input("hidden");
op.add_output("out");
}
net_def.add_external_input("in");
net_def.add_external_output("out");
return CreateNet(net_def, ws);
}
} // namespace
TEST(TimeObserverTest, Test3Seconds) {
Workspace ws;
ws.CreateBlob("in");
NetDef net_def;
unique_ptr<NetBase> net(CreateNetTestHelper(&ws));
auto net_ob = std::make_unique<TimeObserver>(net.get());
const auto* ob = net_ob.get();
net->AttachObserver(std::move(net_ob));
net->Run();
CAFFE_ENFORCE(ob);
// NOLINTNEXTLINE(clang-analyzer-cplusplus.NewDelete)
LOG(INFO) << "av time children: " << ob->average_time_children();
LOG(INFO) << "av time: " << ob->average_time();
CAFFE_ENFORCE(ob->average_time_children() > 3000);
CAFFE_ENFORCE(ob->average_time_children() < 3500);
CAFFE_ENFORCE(ob->average_time() > 6000);
CAFFE_ENFORCE(ob->average_time() < 6500);
}
} // namespace caffe2
|
