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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
//
// Copyright © 2020 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
#include <armnn/profiling/ITimelineDecoder.hpp>
#include <map>
#include <vector>
#include <boost/filesystem/path.hpp>
#include <boost/filesystem.hpp>
namespace armnn
{
namespace timelinedecoder
{
class JSONTimelineDecoder : public ITimelineDecoder
{
public:
struct JSONEntity
{
public:
std::vector<uint64_t> connected_entities;
std::vector<uint64_t> childEntities;
JSONEntity(uint64_t guid): m_Guid(guid){}
uint64_t GetGuid();
std::string GetName();
std::string GetType();
void SetName(std::string entityName);
void SetType(std::string entityType);
void SetParent(JSONEntity& parent);
void AddConnection(JSONEntity& headEntity, JSONEntity& connectedEntity);
std::map<std::string, std::string> extendedData;
private:
uint64_t m_Guid;
std::string name;
std::string type;
};
struct Model
{
std::map<uint64_t, JSONEntity> jsonEntities;
std::map<uint64_t, Relationship> relationships;
std::map<uint64_t, Label> labels;
std::map<uint64_t, Event> events;
std::map<uint64_t, EventClass> eventClasses;
};
void PrintJSON(JSONEntity& entity);
std::string GetJSONString(JSONEntity& rootEntity);
std::string GetJSONEntityString(JSONEntity& entity, int& counter);
virtual TimelineStatus CreateEntity(const Entity&) override;
virtual TimelineStatus CreateEventClass(const EventClass&) override;
virtual TimelineStatus CreateEvent(const Event&) override;
virtual TimelineStatus CreateLabel(const Label&) override;
virtual TimelineStatus CreateRelationship(const Relationship&) override;
const Model& GetModel();
void SetOutgoingCaptureFile(const std::string& basicString);
private:
Model m_Model;
boost::filesystem::path fileDir = boost::filesystem::temp_directory_path();
boost::filesystem::path p{fileDir / boost::filesystem::unique_path("output.json")};
std::string outputJSONFile = p.string();
void HandleRetentionLink(const Relationship& relationship);
void HandleLabelLink(const Relationship& relationship);
void HandleExecutionLink(const Relationship& relationship);
void HandleConnectionLabel(const Relationship& relationship);
void HandleBackendIdLabel(const Relationship& relationship);
void HandleNameLabel(const Relationship& relationship);
void HandleTypeLabel(const Relationship& relationship);
std::string GetLayerJSONString(JSONEntity& entity, int& counter, std::string& jsonEntityString);
std::string GetWorkloadJSONString(const JSONEntity& entity, int& counter, std::string& jsonEntityString);
std::string GetWorkloadExecutionJSONString(const JSONEntity& entity, std::string& jsonEntityString) const;
};
}
}
|
