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
|
// (c) Meta Platforms, Inc. and affiliates. Confidential and proprietary.
#include <memory>
#include <set>
#include <vector>
#include "output_base.h"
#include "test/MockActivitySubProfiler.h"
namespace libkineto {
const std::set<ActivityType> supported_activities {ActivityType::CPU_OP};
const std::string profile_name{"MockProfiler"};
void MockProfilerSession::processTrace(ActivityLogger& logger) {
for (const auto& activity: test_activities_) {
activity.log(logger);
}
}
const std::string& MockActivityProfiler::name() const {
return profile_name;
}
const std::set<ActivityType>& MockActivityProfiler::availableActivities() const {
return supported_activities;
}
MockActivityProfiler::MockActivityProfiler(
std::deque<GenericTraceActivity>& activities) :
test_activities_(activities) {};
std::unique_ptr<IActivityProfilerSession> MockActivityProfiler::configure(
const std::set<ActivityType>& /*activity_types*/,
const Config& /*config*/) {
auto session = std::make_unique<MockProfilerSession>();
session->set_test_activities(std::move(test_activities_));
return session;
};
std::unique_ptr<IActivityProfilerSession> MockActivityProfiler::configure(
int64_t /*ts_ms*/,
int64_t /*duration_ms*/,
const std::set<ActivityType>& activity_types,
const Config& config) {
return configure(activity_types, config);
};
std::unique_ptr<CpuTraceBuffer> MockProfilerSession::getTraceBuffer() {
auto buf = std::make_unique<CpuTraceBuffer>();
for (auto& i : test_activities_) {
buf->emplace_activity(std::move(i));
}
test_activities_.clear();
return buf;
}
} // namespace libkineto
|
